add equipmentLast functionality
This commit is contained in:
parent
b237b22693
commit
2b4ade6b6e
3 changed files with 52 additions and 9 deletions
|
|
@ -1,6 +1,6 @@
|
|||
from fastapi import APIRouter, Request, Depends, HTTPException
|
||||
from sqlalchemy.orm import Session as DBSession
|
||||
from database import SessionLocal, User, Session as UserSession, Equipment
|
||||
from database import SessionLocal, User, Session as UserSession, Equipment, EquipmentLast
|
||||
from services import auth
|
||||
from config import cipher
|
||||
import json
|
||||
|
|
@ -14,8 +14,8 @@ def get_db():
|
|||
finally:
|
||||
db.close()
|
||||
|
||||
@router.get("/me/equipment/history")
|
||||
async def get_history(request: Request, db: DBSession = Depends(get_db)):
|
||||
@router.get("/me/equipment")
|
||||
async def get_equipment_composite(request: Request, db: DBSession = Depends(get_db)):
|
||||
session_id = request.cookies.get("session_id")
|
||||
db_session = db.query(UserSession).filter(UserSession.id == session_id).first()
|
||||
|
||||
|
|
@ -29,10 +29,13 @@ async def get_history(request: Request, db: DBSession = Depends(get_db)):
|
|||
token_data = auth.get_token(user.username, decrypted_pass)
|
||||
profile = json.loads(auth.get_profile(token_data["token"]))
|
||||
pid_val = int(profile.get("pid"))
|
||||
print(f"Fetching History for PID: {pid_val}")
|
||||
|
||||
results = db.query(Equipment).filter(Equipment.PId == pid_val).all()
|
||||
return results
|
||||
mii_name = profile.get("name", user.username)
|
||||
last_gear = db.query(EquipmentLast).filter(EquipmentLast.PId == pid_val).first()
|
||||
|
||||
return {
|
||||
"mii_name": mii_name,
|
||||
"last_equipped": last_gear
|
||||
}
|
||||
except Exception as e:
|
||||
print(f"Error in history route: {e}")
|
||||
return []
|
||||
print(f"Equipment Route Error: {e}")
|
||||
return {"mii_name": user.username, "last_equipped": None}
|
||||
38
routes/equipment/equipment_history.py
Normal file
38
routes/equipment/equipment_history.py
Normal file
|
|
@ -0,0 +1,38 @@
|
|||
from fastapi import APIRouter, Request, Depends, HTTPException
|
||||
from sqlalchemy.orm import Session as DBSession
|
||||
from database import SessionLocal, User, Session as UserSession, Equipment
|
||||
from services import auth
|
||||
from config import cipher
|
||||
import json
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
def get_db():
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
@router.get("/me/equipment/history")
|
||||
async def get_history(request: Request, db: DBSession = Depends(get_db)):
|
||||
session_id = request.cookies.get("session_id")
|
||||
db_session = db.query(UserSession).filter(UserSession.id == session_id).first()
|
||||
|
||||
if not db_session:
|
||||
raise HTTPException(status_code=401)
|
||||
|
||||
user = db.query(User).filter(User.username == db_session.username).first()
|
||||
|
||||
try:
|
||||
decrypted_pass = cipher.decrypt(user.spfn_pass_enc.encode()).decode()
|
||||
token_data = auth.get_token(user.username, decrypted_pass)
|
||||
profile = json.loads(auth.get_profile(token_data["token"]))
|
||||
pid_val = int(profile.get("pid"))
|
||||
print(f"Fetching History for PID: {pid_val}")
|
||||
|
||||
results = db.query(Equipment).filter(Equipment.PId == pid_val).all()
|
||||
return results
|
||||
except Exception as e:
|
||||
print(f"Error in history route: {e}")
|
||||
return []
|
||||
Loading…
Reference in a new issue