From 538f2cef71762bfe7c972ca989081a9b880b446d Mon Sep 17 00:00:00 2001 From: dvirlabs Date: Sun, 21 Dec 2025 04:41:43 +0200 Subject: [PATCH] Fix prod banner --- backend/backup_restore_api.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/backend/backup_restore_api.py b/backend/backup_restore_api.py index f22bcea..12e14ff 100644 --- a/backend/backup_restore_api.py +++ b/backend/backup_restore_api.py @@ -16,10 +16,23 @@ load_dotenv() def get_environment() -> str: - """Detect environment based on FRONTEND_URL""" - frontend_url = os.getenv('FRONTEND_URL', 'http://localhost:5174') + """Detect environment based on FRONTEND_URL or deployment environment""" + # Check FRONTEND_URL first + frontend_url = os.getenv('FRONTEND_URL', '') if 'myrecipes.dvirlabs.com' in frontend_url or 'my-recipes.dvirlabs.com' in frontend_url: return 'prod' + + # Check if explicitly set via ENV variable + env_var = os.getenv('ENVIRONMENT', '').lower() + if env_var in ['prod', 'production']: + return 'prod' + + # Check database host (if not localhost, likely production) + db_host = os.getenv('DB_HOST', 'localhost') + if db_host not in ['localhost', '127.0.0.1', 'postgres']: + return 'prod' + + # Default to dev return 'dev'