ewandor/roleplay-contract
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Improving user management and auto-refreshing user firms

Browse Source
This commit is contained in:
ewandor
2025-04-10 01:31:37 +02:00
parent bc059de65b
commit f1fe81a146
13 changed files with 131 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
api/rpk-api/hub/firm/routes.py
View File

@@ -4,44 +4,41 @@ from fastapi import APIRouter, Depends, HTTPException
from hub.auth import get_current_user
from hub.firm import Firm, FirmRead, FirmCreate, FirmUpdate
model = Firm
model_read = FirmRead
model_create = FirmCreate
model_update = FirmUpdate
router = APIRouter()
@router.post("/", response_description="{} added to the database".format(model.__name__))
async def create(item: model_create, user=Depends(get_current_user)) -> model_read:
exists = await Firm.find_one({"name": item.name, "instance": item.instance})
@router.post("/", response_description="{} added to the database".format(Firm.__name__))
async def create(item: FirmCreate, user=Depends(get_current_user)) -> FirmRead:
firm_dict = {"name": item.name, "instance": item.instance}
exists = await Firm.find_one(firm_dict)
if exists:
raise HTTPException(status_code=400, detail="Firm already exists")
record = model(created_by=user.id, updated_by=user.id, owner=user.id, **item.model_dump())
record = Firm(created_by=user.id, updated_by=user.id, owner=user.id, **item.model_dump())
o = await record.create()
user.firms.append(o.id)
user.save()
return model_read(**o.model_dump())
user.firms.append(firm_dict)
await user.save()
return FirmRead(**o.model_dump())
@router.get("/{id}", response_description="{} record retrieved".format(model.__name__))
async def read_id(id: PydanticObjectId, user=Depends(get_current_user)) -> model_read:
item = await model.get(id)
return model_read(**item.model_dump())
@router.get("/{id}", response_description="{} record retrieved".format(Firm.__name__))
async def read_id(id: PydanticObjectId, user=Depends(get_current_user)) -> FirmRead:
item = await Firm.get(id)
return FirmRead(**item.model_dump())
@router.put("/{id}", response_description="{} record updated".format(model.__name__))
async def update(id: PydanticObjectId, req: model_update, user=Depends(get_current_user)) -> model_read:
item = await model.get(id)
@router.put("/{id}", response_description="{} record updated".format(Firm.__name__))
async def update(id: PydanticObjectId, req: FirmUpdate, user=Depends(get_current_user)) -> FirmRead:
item = await Firm.get(id)
if not item:
raise HTTPException(
status_code=404,
detail="{} record not found!".format(model.__name__)
detail="{} record not found!".format(Firm.__name__)
)
if item.owner != user.id:
raise HTTPException(
status_code=403,
detail="Insufficient credentials to modify {} record".format(model.__name__)
detail="Insufficient credentials to modify {} record".format(Firm.__name__)
)
req = {k: v for k, v in req.model_dump().items() if v is not None}
@@ -50,23 +47,23 @@ async def update(id: PydanticObjectId, req: model_update, user=Depends(get_curre
}}
await item.update(update_query)
return model_read(**item.dict())
return FirmRead(**item.dict())
@router.delete("/{id}", response_description="{} record deleted from the database".format(model.__name__))
@router.delete("/{id}", response_description="{} record deleted from the database".format(Firm.__name__))
async def delete(id: PydanticObjectId, user=Depends(get_current_user)) -> dict:
item = await model.get(id)
item = await Firm.get(id)
if not item:
raise HTTPException(
status_code=404,
detail="{} record not found!".format(model.__name__)
detail="{} record not found!".format(Firm.__name__)
)
if item.owner != user.id:
raise HTTPException(
status_code=403,
detail="Insufficient credentials delete {} record".format(model.__name__)
detail="Insufficient credentials delete {} record".format(Firm.__name__)
)
await item.delete()
return {
"message": "{} deleted successfully".format(model.__name__)
"message": "{} deleted successfully".format(Firm.__name__)
}