Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

if user is trialing, set payment method to trial

Browse Source
This commit is contained in:
Abi Raja 2024-05-30 17:40:26 -04:00
parent d96931eeae
commit d4e3405d0a
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/routes/generate_code.py
View File

@ -118,8 +118,15 @@ async def stream_code(websocket: WebSocket):
# TODO: Rename does_user_have_subscription_credits # TODO: Rename does_user_have_subscription_credits
res = await does_user_have_subscription_credits(auth_token) res = await does_user_have_subscription_credits(auth_token)
if res.status != "not_subscriber": if res.status != "not_subscriber":
if res.status == "subscriber_has_credits": if (
payment_method = PaymentMethod.SUBSCRIPTION res.status == "subscriber_has_credits"
or res.status == "subscriber_is_trialing"
):
payment_method = (
PaymentMethod.SUBSCRIPTION
if res.status == "subscriber_has_credits"
else PaymentMethod.TRIAL
)
openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY") openai_api_key = os.environ.get("PLATFORM_OPENAI_API_KEY")
print("Subscription - using platform API key") print("Subscription - using platform API key")
elif res.status == "subscriber_has_no_credits": elif res.status == "subscriber_has_no_credits":

1
backend/routes/logging_utils.py
View File

@ -13,6 +13,7 @@ class PaymentMethod(Enum):
UNKNOWN = "unknown" UNKNOWN = "unknown"
OPENAI_API_KEY = "openai_api_key" OPENAI_API_KEY = "openai_api_key"
SUBSCRIPTION = "subscription" SUBSCRIPTION = "subscription"
TRIAL = "trial"
async def send_to_saas_backend( async def send_to_saas_backend(