import sqlite3 from fastapi import FastAPI, status from fastapi.responses import JSONResponse, Response from backend.adapters.sqlite import create_db from backend.api.model import StudentModel, TribeModel from backend.model.student import Student from backend.model.tribe import Tribe from backend.repository.student_sqlite_repository import StudentSQLiteRepository from backend.repository.tribe_sqlite_repository import ( TribeRepositoryError, TribeSQLiteRepository, ) from backend.service import services from backend.service.services import StudentDoesExist, TribeDoesNotExist, TribeExists # from sqlalchemy import create_engine # from sqlalchemy.orm import clear_mappers, sessionmaker # import backend.adapters.orm as orm # from backend.repository.tribe_sqlalchemy_repository import TribeSQLAlchemyRepository # orm.start_mappers() # engine = create_engine("sqlite:///:memory:") # orm.metadata.create_all(engine) # session = sessionmaker(bind=engine)() # tribe_repo = TribeSQLAlchemyRepository(session) conn = sqlite3.connect("sqlite.db") create_db(conn) tribe_repo = TribeSQLiteRepository(conn) student_repo = StudentSQLiteRepository(conn) app = FastAPI() @app.post("/tribes", status_code=status.HTTP_201_CREATED, response_model=TribeModel) async def post_tribe(item: TribeModel): try: tribe = services.add_tribe( name=item.name, level=item.level, tribe_repo=tribe_repo, conn=conn ) except TribeExists: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The tribe {item.name} already exists", ) return tribe.to_dict() @app.put("/tribes/{name}", status_code=status.HTTP_200_OK, response_model=TribeModel) async def put_tribe(name: str, item: TribeModel): try: tribe = services.update_tribe( name=item.name, level=item.level, tribe_repo=tribe_repo, conn=conn ) except TribeDoesNotExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The tribe {name} does not exists", ) return tribe.to_dict() @app.delete("/tribes/{name}") async def delete_tribe(name: str): try: services.delete_tribe(name=name, tribe_repo=tribe_repo, conn=conn) except TribeDoesNotExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The tribe {name} does not exists", ) return Response( status_code=status.HTTP_204_NO_CONTENT, ) @app.get("/tribes", response_model=list[TribeModel]) async def list_tribes(): tribes = tribe_repo.list() return [t.to_dict() for t in tribes] @app.get("/tribes/{name}", response_model=TribeModel) async def get_tribe(name: str): tribe = tribe_repo.get(name) return tribe.to_dict() @app.post("/students", status_code=status.HTTP_201_CREATED, response_model=StudentModel) async def post_student(item: StudentModel): if item.id is not None: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"You can't post a student with an id. It is already registrered. Use PUT to modify it.", ) try: student = services.add_student( name=item.name, tribe=item.tribe_name, tribe_repo=tribe_repo, student_repo=student_repo, conn=conn, ) except TribeDoesNotExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The tribe {item.tribe_name} does not exists. You can't add a student in it.", ) return student.to_dict() @app.get("/students/{id}", status_code=status.HTTP_200_OK, response_model=StudentModel) async def get_student(id: str): tribes = tribe_repo.list() student = student_repo.get(id, tribes) return student.to_dict() @app.get("/students", status_code=status.HTTP_200_OK, response_model=list[StudentModel]) async def list_students(): tribes = tribe_repo.list() students = student_repo.list(tribes) return [t.to_dict() for t in students] @app.put( "/students/{student_id}", status_code=status.HTTP_200_OK, response_model=StudentModel, ) async def put_student(student_id, item: StudentModel): if student_id != item.id: return JSONResponse( status_code=status.HTTP_400_BAD_REQUEST, content=f"Url and student id are the same", ) try: student = services.update_student( id=item.id, name=item.name, tribe=item.tribe_name, tribe_repo=tribe_repo, student_repo=student_repo, conn=conn, ) except TribeDoesNotExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The tribe {tribe_name} does not exists. You can't add a student in it.", ) except StudentDoesExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The student {item.name} ({item.id=}) does not exists. You can't modify it.", ) return student.to_dict() @app.delete( "/students/{student_id}", ) async def delete_student(student_id): try: student = services.delete_student( id=student_id, student_repo=student_repo, conn=conn, ) except StudentDoesExist: return JSONResponse( status_code=status.HTTP_409_CONFLICT, content=f"The student ({student_id=}) does not exists. You can't delete it.", ) return Response( status_code=status.HTTP_204_NO_CONTENT, )