116 lines
4.2 KiB
Python
116 lines
4.2 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy import select
|
|
from typing import List
|
|
from app.db.session import get_db
|
|
from app.schemas.organization import CorpOnboardIn, CorpOnboardResponse
|
|
from app.models.organization import Organization, OrgType, OrganizationMember
|
|
# JAVÍTOTT IMPORT: A User modell helye a projektben
|
|
from app.models.user import User
|
|
from app.core.config import settings
|
|
import os
|
|
import re
|
|
import logging
|
|
|
|
router = APIRouter()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@router.post("/onboard", response_model=CorpOnboardResponse, status_code=status.HTTP_201_CREATED)
|
|
async def onboard_organization(
|
|
org_in: CorpOnboardIn,
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""
|
|
Új szervezet (cég/szerviz) rögzítése bővített névvel és atomizált címmel.
|
|
"""
|
|
|
|
# 1. Magyar adószám validáció (XXXXXXXX-Y-ZZ)
|
|
if org_in.country_code == "HU":
|
|
if not re.match(r"^\d{8}-\d-\d{2}$", org_in.tax_number):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="Érvénytelen magyar adószám formátum!"
|
|
)
|
|
|
|
# 2. Duplikáció ellenőrzés
|
|
stmt_exist = select(Organization).where(Organization.tax_number == org_in.tax_number)
|
|
result_exist = await db.execute(stmt_exist)
|
|
if result_exist.scalar_one_or_none():
|
|
raise HTTPException(
|
|
status_code=status.HTTP_409_CONFLICT,
|
|
detail="Ezzel az adószámmal már regisztráltak céget!"
|
|
)
|
|
|
|
# 3. Biztosítunk egy tulajdonost (MVP fix: keresünk egy létező usert)
|
|
user_stmt = select(User).limit(1)
|
|
user_res = await db.execute(user_stmt)
|
|
test_user = user_res.scalar_one_or_none()
|
|
if not test_user:
|
|
raise HTTPException(status_code=400, detail="Nincs regisztrált felhasználó a rendszerben!")
|
|
|
|
# 4. Mentés (Szervezet létrehozása atomizált adatokkal és név-hierarchiával)
|
|
new_org = Organization(
|
|
full_name=org_in.full_name,
|
|
name=org_in.name,
|
|
display_name=org_in.display_name,
|
|
tax_number=org_in.tax_number,
|
|
reg_number=org_in.reg_number,
|
|
address_zip=org_in.address_zip,
|
|
address_city=org_in.address_city,
|
|
address_street_name=org_in.address_street_name,
|
|
address_street_type=org_in.address_street_type,
|
|
address_house_number=org_in.address_house_number,
|
|
address_hrsz=org_in.address_hrsz,
|
|
address_stairwell=org_in.address_stairwell,
|
|
address_floor=org_in.address_floor,
|
|
address_door=org_in.address_door,
|
|
country_code=org_in.country_code,
|
|
org_type=OrgType.business,
|
|
status="pending_verification"
|
|
)
|
|
|
|
db.add(new_org)
|
|
await db.flush()
|
|
|
|
# 5. TULAJDONOS RÖGZÍTÉSE (Membership lánc)
|
|
owner_member = OrganizationMember(
|
|
organization_id=new_org.id,
|
|
user_id=test_user.id,
|
|
role="owner"
|
|
)
|
|
db.add(owner_member)
|
|
|
|
# 6. NAS Mappa létrehozása (Org izoláció)
|
|
try:
|
|
base_path = getattr(settings, "NAS_STORAGE_PATH", "/mnt/nas/app_data")
|
|
org_path = os.path.join(base_path, "organizations", str(new_org.id))
|
|
os.makedirs(os.path.join(org_path, "documents"), exist_ok=True)
|
|
logger.info(f"NAS mappa struktúra kész: {org_path}")
|
|
except Exception as e:
|
|
logger.error(f"NAS hiba: {e}")
|
|
|
|
await db.commit()
|
|
await db.refresh(new_org)
|
|
|
|
return {"organization_id": new_org.id, "status": new_org.status}
|
|
|
|
@router.get("/my", response_model=List[CorpOnboardResponse])
|
|
async def get_my_organizations(
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""
|
|
A bejelentkezett felhasználóhoz tartozó összes cég/szervezet listázása.
|
|
"""
|
|
# MVP Teszt: Kézzel keresünk egy létező usert (később: current_user.id)
|
|
user_stmt = select(User).limit(1)
|
|
user_res = await db.execute(user_stmt)
|
|
test_user = user_res.scalar_one_or_none()
|
|
|
|
if not test_user:
|
|
return []
|
|
|
|
stmt = select(Organization).join(OrganizationMember).where(OrganizationMember.user_id == test_user.id)
|
|
result = await db.execute(stmt)
|
|
orgs = result.scalars().all()
|
|
|
|
return [{"organization_id": o.id, "status": o.status} for o in orgs] |