forked from spacebar/SplatNet-Backend
add equipmentLast functionality
This commit is contained in:
parent
b237b22693
commit
2b4ade6b6e
3 changed files with 52 additions and 9 deletions
2
main.py
2
main.py
|
|
@ -8,6 +8,7 @@ from routes import logout
|
||||||
from routes import boss
|
from routes import boss
|
||||||
#yes meow meow import me please... i want to be embedded into my code... :pleading_face:
|
#yes meow meow import me please... i want to be embedded into my code... :pleading_face:
|
||||||
from routes import me
|
from routes import me
|
||||||
|
from routes.equipment import equipment_history
|
||||||
from routes.equipment import equipment
|
from routes.equipment import equipment
|
||||||
from services.boss_retrieval import process_boss_file
|
from services.boss_retrieval import process_boss_file
|
||||||
from contextlib import asynccontextmanager
|
from contextlib import asynccontextmanager
|
||||||
|
|
@ -116,6 +117,7 @@ app.include_router(sessionid_check.router, prefix="/api/v1")
|
||||||
app.include_router(logout.router, prefix="/api/v1")
|
app.include_router(logout.router, prefix="/api/v1")
|
||||||
app.include_router(boss.router, prefix="/api/v1")
|
app.include_router(boss.router, prefix="/api/v1")
|
||||||
app.include_router(me.router, prefix="/api/v1")
|
app.include_router(me.router, prefix="/api/v1")
|
||||||
|
app.include_router(equipment_history.router, prefix="/api/v1")
|
||||||
app.include_router(equipment.router, prefix="/api/v1")
|
app.include_router(equipment.router, prefix="/api/v1")
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
from fastapi import APIRouter, Request, Depends, HTTPException
|
from fastapi import APIRouter, Request, Depends, HTTPException
|
||||||
from sqlalchemy.orm import Session as DBSession
|
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 services import auth
|
||||||
from config import cipher
|
from config import cipher
|
||||||
import json
|
import json
|
||||||
|
|
@ -14,8 +14,8 @@ def get_db():
|
||||||
finally:
|
finally:
|
||||||
db.close()
|
db.close()
|
||||||
|
|
||||||
@router.get("/me/equipment/history")
|
@router.get("/me/equipment")
|
||||||
async def get_history(request: Request, db: DBSession = Depends(get_db)):
|
async def get_equipment_composite(request: Request, db: DBSession = Depends(get_db)):
|
||||||
session_id = request.cookies.get("session_id")
|
session_id = request.cookies.get("session_id")
|
||||||
db_session = db.query(UserSession).filter(UserSession.id == session_id).first()
|
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)
|
token_data = auth.get_token(user.username, decrypted_pass)
|
||||||
profile = json.loads(auth.get_profile(token_data["token"]))
|
profile = json.loads(auth.get_profile(token_data["token"]))
|
||||||
pid_val = int(profile.get("pid"))
|
pid_val = int(profile.get("pid"))
|
||||||
print(f"Fetching History for PID: {pid_val}")
|
mii_name = profile.get("name", user.username)
|
||||||
|
last_gear = db.query(EquipmentLast).filter(EquipmentLast.PId == pid_val).first()
|
||||||
results = db.query(Equipment).filter(Equipment.PId == pid_val).all()
|
|
||||||
return results
|
return {
|
||||||
|
"mii_name": mii_name,
|
||||||
|
"last_equipped": last_gear
|
||||||
|
}
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"Error in history route: {e}")
|
print(f"Equipment Route Error: {e}")
|
||||||
return []
|
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