· 4 years ago · Apr 11, 2021, 09:36 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 pass
23 url = os.environ.get('DATABASE_URL')
24 if url.startswith("postgres://"):
25 url = url.replace("postgres://", "postgresql://", 1)
26 SQLALCHEMY_DATABASE_URI = url
27