init commit

2024-11-17 02:31:42 +03:00
commit cf933c770c
217 changed files with 19340 additions and 0 deletions
@@ -0,0 +1,151 @@
+import typing
+import uuid
+
+import fastapi
+import sqlmodel
+
+import app.api.auth.deps
+from app.api.auth.deps import BearerAuth
+from app.api.events.routers import events_router
+import app.core.db
+from app.models import Event
+from app.models import OutputEvent
+import app.models.event
+from app.models.user import User
+
+
+@events_router.get(
+    '/',
+    response_model=list[app.models.event.OutputEvent],
+    description='Return events containing given user (by token)',
+    dependencies=[fastapi.Depends(BearerAuth())],
+)
+def list_events(
+    user: typing.Annotated[
+        User, fastapi.Depends(app.api.auth.deps.get_current_user)
+    ],
+):
+    output = []
+
+    with sqlmodel.Session(app.core.db.engine) as session:
+        user = session.get(User, user.id)
+
+        for event in user.events:
+            new_output = app.models.event.OutputEvent(
+                id=event.id,
+                owner=event.owner_id,
+                users=event.users,
+                invite=event.invite,
+                name=event.name,
+            )
+            output.append(new_output)
+
+    return output
+
+
+@events_router.post(
+    '/add/{event_id}',
+    response_model=Event,
+    description='Add user to event',
+    dependencies=[fastapi.Depends(BearerAuth())],
+    responses={
+        fastapi.status.HTTP_404_NOT_FOUND: {
+            'model': app.models.base.BasicResponse,
+            'detail': r'Event \ user not found',
+        },
+    },
+)
+async def add_to_event(
+    user: typing.Annotated[
+        User, fastapi.Depends(app.api.auth.deps.get_current_user)
+    ],
+    event_id: uuid.UUID,
+):
+    with sqlmodel.Session(app.core.db.engine) as session:
+        event = session.get(Event, event_id)
+
+        if not event:
+            raise fastapi.HTTPException(
+                detail='Event not found', status_code=404
+            )
+
+        user = session.get(User, user.id)
+
+        if not user:
+            raise fastapi.HTTPException(
+                detail='User not found', status_code=404
+            )
+
+        event.users.append(user)
+
+        session.add(event)
+        session.commit()
+        session.refresh(event)
+
+    return event
+
+
+@events_router.post(
+    '/',
+    description='Create event',
+    dependencies=[fastapi.Depends(BearerAuth())],
+)
+def create_event(
+    event: app.models.event.BaseEvent,
+    user: typing.Annotated[
+        User, fastapi.Depends(app.api.auth.deps.get_current_user)
+    ],
+) -> app.models.event.Event:
+    with sqlmodel.Session(app.core.db.engine) as session:
+        new_event = app.models.event.Event(
+            name=event.name, owner_id=user.id, users=[user]
+        )
+        session.add(new_event)
+        session.commit()
+        session.refresh(new_event)
+
+    return new_event
+
+
+# @events_router.delete(
+#     '/{event_id}',
+#     dependencies=[fastapi.Depends(app.api.auth.deps.get_current_user)],
+#     description='Delete an event',
+# )
+# def delete_event(
+#     event_id: uuid.UUID,
+# ):
+#     with sqlmodel.Session(app.core.db.engine) as session:
+#         session.delete(session.get(app.models.event.Event, event_id))
+#         session.commit()
+
+
+@events_router.get(
+    '/{event_id}',
+    response_model=app.models.event.OutputEvent,
+    dependencies=[fastapi.Depends(app.api.auth.deps.get_current_user)],
+    description='Get info for an event by the id',
+    responses={
+        fastapi.status.HTTP_404_NOT_FOUND: {
+            'model': app.models.base.BasicResponse,
+            'detail': r'Event \ user not found',
+        },
+    },
+)
+def event_by_id(event_id: uuid.UUID):
+    with sqlmodel.Session(app.core.db.engine) as session:
+        event = session.get(app.models.event.Event, event_id)
+
+        if not event:
+            raise fastapi.HTTPException(
+                detail='Event not found', status_code=404
+            )
+
+        new_event = app.models.event.OutputEvent(
+            id=event.id,
+            owner=event.owner_id,
+            users=event.users,
+            invite=event.invite,
+            name=event.name,
+        )
+        return new_event