File size: 2,624 Bytes
fc76af0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
from users.models import User, UserEmbeddings
from fastapi.exceptions import HTTPException
from core.security import get_password_hash
from datetime import datetime
from sqlalchemy.orm import Session
from users.schemas import UserCreate, UserUpdate, UserEmbeddingsModel
async def create_user_account(data:UserCreate, db:Session):
user = db.query(User).filter(User.email == data.email).first()
if user:
raise HTTPException(status_code=422, detail="Email already registered")
new_user = User(
email=data.email,
first_name=data.first_name,
last_name=data.last_name,
age=data.age,
preferences=data.preferences,
password = get_password_hash(data.password),
registered_at=datetime.now(),
updated_at=datetime.now(),
)
db.add(new_user)
db.commit()
db.refresh(new_user)
return new_user
async def create_user_embeddings(user_id:int, embeddings:UserEmbeddingsModel, db:Session):
user = db.query(User).filter(User.id == user_id).first()
if not user:
embeddings = UserEmbeddings(user_id=user_id, embeddings=embeddings.embeddings)
db.add(embeddings)
db.commit()
db.refresh(embeddings)
return embeddings
def get_user_by_id(user_id, db:Session):
return db.query(User).filter(User.id == user_id).first()
def get_user_by_email(email, db:Session):
return db.query(User).filter(User.email == email).first()
def get_users(db:Session, skip=0, limit=100):
return db.query(User).offset(skip).limit(limit).all()
def update_user(db:Session, user_id:int, user:UserUpdate):
user = db.query(User).filter(User.id == user_id).first()
if user:
update_data = user.dict(exclude_unset=True)
if 'password' in update_data:
update_data['password'] = get_password_hash(update_data['password'])
del update_data['password']
for key, value in update_data.items():
setattr(user, key, value)
db.commit()
db.refresh(user)
return user
def delete_user(user_id, db:Session):
user = db.query(User).filter(User.id == user_id).first()
if user:
db.delete(user)
db.commit()
return user
return None
async def update_user_embeddings(user_id:int, embeddings:UserEmbeddingsModel, db:Session):
embeddings = db.query(UserEmbeddings).filter(UserEmbeddings.user_id == user_id).first()
if embeddings:
embeddings.embeddings = embeddings.embeddings
db.commit()
db.refresh(embeddings)
return embeddings
return None
|