Restauration : fichiers Python depuis commit 4080a308ea
This commit is contained in:
+141
@@ -0,0 +1,141 @@
|
||||
from fastapi import FastAPI, Depends, HTTPException, status
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from datetime import timedelta
|
||||
from database.database import SessionLocal
|
||||
from database.models import Document, User
|
||||
from api.auth import (
|
||||
UserLogin, UserCreate, Token, UserResponse, get_password_hash, verify_password,
|
||||
create_access_token, decode_token, ACCESS_TOKEN_EXPIRE_MINUTES
|
||||
)
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
@app.get("/")
|
||||
def root():
|
||||
return {"message": "RegWatch API running"}
|
||||
|
||||
@app.post("/test-login")
|
||||
def test_login(credentials: UserLogin):
|
||||
return {"received": credentials.dict()}
|
||||
|
||||
@app.post("/signup", response_model=UserResponse)
|
||||
def signup(user: UserCreate):
|
||||
db = SessionLocal()
|
||||
|
||||
existing_user = db.query(User).filter(User.email == user.email).first()
|
||||
if existing_user:
|
||||
db.close()
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_400_BAD_REQUEST,
|
||||
detail="Email already registered"
|
||||
)
|
||||
|
||||
hashed_password = get_password_hash(user.password)
|
||||
db_user = User(
|
||||
email=user.email,
|
||||
password=hashed_password,
|
||||
full_name=user.full_name
|
||||
)
|
||||
db.add(db_user)
|
||||
db.commit()
|
||||
db.refresh(db_user)
|
||||
db.close()
|
||||
|
||||
return UserResponse(
|
||||
id=db_user.id,
|
||||
email=db_user.email,
|
||||
full_name=db_user.full_name
|
||||
)
|
||||
|
||||
@app.post("/login", response_model=Token)
|
||||
def login(credentials: UserLogin):
|
||||
try:
|
||||
db = SessionLocal()
|
||||
|
||||
user = db.query(User).filter(User.email == credentials.email).first()
|
||||
db.close()
|
||||
|
||||
if not user:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="User not found"
|
||||
)
|
||||
|
||||
if not verify_password(credentials.password, user.password):
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="Invalid password"
|
||||
)
|
||||
|
||||
access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
|
||||
access_token = create_access_token(
|
||||
data={"sub": user.email}, expires_delta=access_token_expires
|
||||
)
|
||||
|
||||
return Token(access_token=access_token, token_type="bearer")
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
print(f"ERROR in login: {str(e)}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
raise HTTPException(
|
||||
status_code=500,
|
||||
detail=f"Server error: {str(e)}"
|
||||
)
|
||||
|
||||
@app.get("/me", response_model=UserResponse)
|
||||
def get_current_user(token: str):
|
||||
email = decode_token(token)
|
||||
if not email:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="Invalid token"
|
||||
)
|
||||
|
||||
db = SessionLocal()
|
||||
user = db.query(User).filter(User.email == email).first()
|
||||
db.close()
|
||||
|
||||
if not user:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_404_NOT_FOUND,
|
||||
detail="User not found"
|
||||
)
|
||||
|
||||
return UserResponse(
|
||||
id=user.id,
|
||||
email=user.email,
|
||||
full_name=user.full_name
|
||||
)
|
||||
|
||||
@app.get("/documents")
|
||||
def get_documents():
|
||||
db = SessionLocal()
|
||||
|
||||
documents = db.query(Document).all()
|
||||
|
||||
result = []
|
||||
|
||||
for doc in documents:
|
||||
result.append({
|
||||
"id": doc.id,
|
||||
"title": doc.title,
|
||||
"ingredient": doc.ingredient,
|
||||
"source": doc.source,
|
||||
"type": doc.document_type,
|
||||
"date": doc.meeting_date,
|
||||
"pdf_url": doc.pdf_url
|
||||
})
|
||||
|
||||
db.close()
|
||||
|
||||
return result
|
||||
Reference in New Issue
Block a user