Upload main.py

main.py

@@ -0,0 +1,37 @@
+
+# main.py
+
+from fastapi import FastAPI, Depends, HTTPException
+from fastapi.security import OAuth2PasswordBearer
+from jose import JWTError, jwt
+from sqlalchemy.orm import Session
+
+from auth import SECRET_KEY, ALGORITHM
+from database import get_db  # فرض بر این که چنین فایلی داری
+from models import User       # مدل کاربر
+# از فایل‌های خودت import را تنظیم کن
+
+app = FastAPI()
+
+# احراز هویت با OAuth2
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="login")
+
+# بررسی توکن و دریافت کاربر جاری
+async def get_current_user(token: str = Depends(oauth2_scheme), db: Session = Depends(get_db)):
+    try:
+        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
+        user = db.query(User).filter(User.username == payload["sub"]).first()
+        if not user:
+            raise HTTPException(status_code=404, detail="User not found")
+        return user
+    except JWTError:
+        raise HTTPException(status_code=401, detail="Invalid token")
+
+# مسیر حفاظت‌شده
+@app.get("/users/me")
+async def read_users_me(current_user: User = Depends(get_current_user)):
+    return {"username": current_user.username}
+
+# اگر فایل دیگری مثل auth_routes نداری، این خط را حذف کن:
+# from routers import auth_routes
+# app.include_router(auth_routes.router)