11import os
22
3- class BaseConfig :
4- ENVIRONMENT = os .getenv ('ENVIRONMENT' , 'development' ).lower ()
5- LOG_LEVEL = os .getenv ('LOG_LEVEL' , 'INFO' if ENVIRONMENT == 'production' else 'DEBUG' ).upper ()
6-
7- class MainConfig (BaseConfig ):
3+ class Config ():
84 # Flask
95 FLASK_HOST = os .getenv ('FLASK_HOST' )
106 SECRET_KEY = os .getenv ('SECRET_KEY' )
@@ -18,34 +14,23 @@ class MainConfig(BaseConfig):
1814
1915 # Postmark
2016 POSTMARK_API_KEY = os .getenv ('POSTMARK_API_KEY' )
17+
18+ # Environment
19+ ENVIRONMENT = os .getenv ('ENVIRONMENT' , 'development' ).lower ()
20+ LOG_LEVEL = os .getenv ('LOG_LEVEL' , 'INFO' if ENVIRONMENT == 'production' else 'DEBUG' ).upper ()
2121
2222 @staticmethod
2323 def validate ():
2424 required = {
25- 'ENVIRONMENT' : MainConfig .ENVIRONMENT ,
26- 'SECRET_KEY' : MainConfig .SECRET_KEY ,
27- 'FLASK_HOST' : MainConfig .FLASK_HOST ,
28- 'POSTMARK_API_KEY' : MainConfig .POSTMARK_API_KEY ,
29- 'POSTGRES_HOST' : MainConfig .POSTGRES_HOST ,
30- 'POSTGRES_PORT' : MainConfig .POSTGRES_PORT ,
31- 'POSTGRES_DB' : MainConfig .POSTGRES_DB ,
32- 'POSTGRES_USER' : MainConfig .POSTGRES_USER ,
33- 'POSTGRES_PASSWORD' : MainConfig .POSTGRES_PASSWORD ,
34- }
35- for key , val in required .items ():
36- if not val :
37- raise ValueError (f"{ key } is not set" )
38-
39-
40- class RelayConfig (BaseConfig ):
41- HOST = os .getenv ('HOST' )
42- OPENBMP_CONNECT = os .getenv ('OPENBMP_CONNECT' )
43-
44- @staticmethod
45- def validate ():
46- required = {
47- 'HOST' : RelayConfig .HOST ,
48- 'OPENBMP_CONNECT' : RelayConfig .OPENBMP_CONNECT ,
25+ 'ENVIRONMENT' : Config .ENVIRONMENT ,
26+ 'SECRET_KEY' : Config .SECRET_KEY ,
27+ 'FLASK_HOST' : Config .FLASK_HOST ,
28+ 'POSTMARK_API_KEY' : Config .POSTMARK_API_KEY ,
29+ 'POSTGRES_HOST' : Config .POSTGRES_HOST ,
30+ 'POSTGRES_PORT' : Config .POSTGRES_PORT ,
31+ 'POSTGRES_DB' : Config .POSTGRES_DB ,
32+ 'POSTGRES_USER' : Config .POSTGRES_USER ,
33+ 'POSTGRES_PASSWORD' : Config .POSTGRES_PASSWORD ,
4934 }
5035 for key , val in required .items ():
5136 if not val :
0 commit comments