SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 23:22:57 +01:00
Code Issues Projects Releases Packages Activity

Updated with active subscription

Browse source
This commit is contained in:
Saurab-Shrestha 2024-02-25 12:47:35 +05:45
parent 808a0fa6fc
commit 8271fedb78
15 changed files with 121 additions and 240 deletions
Download patch file Download diff file Expand all files Collapse all files

7
private_gpt/users/api/deps.py
View file

@ -117,17 +117,16 @@ def get_company_name(company_id: int, db: Session = Depends(get_db)) -> str:
def get_active_subscription(
current_user: models.User = Depends(get_current_user),
db: Session = Depends(get_db),
):
company_id = current_user.user_role.company_id
company_id = 1
if company_id:
company = crud.company.get(db, company_id)
if company and company.subscriptions:
active_subscription = next((sub for sub in company.subscriptions if sub.is_active), None)
if active_subscription:
return company
print("Has active Subscription")
return active_subscription
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail="Access Forbidden - No Active Subscription",