From 9fd201c10adf11802d2326ecf919f42c85c6425a Mon Sep 17 00:00:00 2001 From: Gentile G Date: Wed, 7 May 2025 17:39:37 +0200 Subject: [PATCH] WIP trying to correct firm initialization --- api/rpk-api/firm/current_firm/routes.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/api/rpk-api/firm/current_firm/routes.py b/api/rpk-api/firm/current_firm/routes.py index b2afd33..bec0b2c 100644 --- a/api/rpk-api/firm/current_firm/routes.py +++ b/api/rpk-api/firm/current_firm/routes.py @@ -31,14 +31,22 @@ async def create(schema: CurrentFirmSchemaCreate, reg=Depends(get_uninitialized_ corporation_schema.entity_data.employees.append(Employee(entity_id=owner_entity.id, position=schema.position)) corp = await Entity.create(reg.db, corporation_schema) - document = await CurrentFirm.create(reg.db, CurrentFirm( + firm = await CurrentFirm.create(reg.db, CurrentFirm( instance=reg.instance, firm=reg.firm, entity_id=corp.id, primary_color=schema.primary_color, secondary_color=schema.secondary_color, )) - return CurrentFirmSchemaRead.from_model_and_entities(document, EntityRead.from_model(corp), EntityRead.from_model(owner_entity)) + + await Partner.create(Partner(user_id=reg.user.id, entity_id=owner_entity.id)) + + return CurrentFirmSchemaRead.from_model_and_entities( + firm, + EntityRead.from_model(corp), + EntityRead.from_model(owner_entity), + [EntityRead.from_model(owner_entity)] + ) @current_firm_router.put("/", response_description=f"Current Firm record updated") async def update(schema: CurrentFirmSchemaUpdate, reg=Depends(get_authed_tenant_registry)) -> CurrentFirmSchemaRead: