Fix prod banner

This commit is contained in:
dvirlabs 2025-12-21 04:41:43 +02:00
parent d36f4bcc5c
commit 538f2cef71

View File

@ -16,10 +16,23 @@ load_dotenv()
def get_environment() -> str: def get_environment() -> str:
"""Detect environment based on FRONTEND_URL""" """Detect environment based on FRONTEND_URL or deployment environment"""
frontend_url = os.getenv('FRONTEND_URL', 'http://localhost:5174') # 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: if 'myrecipes.dvirlabs.com' in frontend_url or 'my-recipes.dvirlabs.com' in frontend_url:
return 'prod' 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' return 'dev'