From 745a3df44e9c7266e83b497e3133ef10e7347aa4 Mon Sep 17 00:00:00 2001
From: hackrobot <xiaoshuang.eric@gmail.com>
Date: Mon, 15 Jul 2024 18:43:15 +0800
Subject: [PATCH] 'dev'

---
 app/__pycache__/crud.cpython-310.pyc     | Bin 0 -> 1800 bytes
 app/__pycache__/database.cpython-310.pyc | Bin 0 -> 504 bytes
 app/__pycache__/main.cpython-310.pyc     | Bin 0 -> 2345 bytes
 app/__pycache__/models.cpython-310.pyc   | Bin 0 -> 681 bytes
 app/__pycache__/schemas.cpython-310.pyc  | Bin 0 -> 1215 bytes
 app/crud.py                              |  37 ++++++++++++++
 app/database.py                          |  11 ++++
 app/main.py                              |  61 +++++++++++++++++++++++
 app/models.py                            |  14 ++++++
 app/schemas.py                           |  23 +++++++++
 requirements.txt                         |   4 ++
 run.py                                   |   4 ++
 12 files changed, 154 insertions(+)
 create mode 100644 app/__pycache__/crud.cpython-310.pyc
 create mode 100644 app/__pycache__/database.cpython-310.pyc
 create mode 100644 app/__pycache__/main.cpython-310.pyc
 create mode 100644 app/__pycache__/models.cpython-310.pyc
 create mode 100644 app/__pycache__/schemas.cpython-310.pyc
 create mode 100644 app/crud.py
 create mode 100644 app/database.py
 create mode 100644 app/main.py
 create mode 100644 app/models.py
 create mode 100644 app/schemas.py
 create mode 100644 requirements.txt
 create mode 100644 run.py

diff --git a/app/__pycache__/crud.cpython-310.pyc b/app/__pycache__/crud.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..c7bc77ea863a3d5caf14352a16d8c3e283f2de6f
GIT binary patch
literal 1800
zcmZux&2Ah;5bo~z+ke~ZO=6PBI0%6zf}A*^2qBzEaIui=2x+xwH9Os#^zO{gxVx8l
zqxA)H4dTE9*b)-T3-B;~<&+oTfIy<EXV<YEMw;&G>gt*AtNOZ%cDq5~`Teh7$DM$X
zKXGySiD2<1y!sCqIN>xSIqg%5Hp^J<^}XEh`)=)JK_2!)XZu-{*ZMWE13`Hx>SLcr
zy!O)XH=L&~n&4^hCU{!>w)3?`op&HI<QshRWzcUs@0Rc<9f<AnZHV1i`Mk*{_<H;X
z__p5q_)UoJ!WVAw9oX6C?;Vo(&Ktl^Vrrv9p|vb#F|%P_aFJ;n>GYY%6AinaZ)Qe3
zRnjaTD^)1)w7)(&`u6cjDoO*VEaM{^&9zWN$y0{kTs>&}@b1H_cVSqPGh*n3ozpW9
z=ZU|hQ(Oh-<dD3k2XSbF<GD}^<w3ahpTJjbIF^}#nMrPgu~gc`Okn|34Tjf5ZNT*R
zXZxQIFkBCWlIh@)&{I>CgM2X@C)y;X9G0q>h|~;{vK*vp&L5NuTYD<Z5V42w4UK?z
zsoH|cHY67_BgayxcSwOEDpL2s_<vH4UV!wzB0w-if~pSTaimaMj-YCy)q=r>MJZ-L
z(0_g+x$0oxiewjKkYY7#V)X4b?BQrJtWFLKHN>;{4@cxaRX4zXgMYKWp32e&narh0
zF}LdnZG89P)pudIOlL^}y1%@rP(>iTMgXvn$66S*1IMkGWLfN0^|=KO=e>@<zQ(VS
zyn}m}sylez$7)>3IKsn9JBWQS44F{wOR}VA?40~Yf2ODG#TEef5Gq9q{yA}TP<g&1
zOLhoaG~wbC3OjBu!WdKIlr!$VVn5R7Bn}SbFO-lM+nCC|e6G#jNbJpL5>yG-qX|kf
zb3IJqme`OBlgP|*i|cO-YJDzK1Bu|=hG~&Q%59{?SPA`1p>%9b6<Ib)(y5c(I$N0i
ze#A#kkJmHSOqEECaC;~Q-GiY=8Ew!SZBajH{R1;&|EdpR<4XQe)fXkc2@5PSmeyr~
zx=hivI-vzX0}ycE?m^^Qg9z;%L{*$agq72Hg%FKT)kk33CRTDdFFD}8qB+dL#T9x8
zizS;t3+3z;Jp(w8$oGu_JM~U|!<KaEE&ZQ7N{(w-U<Ve;yMhm}WAESydy+UwvpI(Z
z%|MEApx#Fafixn=ZU&VyMmbLE1K5E8*FhYyk1^DniiHh+NHS<Y*KS+0;^3I>L9E6r
zwL?3=wSjCuh2;UH3WP%xCOGcdjy}$k44dfUL7{SmMF2^ZWu+5ri~Y+n0Kd{kv}#vQ
z8rG9DQ}rWH1Ae^F)ir5guiJN;a22mwiOaj&dg9XVpk1Qlh`A0@dkCF=o{29Iv=dW{
TY=V$lw9C5Qj@R>gt=-^1UTln8

literal 0
HcmV?d00001

diff --git a/app/__pycache__/database.cpython-310.pyc b/app/__pycache__/database.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..01fcc12267c1e6aab4afa4bc7af7359f23c02405
GIT binary patch
literal 504
zcmYjOJ&zMH5VdzVn`E_#4qbu7KS+~p(Fq}PABZ9#AY6g9lI7Twi^+PuXFKRZizu(-
zPw<!AQc=@UF?(0sjpWgr(Y)t*ww+80f>Hhbx%oi}`P~<fV}QjgJpMf<MHFp_p^VZV
zCrx5fmfqpC$;^Na5N9k03b`7nTvIi?P1q0+s~m=Ebepmfq~8e}tMQ7=ia)4IW;9Nu
z0}+5jd!<{z#*qTq2q%L63Vbbm-zz<MuWf6@4&1k6f8FQ+CskF=s{lUC)tWt#TaY{M
zg#jM6sHtA}L}}X=WZ=SG`B@qZu@6>SV|0k)9&ehx-^OgMTlH>M!~&&7gHq<;5?qiy
zduYtz25)*;&ZRgIr;i4#GcM~{Pgft7i{;syi?^To`Qp>!bg{bNmmilgzqc97)xDvm
zm7<B+Dcbx++!VQY>EH9Sllszw^EEgv>vQnCV7uBJ_(u33I?X$0zo4;NbX~1P5dBNd
eyF(nkFjnmwc!_^Xe6LuNl9n_}j*@9oB>w?h8k4F3

literal 0
HcmV?d00001

diff --git a/app/__pycache__/main.cpython-310.pyc b/app/__pycache__/main.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..1447098b5b89cedea8b6f18a0cd812634fe328b2
GIT binary patch
literal 2345
zcma)6-EZ7P5Z_(<Y~M%jE}w1Ev`xP;pavl%Bm@X(K_FG3Qd^L$NS5Q>OMUUjvFlTt
zaFKxYHE(_70Uq-w@UQGEPyGuhg+gcceCd@Ifh~`BXJ=>UH}l&mW@Z`$o)3O|*!#sJ
z<X4<bKXWko2tM^AC4>`BQ<BkuW-MTt6<8E?EVZ**P%~pIbuu?_GcWMW+)nFRBWPsJ
zpqaITmeJMHnXDbOvrf>VL^S%{{%n6vbhvX!VNMT7Q0Ff99*|&O)cXs(E}H#C-Z*5u
zIsFPt%pcI6$6LY^OUs1MRR4DM?^OS8^`EW&bJc%d)OuaMz!x9X4rp+~oV6reJPo|z
zd4!*+PO<n&agr~?hr1lU!cUo9%VyW=B*My?^3!6;SlavyKWi*k!15Gdd&%-#<<sWp
z`2}OS3YMq&t9@RC{b#BeF7eC8{w#0aBL0;}SU2C&?q`uIKfd{ep85Rt?VF$8kHw%&
zid?(b#X#g-`Bb~Ngi@gMnYM2vs?^?>#wJQ(zmpYQq)NLgzALgwX*-rf4yLVY*Nq~M
zQtgO*JIMujv$v9*$9VeW0j>~W6G5eV4aE1PBnM=l>@`N@D?-WFWJG?VKf{QVTgT&j
z#HSlRlTFy6YjIjAbkF$K5s_WcJ$dKadz*KZkZM!NB;LF()J|CpHnZKZ2biNl5)Ne1
z7jd~64L}_w`PIR$cD6+s@~t)ogKTHtIh0Y4E-{Jp-m}*u&V3rRrBH(+S0XeS*w9{{
z#5;ME3GGP8gxt|H_e6X*D#Lu3Z3(IEd-oHL{&NnqPERH<3c*`gCb+ABgeP2f;FK8i
zF?n!|Ag{rv&VwjPpN`y;^{o*Z(GeS2BYRXEIh-E4oE<W#ueIU3avp5u0*XZxODImD
zIEi8z#1S*cX?qBH@@<KLgJvv6REiLFvW}VtrtzB5`riL=Yn+-QjKwTC(H<g2pBRiu
zZKLT>Om5?;Tm{{pyN=6h{Sk(tZMBmObS+J?q>L@lOdcH9;4}HaN}GHz%mrk?0`zRd
zx5hD?fn|9X1Q5l5t7Cc1F<^4M`pSSC0C*g+x&$5+1JPx24SsubCe-zB6E5;H=_Nw$
z>&Ej*jJA^p<F(;4@m_|3iI<L`KOVAuHiFJEByzT7oN<fWhwKNaDWIrr98q}=>>oXZ
zEApg`MOe#=a;;YkbAH5h6IfghRT#q+3>+@XD1j?!aO#x_&Twlt95X8{WQY`$7!cT@
z>oM?CIKoU$m8*V?-iHN+By2N}wx~@dUedizl>$s=nDBk=R-=Ch{}$}~Z}4DYA+34M
zP}#vtj+a7WAo2o=izu)tas|YMxZVV_L8zP1m_)^6fNOReKd*o*Uk5)J-W1R_HCo*?
zxP}7`m*NE?Vo{HY_%=)?L^Or9M%K%PHQ4aNQb31Pbv7hk>3_h2aGyu|2CP>Q<rTEW
z`ax7CAj4%?HN`~Bsf{mIRK5wT@4`0)5$KFse=TsUo+y#I<^+lGg0}i5jZ*CGyH^XD
zNyMw2a(9sA+tNW>7c~~NP>CEd4P$)OTA;fMNsG23pa^y_*#Z&RpTJyu8SoWXgGSYF
zqBJ!vR$EZVz(%@jOO?_z2W=&JY3`n3p9Z>PamN+12E%gPr0$rc7*JD$Hi;NFq4%L%
z4^#0W+$f@MgIFLf+NBGuOS_iCJge<8#-J%pUWeL`p-KJiF!q;Yv8RsB{&Z^W5659o
X9GCshyw>Bn&PCd`R%zS8?=pP`hFbwC

literal 0
HcmV?d00001

diff --git a/app/__pycache__/models.cpython-310.pyc b/app/__pycache__/models.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..ab08cdf68a0189707aaecd00c002126a31218622
GIT binary patch
literal 681
zcmYjPKW`H;6t{hUa->Ph5W#1-r4wR62vrmbv2-acRwT>4_mY~N?=JS8+RSC`#+Tre
z@EyD&Au+Qt@tmC0+Vb!H^ZRSJ^ZAs~`u+F&`X^`XR}`MEl!S9S{uKc*;5D;cax0{;
ziA;FZi8{5J%sBhVAc6E7gA}v-MCOoPu`>Tdo~%qle$m#0ZNlVJ)8htR$gg^5noY?=
z`d)ji;=QLUBjKEm2Lfc01Cs){OpxM(pq)HW7Lq2!rw>9Fk-z8i*+{3OO+ZFBPe$!*
zr1Q}lJr|Hut&@;`@#thW5c<!h2q&F0R=d60;yz@i0lX`@E2yAS97xEA#_R^XEKftB
zRHH2_6{bqr7KS?VGo^Neu1CvUsa{{#codzWNG>4=!hnm@4n<?CZ8V46A?4nN`7KsA
zx>wC$uW4@S?VW*;Z;+~S;dJc`aY_TI-bAmq!^Xf5=3d0$5&pftc(aO~`4u`-tv;aN
z_HDPad$rcS*PT(FYrkUEuXIQ17O?hb-QMNo9E<Wxf`X-dA!fXwy*S`=ez<W-gPZzY
xt?TLr?f$W;V?#5Fc?O!=rdM|HRS#Z|<sG?~WC+FE<J{iSXJkI6U$PlrivOFltULe!

literal 0
HcmV?d00001

diff --git a/app/__pycache__/schemas.cpython-310.pyc b/app/__pycache__/schemas.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..42865096a3d2f2f667359d74bd8ccbe7693706d6
GIT binary patch
literal 1215
zcmZ{j&2AGh5XbHH?#9_95K8$HS2*mY*Gi~D6-DAiI6y9|l`OM%+NIeK?5#>Ta4C<#
zOZ1g|<-|ERE->RXp=!fM{>wj)J)Rj)n$ajB@I7DtnEj!I{6x6^g#d>zY!0G`qH|Kw
zj8dEj^PplGqvQ=yOoiWx3N^i8SqL1d7&z7;a0J{_9GH8Kf%_@}PCWP2;5|u)zu*Rv
zQrCYat$y1mJqO_L+OBMBIWN$9{bK-!Fzh!FO){!T7EB?+jOkEEPl%3HaDd#~F!Kyi
zT-ig<{S8N+A*w5T?0K-^p6B6)IWolFcYJEK!N4bu*JW{9%SyZ0=t`PXH#*nFiR?su
zR(;gQh38*N<>F&qE1<hGQ^HJ0r9@ZasixI+snU_-LZBZZTp~o(sIxiFheCWllk=6u
zM3B9~S~zAq@Fs+;>!y=fH7kUP(cVMhD18)sax(zAB<3E@QGV|qy~r^On`=`R`D<-Y
zyQa;ng_ud($+i@&X+G(q%Vpc<wm5<0>|VPt*dsXX7$hNM`tP?(z4g}PL96-4TMl8e
zemq~wz$F;Ok<qf#lS^iBhZ&*VLD|H+n(g4)E(q>eVicT&)UW*aQga`+rB@eCTZI>y
z2e7u)`6Do3V&)+V(k;hW{zDk{8OS=DDbdh~iXPzKBTx+5om~W3q(W$z$c31t3rcm%
z+qyQsoNI5`)+&IL?CSte?j1GttUUfONy8fza=bAWc8lJYj*C|ZF&Vom7pLYa&?Jp-
zdSYeW8J|0D*yheW1_PGUVxJ|1)1APtO;YZ7yHK+3%EAO-baA(6%lg<itM3Z`O1`50
b$Tt6uAFTTRC04~^FAT|$j)QT3%y!vdOrh!`

literal 0
HcmV?d00001

diff --git a/app/crud.py b/app/crud.py
new file mode 100644
index 0000000..f5ea976
--- /dev/null
+++ b/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
diff --git a/app/database.py b/app/database.py
new file mode 100644
index 0000000..331a877
--- /dev/null
+++ b/app/database.py
@@ -0,0 +1,11 @@
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+
+SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()
diff --git a/app/main.py b/app/main.py
new file mode 100644
index 0000000..02f1197
--- /dev/null
+++ b/app/main.py
@@ -0,0 +1,61 @@
+from fastapi import FastAPI, HTTPException, Depends
+from sqlalchemy.orm import Session
+from typing import List, Optional
+
+from app import models, schemas, crud
+from app.database import SessionLocal, engine
+
+# 初始化数据库
+models.Base.metadata.create_all(bind=engine)
+
+app = FastAPI()
+
+# 依赖注入
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+@app.get("/add", response_model=schemas.User)
+def add_user(
+    nickname: Optional[str] = None,
+    remark: Optional[str] = None,
+    wechat_number: Optional[str] = None,
+    wxid: Optional[str] = None,
+    gender: Optional[str] = None,
+    friend_add_time: Optional[str] = None,
+    openid: Optional[str] = None,
+    db: Session = Depends(get_db)
+):
+    return crud.create_user(db=db, user=schemas.UserCreate(
+        nickname=nickname, remark=remark, wechat_number=wechat_number,
+        wxid=wxid, gender=gender, friend_add_time=friend_add_time,
+        openid=openid))
+
+@app.get("/users/", response_model=List[schemas.User])
+def read_users(skip: int = 0, limit: int = 10, db: Session = Depends(get_db)):
+    return crud.get_users(db=db, skip=skip, limit=limit)
+
+@app.get("/user/{identifier}", response_model=schemas.User)
+def read_user(identifier: str, db: Session = Depends(get_db)):
+    user = crud.get_user_by_openid_or_wxid(db, identifier=identifier)
+    if user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    return user
+
+@app.delete("/user/{user_id}", response_model=schemas.User)
+def delete_user(user_id: int, db: Session = Depends(get_db)):
+    user = crud.get_user(db=db, user_id=user_id)
+    if user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    return crud.delete_user(db=db, user=user)
+
+@app.put("/user/{identifier}", response_model=schemas.User)
+def update_user(identifier: str, user_update: schemas.UserUpdate, db: Session = Depends(get_db)):
+    user = crud.get_user_by_openid_or_wxid(db, identifier=identifier)
+    if user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    
+    return crud.update_user(db=db, user=user, user_update=user_update)
\ No newline at end of file
diff --git a/app/models.py b/app/models.py
new file mode 100644
index 0000000..afd64de
--- /dev/null
+++ b/app/models.py
@@ -0,0 +1,14 @@
+from sqlalchemy import Column, Integer, String
+from app.database import Base
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    nickname = Column(String, index=True)
+    remark = Column(String)
+    wechat_number = Column(String)
+    wxid = Column(String, index=True, unique=True)
+    gender = Column(String)
+    friend_add_time = Column(String)
+    openid = Column(String, index=True)
diff --git a/app/schemas.py b/app/schemas.py
new file mode 100644
index 0000000..9a3a9d3
--- /dev/null
+++ b/app/schemas.py
@@ -0,0 +1,23 @@
+from pydantic import BaseModel
+from typing import Optional
+
+class UserBase(BaseModel):
+    nickname: Optional[str] = None
+    remark: Optional[str] = None
+    wechat_number: Optional[str] = None
+    wxid: Optional[str] = None
+    gender: Optional[str] = None
+    friend_add_time: Optional[str] = None
+    openid: Optional[str] = None
+
+class UserCreate(UserBase):
+    pass
+
+class UserUpdate(UserBase):
+    pass
+
+class User(UserBase):
+    id: int
+
+    class Config:
+        orm_mode = True
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..a70e8ac
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+sqlalchemy
+pydantic
\ No newline at end of file
diff --git a/run.py b/run.py
new file mode 100644
index 0000000..4753965
--- /dev/null
+++ b/run.py
@@ -0,0 +1,4 @@
+import uvicorn
+
+if __name__ == "__main__":
+    uvicorn.run("app.main:app", host="127.0.0.1", port=8000, reload=True)