· 4 years ago · Apr 11, 2021, 09:44 PM
1# src/config.py
2
3
4import os
5
6class BaseConfig:
7 TESTING = False
8 SQLALCHEMY_TRACK_MODIFICATIONS = False
9 SECRET_KEY = 'my_precious'
10
11
12class DevelopmentConfig(BaseConfig):
13 SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL')
14
15
16class TestingConfig(BaseConfig):
17 TESTING = True
18 SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_TEST_URL')
19
20
21class ProductionConfig(BaseConfig):
22 url = os.environ.get('DATABASE_URL')
23 if url.startswith("postgres://"):
24 url = url.replace("postgres://", "postgresql://", 1)
25 SQLALCHEMY_DATABASE_URI = url
26