Add authentification
This commit is contained in:
@@ -16,6 +16,23 @@ def get_all(
|
||||
def get_one(session: Session, user_id: int) -> models.UserPublic:
|
||||
return session.get(models.User, user_id)
|
||||
|
||||
def get_or_create_roles(session: Session, role_names) -> list[models.ContractType]:
|
||||
statement = select(models.ContractType).where(models.ContractType.name.in_(role_names))
|
||||
existing = session.exec(statement).all()
|
||||
existing_roles = {role.name for role in existing}
|
||||
missing_role = set(role_names) - existing_roles
|
||||
|
||||
new_roles = []
|
||||
for role_name in missing_role:
|
||||
role = models.ContractType(name=role_name)
|
||||
session.add(role)
|
||||
new_roles.append(role)
|
||||
|
||||
session.commit()
|
||||
for role in new_roles:
|
||||
session.refresh(role)
|
||||
return existing + new_roles
|
||||
|
||||
def get_or_create_user(session: Session, user_create: models.UserCreate):
|
||||
statement = select(models.User).where(models.User.email == user_create.email)
|
||||
user = session.exec(statement).first()
|
||||
@@ -24,9 +41,20 @@ def get_or_create_user(session: Session, user_create: models.UserCreate):
|
||||
user = create_one(session, user_create)
|
||||
return user
|
||||
|
||||
def get_roles(session: Session):
|
||||
statement = select(models.ContractType)
|
||||
return session.exec(statement.order_by(models.ContractType.name)).all()
|
||||
|
||||
def create_one(session: Session, user: models.UserCreate) -> models.UserPublic:
|
||||
user_create = user.model_dump(exclude_unset=True)
|
||||
new_user = models.User(**user_create)
|
||||
print("USER CREATE", user)
|
||||
new_user = models.User(
|
||||
name=user.name,
|
||||
email=user.email
|
||||
)
|
||||
|
||||
roles = get_or_create_roles(session, user.role_names)
|
||||
new_user.roles = roles
|
||||
|
||||
session.add(new_user)
|
||||
session.commit()
|
||||
session.refresh(new_user)
|
||||
|
||||
Reference in New Issue
Block a user