cloud-api/src/user/service.py

"""
Module specific business logic for user module

Functions:
 - add_user_to_db

Exports:
 - add_user_to_db
"""
from typing import Any

from src.database import get_db
from src.exceptions import UnprocessableContent

from src.user.schemas import OIDCUser
from src.user.models import User


async def add_user_to_db(user_claims: dict[str, Any]) -> int:
	try:
		valid_user = OIDCUser(first_name=user_claims["given_name"], last_name=user_claims["family_name"], email=user_claims["email"], oidc_id=user_claims["sub"])
	except Exception as e:
		print(e)
		raise UnprocessableContent("Invalid or missing OIDC data")

	db = next(get_db())
	db_user = db.query(User).filter(User.oidc_id == valid_user.oidc_id).first()

	if not db_user:
		user_model = User(**valid_user.model_dump())
		db.add(user_model)
		db.commit()
		return user_model.id
	else:
		change = False
		if db_user.first_name != valid_user.first_name:
			db_user.first_name = valid_user.first_name
			change = True
		if db_user.last_name != valid_user.last_name:
			db_user.last_name = valid_user.last_name
			change = True
		if change:
			db.add(db_user)
			db.commit()
		return db_user.id