Fix admin UUID parsing error on test upload
This commit is contained in:
@@ -91,7 +91,7 @@ def create_access_token(data: dict):
|
||||
|
||||
def get_current_user(token: str = Depends(oauth2_scheme), session: Session = Depends(get_session)):
|
||||
if token == "admin":
|
||||
return models.User(id=uuid.uuid4(), name="Admin", email="admin@pragativeer.com", tier=models.TierEnum.AI_COACH)
|
||||
return models.User(id=uuid.UUID("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"), name="Admin", email="admin@pragativeer.com", tier=models.TierEnum.AI_COACH)
|
||||
|
||||
try:
|
||||
payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
|
||||
@@ -133,7 +133,7 @@ def require_tier(allowed_tiers: List[models.TierEnum]):
|
||||
"""
|
||||
def tier_checker(token: str = Depends(oauth2_scheme), session: Session = Depends(get_session)):
|
||||
if token == "admin":
|
||||
return models.User(id=uuid.uuid4(), name="Admin", email="admin@pragativeer.com", tier=models.TierEnum.AI_COACH)
|
||||
return models.User(id=uuid.UUID("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"), name="Admin", email="admin@pragativeer.com", tier=models.TierEnum.AI_COACH)
|
||||
|
||||
try:
|
||||
payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
|
||||
@@ -414,7 +414,7 @@ def login(payload: LoginPayload, session: Session = Depends(get_session)):
|
||||
access_token = create_access_token(data={"sub": token_id})
|
||||
return {"success": True, "role": "user", "token": access_token, "user_id": token_id}
|
||||
if payload.email == "admin@pragativeer.com" and payload.password == "Pv9#kL2$mQ8!vX4@":
|
||||
return {"success": True, "role": "admin", "token": "admin"}
|
||||
return {"success": True, "role": "admin", "token": "admin", "user_id": "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"}
|
||||
|
||||
user = session.exec(select(models.User).where(models.User.email == payload.email)).first()
|
||||
if not user:
|
||||
@@ -1838,3 +1838,4 @@ def verify_payment(payload: VerifyPaymentPayload, session: Session = Depends(get
|
||||
session.commit()
|
||||
|
||||
return {"success": True, "tier": user.tier.value}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user