57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from sqlalchemy.orm import Session
|
|
from sqlalchemy import select
|
|
|
|
|
|
from . import models, schemas
|
|
|
|
|
|
# 删除
|
|
def delete_user(db: Session, user_id: int):
|
|
db.query(models.User).where(models.User.id == user_id).delete()
|
|
db.commit()
|
|
# db.refresh(db_user)
|
|
|
|
|
|
# 更新
|
|
def update_user(db: Session, user_id: int):
|
|
db_test=db.query(models.User).filter(models.User.id == user_id).update({"email":'111'},synchronize_session=False)
|
|
db.commit()
|
|
# db_test=db.query(models.User).filter(models.User.id == user_id).first()
|
|
# db_test.email='12345@qq.com'
|
|
# db.commit()
|
|
|
|
|
|
# 查询
|
|
def get_user(db: Session, user_id: int):
|
|
return db.query(models.User).filter(models.User.id == user_id).first()
|
|
|
|
|
|
def get_user_by_email(db: Session, email: str):
|
|
return db.query(models.User).filter(models.User.email == email).first()
|
|
|
|
|
|
def get_users(db: Session, skip: int = 0, limit: int = 100):
|
|
return db.query(models.User).offset(skip).limit(limit).all()
|
|
|
|
# 新增
|
|
def create_user(db: Session, user: schemas.UserCreate):
|
|
|
|
fake_hashed_password = user.password + "notreallyhashed"
|
|
db_user = models.User(email=user.email, hashed_password=fake_hashed_password,is_active=False)
|
|
db.add(db_user)
|
|
db.commit()
|
|
db.refresh(db_user)
|
|
return db_user
|
|
|
|
|
|
def get_items(db: Session, skip: int = 0, limit: int = 100):
|
|
return db.query(models.Item).offset(skip).limit(limit).all()
|
|
|
|
|
|
def create_user_item(db: Session, item: schemas.ItemCreate, user_id: int):
|
|
db_item = models.Item(**item.dict(), owner_id=user_id)
|
|
db.add(db_item)
|
|
db.commit()
|
|
db.refresh(db_item)
|
|
return db_item
|