'dev'

app/crud.py

@@ -0,0 +1,37 @@
+from sqlalchemy.orm import Session
+from app import models, schemas
+from sqlalchemy.exc import IntegrityError
+from fastapi import HTTPException
+
+def get_user(db: Session, user_id: int):
+    return db.query(models.User).filter(models.User.id == user_id).first()
+
+def get_user_by_openid_or_wxid(db: Session, identifier: str):
+    return db.query(models.User).filter((models.User.openid == identifier) | (models.User.wxid == identifier)).first()
+
+def get_users(db: Session, skip: int = 0, limit: int = 10):
+    return db.query(models.User).offset(skip).limit(limit).all()
+
+def create_user(db: Session, user: schemas.UserCreate):
+    db_user = models.User(**user.dict())
+    try:
+        db.add(db_user)
+        db.commit()
+        db.refresh(db_user)
+    except IntegrityError:
+        db.rollback()
+        # wxid的唯一性
+        raise HTTPException(status_code=400, detail="wxid must be unique")
+    return db_user
+
+def delete_user(db: Session, user: models.User):
+    db.delete(user)
+    db.commit()
+    return user
+
+def update_user(db: Session, user: models.User, user_update: schemas.UserUpdate):
+    for key, value in user_update.dict(exclude_unset=True).items():
+        setattr(user, key, value)
+    db.commit()
+    db.refresh(user)
+    return user