eetlijst-python 0.0.1__py3-none-any.whl

eetlijst/generated/join_group.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+
+
+class JoinGroup(BaseModel):
+    join_group: Optional["JoinGroupJoinGroup"] = Field(alias="joinGroup")
+
+
+class JoinGroupJoinGroup(BaseModel):
+    accepted: bool
+    error: Optional[str]
+
+
+JoinGroup.model_rebuild()

eetlijst/generated/list_items.py

@@ -0,0 +1,13 @@
+from .base_model import BaseModel
+from .fragments import ItemFields
+
+
+class ListItems(BaseModel):
+    eetschema_list: list["ListItemsEetschemaList"]
+
+
+class ListItemsEetschemaList(ItemFields):
+    pass
+
+
+ListItems.model_rebuild()

eetlijst/generated/remove_account.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+from .base_model import BaseModel
+
+
+class RemoveAccount(BaseModel):
+    update_eetschema_user_by_pk: Optional["RemoveAccountUpdateEetschemaUserByPk"]
+
+
+class RemoveAccountUpdateEetschemaUserByPk(BaseModel):
+    reason_to_remove_account_text: Optional[str]
+    reason_to_remove_account_selection: Optional[list[str]]
+    reason_to_remove_account_selection_all_options: Optional[list[str]]
+
+
+RemoveAccount.model_rebuild()

eetlijst/generated/settle_unsettled_expenses.py

@@ -0,0 +1,21 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import ExpenseFields
+
+
+class SettleUnsettledExpenses(BaseModel):
+    update_eetschema_expense: Optional["SettleUnsettledExpensesUpdateEetschemaExpense"]
+
+
+class SettleUnsettledExpensesUpdateEetschemaExpense(BaseModel):
+    number_expenses: int
+    expenses: list["SettleUnsettledExpensesUpdateEetschemaExpenseExpenses"]
+
+
+class SettleUnsettledExpensesUpdateEetschemaExpenseExpenses(ExpenseFields):
+    pass
+
+
+SettleUnsettledExpenses.model_rebuild()
+SettleUnsettledExpensesUpdateEetschemaExpense.model_rebuild()

eetlijst/generated/settlement_expenses.py

@@ -0,0 +1,13 @@
+from .base_model import BaseModel
+from .fragments import ExpenseFields
+
+
+class SettlementExpenses(BaseModel):
+    eetschema_expense: list["SettlementExpensesEetschemaExpense"]
+
+
+class SettlementExpensesEetschemaExpense(ExpenseFields):
+    pass
+
+
+SettlementExpenses.model_rebuild()

eetlijst/generated/update_attendance.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import AttendanceFields
+
+
+class UpdateAttendance(BaseModel):
+    update_eetschema_event_attendees_by_pk: Optional[
+        "UpdateAttendanceUpdateEetschemaEventAttendeesByPk"
+    ]
+
+
+class UpdateAttendanceUpdateEetschemaEventAttendeesByPk(AttendanceFields):
+    pass
+
+
+UpdateAttendance.model_rebuild()

eetlijst/generated/update_event.py

@@ -0,0 +1,15 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import EventFields
+
+
+class UpdateEvent(BaseModel):
+    update_eetschema_event_by_pk: Optional["UpdateEventUpdateEetschemaEventByPk"]
+
+
+class UpdateEventUpdateEetschemaEventByPk(EventFields):
+    pass
+
+
+UpdateEvent.model_rebuild()

eetlijst/generated/update_expense.py

@@ -0,0 +1,20 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import ExpenseFields
+
+
+class UpdateExpense(BaseModel):
+    update_eetschema_expense: Optional["UpdateExpenseUpdateEetschemaExpense"]
+
+
+class UpdateExpenseUpdateEetschemaExpense(BaseModel):
+    returning: list["UpdateExpenseUpdateEetschemaExpenseReturning"]
+
+
+class UpdateExpenseUpdateEetschemaExpenseReturning(ExpenseFields):
+    pass
+
+
+UpdateExpense.model_rebuild()
+UpdateExpenseUpdateEetschemaExpense.model_rebuild()

eetlijst/generated/update_expense_distribution.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+from .base_model import BaseModel
+
+
+class UpdateExpenseDistribution(BaseModel):
+    delete_eetschema_expense_distribution: Optional[
+        "UpdateExpenseDistributionDeleteEetschemaExpenseDistribution"
+    ]
+    insert_eetschema_expense_distribution: Optional[
+        "UpdateExpenseDistributionInsertEetschemaExpenseDistribution"
+    ]
+
+
+class UpdateExpenseDistributionDeleteEetschemaExpenseDistribution(BaseModel):
+    affected_rows: int
+
+
+class UpdateExpenseDistributionInsertEetschemaExpenseDistribution(BaseModel):
+    affected_rows: int
+
+
+UpdateExpenseDistribution.model_rebuild()

eetlijst/generated/update_group.py

@@ -0,0 +1,15 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import GroupFields
+
+
+class UpdateGroup(BaseModel):
+    group: Optional["UpdateGroupGroup"]
+
+
+class UpdateGroupGroup(GroupFields):
+    pass
+
+
+UpdateGroup.model_rebuild()

eetlijst/generated/update_list_item.py

@@ -0,0 +1,15 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import ItemFields
+
+
+class UpdateListItem(BaseModel):
+    update_eetschema_list_by_pk: Optional["UpdateListItemUpdateEetschemaListByPk"]
+
+
+class UpdateListItemUpdateEetschemaListByPk(ItemFields):
+    pass
+
+
+UpdateListItem.model_rebuild()

eetlijst/generated/update_many_attendance.py

@@ -0,0 +1,22 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import AttendanceFields
+
+
+class UpdateManyAttendance(BaseModel):
+    insert_eetschema_event_attendees: Optional[
+        "UpdateManyAttendanceInsertEetschemaEventAttendees"
+    ]
+
+
+class UpdateManyAttendanceInsertEetschemaEventAttendees(BaseModel):
+    returning: list["UpdateManyAttendanceInsertEetschemaEventAttendeesReturning"]
+
+
+class UpdateManyAttendanceInsertEetschemaEventAttendeesReturning(AttendanceFields):
+    pass
+
+
+UpdateManyAttendance.model_rebuild()
+UpdateManyAttendanceInsertEetschemaEventAttendees.model_rebuild()

eetlijst/generated/update_user.py

@@ -0,0 +1,15 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import UserFields
+
+
+class UpdateUser(BaseModel):
+    update_eetschema_user_by_pk: Optional["UpdateUserUpdateEetschemaUserByPk"]
+
+
+class UpdateUserUpdateEetschemaUserByPk(UserFields):
+    pass
+
+
+UpdateUser.model_rebuild()

eetlijst/generated/update_user_in_group.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import UserInGroupFields
+
+
+class UpdateUserInGroup(BaseModel):
+    update_eetschema_users_in_group_by_pk: Optional[
+        "UpdateUserInGroupUpdateEetschemaUsersInGroupByPk"
+    ]
+
+
+class UpdateUserInGroupUpdateEetschemaUsersInGroupByPk(UserInGroupFields):
+    pass
+
+
+UpdateUserInGroup.model_rebuild()

eetlijst/generated/update_users_in_group.py

@@ -0,0 +1,25 @@
+from typing import Optional
+
+from .base_model import BaseModel
+from .fragments import UserInGroupFields
+
+
+class UpdateUsersInGroup(BaseModel):
+    update_eetschema_users_in_group_many: Optional[
+        list[Optional["UpdateUsersInGroupUpdateEetschemaUsersInGroupMany"]]
+    ]
+
+
+class UpdateUsersInGroupUpdateEetschemaUsersInGroupMany(BaseModel):
+    number_users_in_group: int
+    users_in_group: list[
+        "UpdateUsersInGroupUpdateEetschemaUsersInGroupManyUsersInGroup"
+    ]
+
+
+class UpdateUsersInGroupUpdateEetschemaUsersInGroupManyUsersInGroup(UserInGroupFields):
+    pass
+
+
+UpdateUsersInGroup.model_rebuild()
+UpdateUsersInGroupUpdateEetschemaUsersInGroupMany.model_rebuild()

eetlijst/services/__init__.py

@@ -0,0 +1,9 @@
+from .app import App
+from .event_attendance import EventAttendance
+from .events import Events
+from .expenses import Expenses
+from .group_list import GroupList
+from .group_users import GroupUsers
+from .groups import Groups
+from .settlements import Settlements
+from .users import Users

eetlijst/services/app/__init__.py

@@ -0,0 +1 @@
+from .app import App

eetlijst/services/app/app.py

@@ -0,0 +1,20 @@
+from dataclasses import dataclass
+
+from eetlijst.generated import GraphQlClient
+
+from eetlijst.exceptions import EetlijstException
+
+from eetlijst.services.app.transformers import transform_app_status
+
+
+@dataclass
+class App:
+    _client: GraphQlClient
+
+    async def status(self):
+        result = await self._client.app_status()
+
+        if not result or not result.eetschema_app_status:
+            raise EetlijstException("App status not found")
+
+        return transform_app_status(result)

eetlijst/services/app/transformers.py

@@ -0,0 +1,10 @@
+from eetlijst.generated.app_status import AppStatus, AppStatusEetschemaAppStatus
+
+from eetlijst.exceptions import EetlijstException
+
+
+def transform_app_status(status: AppStatus) -> AppStatusEetschemaAppStatus:
+    if not status.eetschema_app_status:
+        raise EetlijstException("App status not found")
+
+    return status.eetschema_app_status[0]

eetlijst/services/event_attendance/__init__.py

@@ -0,0 +1 @@
+from .event_attendance import EventAttendance

eetlijst/services/event_attendance/event_attendance.py

@@ -0,0 +1,94 @@
+from dataclasses import dataclass
+
+from eetlijst.generated import GraphQlClient
+from eetlijst.generated.base_model import UNSET, UnsetType
+from eetlijst.generated.enums import AttendanceStatus
+from eetlijst.generated.input_types import (
+    eetschema_event_attendees_bool_exp,
+    eetschema_event_attendees_insert_input,
+    eetschema_event_attendees_order_by,
+    eetschema_event_attendees_set_input,
+)
+
+from .transformers import (
+    transform_attendance,
+    transform_update_attendance,
+    transform_update_many_attendance,
+)
+
+
+class EventAttendceAttend(eetschema_event_attendees_insert_input):
+    user_id: str
+    event_id: str
+    status: AttendanceStatus
+
+
+class EventAttendceUpdate(eetschema_event_attendees_set_input):
+    user_id: str
+    event_id: str
+
+
+@dataclass
+class EventAttendance:
+    _client: GraphQlClient
+
+    async def get(
+        self,
+        event_id: str,
+        user_id: str,
+    ):
+        result = await self._client.get_attendance(event_id, user_id)
+        return transform_attendance(result.eetschema_event_attendees_by_pk)
+
+    async def all(
+        self,
+        where: eetschema_event_attendees_bool_exp | UnsetType | None = UNSET,
+        order: list[eetschema_event_attendees_order_by] | UnsetType | None = UNSET,
+        limit: int | UnsetType | None = UNSET,
+    ):
+        result = await self._client.all_attendances(where, order, limit)
+        return [
+            transform_attendance(attendee)
+            for attendee in result.eetschema_event_attendees
+        ]
+
+    async def attend(self, data: EventAttendceAttend):
+        return await self.update_many([data])
+
+    async def update(self, data: EventAttendceUpdate):
+        result = await self._client.update_attendance(data.event_id, data.user_id, data)
+        return transform_update_attendance(result)
+
+    async def update_many(self, data: list[eetschema_event_attendees_insert_input]):
+        result = await self._client.update_many_attendance(data)
+        return transform_update_many_attendance(result)
+
+    async def comment(self, user_id: str, event_id: str, comment: str | None):
+        return await self.update(
+            EventAttendceUpdate(
+                user_id=user_id,
+                event_id=event_id,
+                comment=comment,
+            )
+        )
+
+    async def status(
+        self, user_id: str, event_id: str, status: AttendanceStatus | None
+    ):
+        return await self.update(
+            EventAttendceUpdate(
+                user_id=user_id,
+                event_id=event_id,
+                status=status,
+                user_changed_status=True,
+            )
+        )
+
+    async def guests(self, user_id: str, event_id: str, guests: int | None):
+        return await self.update(
+            EventAttendceUpdate(
+                user_id=user_id,
+                event_id=event_id,
+                number_guests=guests,
+            )
+        )

eetlijst/services/event_attendance/transformers.py

@@ -0,0 +1,58 @@
+from datetime import datetime
+from typing import Optional
+
+from eetlijst.generated.base_model import BaseModel
+from eetlijst.generated.enums import AttendanceStatus
+from eetlijst.generated.fragments import (
+    AttendanceFields,
+    AttendanceFieldsLinkedEvent,
+    AttendanceFieldsUserInGroup,
+)
+from eetlijst.generated.update_attendance import UpdateAttendance
+from eetlijst.generated.update_many_attendance import UpdateManyAttendance
+
+from eetlijst.exceptions import EetlijstException
+
+
+class Attendance(BaseModel):
+    created_at: datetime
+    updated_at: datetime
+    comment: Optional[str]
+    number_guests: int
+    status: AttendanceStatus
+    event: AttendanceFieldsLinkedEvent
+    user: Optional[AttendanceFieldsUserInGroup]
+
+
+def transform_attendance(attendance: Optional[AttendanceFields]) -> Attendance:
+    if not attendance:
+        raise EetlijstException("Attendance not found")
+
+    data = attendance.model_dump()
+
+    # 'linked_event' -> 'event'
+    if "linked_event" in data:
+        data["event"] = data.pop("linked_event")
+
+    # 'user_in_group' -> 'user'
+    if "user_in_group" in data:
+        data["user"] = data.pop("user_in_group")
+
+    return Attendance(**data)
+
+
+def transform_update_attendance(update: UpdateAttendance):
+    if not update.update_eetschema_event_attendees_by_pk:
+        raise EetlijstException("Failed to update attendance")
+
+    return transform_attendance(update.update_eetschema_event_attendees_by_pk)
+
+
+def transform_update_many_attendance(updates: UpdateManyAttendance):
+    if not updates.insert_eetschema_event_attendees:
+        raise EetlijstException("Failed to update attendance")
+
+    return [
+        transform_attendance(update)
+        for update in updates.insert_eetschema_event_attendees.returning
+    ]

eetlijst/services/events/__init__.py

@@ -0,0 +1 @@
+from .events import Events

eetlijst/services/events/events.py

@@ -0,0 +1,148 @@
+from dataclasses import dataclass
+from datetime import datetime
+
+from eetlijst.generated import GraphQlClient, order_by
+from eetlijst.generated.input_types import (
+    eetschema_event_bool_exp,
+    eetschema_event_order_by,
+    eetschema_event_set_input,
+    uuid_comparison_exp,
+)
+
+from eetlijst.services.event_attendance import EventAttendance
+
+from eetlijst.utils.datetime import current_datetime, format_date
+
+from .transformers import (
+    transform_automatic_events,
+    transform_event,
+    transform_update_event,
+)
+
+
+@dataclass
+class Events:
+    _client: GraphQlClient
+    attendance: EventAttendance
+
+    async def populate_dinners(self, group_id: str, date: datetime | str):
+        result = await self._client.automatic_events(
+            group_id,
+            format_date(date),
+        )
+        return transform_automatic_events(result)
+
+    async def get(
+        self,
+        event_id: str,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        result = await self._client.get_event(
+            event_id,
+            include_attendees,
+            include_expenses,
+        )
+        return transform_event(result.eetschema_event_by_pk)
+
+    async def all(
+        self,
+        group_id: str,
+        where: eetschema_event_bool_exp | None = None,
+        order: list[eetschema_event_order_by] | None = None,
+        limit: int | None = None,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        if where is not None:
+            where_data = where.model_copy(
+                update={"group_id": uuid_comparison_exp(_eq=group_id)}
+            )
+        else:
+            where_data = eetschema_event_bool_exp(
+                group_id=uuid_comparison_exp(_eq=group_id)
+            )
+
+        order_data = order
+        if order is None:
+            order_data = [eetschema_event_order_by(start_date=order_by.desc)]
+
+        result = await self._client.all_events(
+            where_data,
+            order_data,
+            limit,
+            include_attendees,
+            include_expenses,
+        )
+        return [transform_event(event) for event in result.eetschema_event]
+
+    async def update(
+        self,
+        event_id: str,
+        data: eetschema_event_set_input,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        result = await self._client.update_event(
+            event_id,
+            data,
+            include_attendees,
+            include_expenses,
+        )
+        return transform_update_event(result)
+
+    async def close(
+        self,
+        event_id: str,
+        closed_by: str,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        return await self.update(
+            event_id,
+            eetschema_event_set_input(
+                open=False,
+                changed_signup_time=True,
+                closed_by=closed_by,
+                signup_deadline=current_datetime(),
+            ),
+            include_attendees,
+            include_expenses,
+        )
+
+    async def open(
+        self,
+        event_id: str,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        return await self.update(
+            event_id,
+            eetschema_event_set_input(
+                open=True,
+                changed_signup_time=True,
+                signup_deadline=current_datetime(),
+            ),
+            include_attendees,
+            include_expenses,
+        )
+
+    async def close_at(
+        self,
+        event_id: str,
+        closed_by: str,
+        signup_deadline: datetime,
+        include_attendees: bool = False,
+        include_expenses: bool = False,
+    ):
+        return await self.update(
+            event_id,
+            eetschema_event_set_input(
+                open=False,
+                changed_signup_time=True,
+                closed_by=closed_by,
+                signup_deadline=signup_deadline,
+            ),
+            include_attendees,
+            include_expenses,
+        )