diff --git a/core/ab_testing.py b/core/ab_testing.py index d7d19d9..00b60ee 100644 --- a/core/ab_testing.py +++ b/core/ab_testing.py @@ -15,6 +15,13 @@ logger = logging.getLogger(__name__) +# Load environment variables for database credentials +DB_HOST = os.getenv('DB_HOST', 'localhost') +DB_PORT = os.getenv('DB_PORT', '5432') +DB_NAME = os.getenv('DB_NAME', 'milo_db') +DB_USER = os.getenv('DB_USER', 'milo_user') +DB_PASSWORD = os.getenv('DB_PASSWORD', 'milo_password') + SIGNIFICANCE_THRESHOLD = 0.15 # 15% difference to declare winner MAX_EXPERIMENT_DAYS = 14 # Cancel experiments older than this diff --git a/core/business_manager.py b/core/business_manager.py index 62a5578..7e368d5 100644 --- a/core/business_manager.py +++ b/core/business_manager.py @@ -9,6 +9,13 @@ import yaml +# Load environment variables for database credentials +DB_HOST = os.getenv('DB_HOST', 'localhost') +DB_PORT = os.getenv('DB_PORT', '5432') +DB_NAME = os.getenv('DB_NAME', 'milo_db') +DB_USER = os.getenv('DB_USER', 'milo_user') +DB_PASSWORD = os.getenv('DB_PASSWORD', 'milo_password') + logger = logging.getLogger(__name__) diff --git a/core/community_manager.py b/core/community_manager.py index b97850b..1efa98b 100644 --- a/core/community_manager.py +++ b/core/community_manager.py @@ -32,6 +32,13 @@ logger = logging.getLogger(__name__) +# Load environment variables for database credentials +DB_HOST = os.getenv('DB_HOST', 'localhost') +DB_PORT = os.getenv('DB_PORT', '5432') +DB_NAME = os.getenv('DB_NAME', 'milo_db') +DB_USER = os.getenv('DB_USER', 'milo_user') +DB_PASSWORD = os.getenv('DB_PASSWORD', 'milo_password') + # Default flair templates per subreddit DEFAULT_FLAIRS = [ {"text": "Discussion", "css_class": "discussion"},