46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from core.database import fake_users_db
|
|
from pydantic import BaseModel
|
|
import uuid
|
|
|
|
router = APIRouter()
|
|
|
|
class CompanyRegistration(BaseModel):
|
|
company_name: str
|
|
user_name: str
|
|
email: str
|
|
password: str
|
|
role: str = "employee"
|
|
|
|
@router.post("/register/company")
|
|
async def register_company_user(registration: CompanyRegistration):
|
|
"""Register a new user in a company"""
|
|
user_id = str(uuid.uuid4())
|
|
|
|
if registration.email in [user["email"] for user in fake_users_db.values()]:
|
|
raise HTTPException(status_code=400, detail="Email already registered")
|
|
|
|
fake_users_db[registration.user_name] = {
|
|
"id": user_id,
|
|
"company_name": registration.company_name,
|
|
"email": registration.email,
|
|
"password": registration.password,
|
|
"role": registration.role,
|
|
"disabled": False
|
|
}
|
|
|
|
return {
|
|
"message": "Company user registered successfully",
|
|
"user_id": user_id,
|
|
"company_name": registration.company_name,
|
|
"username": registration.user_name,
|
|
"next_steps": [
|
|
"Verify your email address",
|
|
"Complete company profile",
|
|
"Set up team members"
|
|
],
|
|
"features": {
|
|
"rate_limit": 100,
|
|
"expires_in": 3600
|
|
}
|
|
} |