46 lines
1019 B
Python
46 lines
1019 B
Python
"""
|
|
Pass in a config file based on your environment.
|
|
|
|
Example:
|
|
|
|
import check_my_environment
|
|
|
|
|
|
class Main:
|
|
def __init__(self, configFile):
|
|
pass
|
|
|
|
def process(self):
|
|
print("ok")
|
|
|
|
if __name__ == "__main__":
|
|
m = Main(some_script.CONFIGFILE)
|
|
m.process()
|
|
|
|
"""
|
|
|
|
|
|
import os
|
|
import sys
|
|
ENVIRONMENT = "development"
|
|
CONFIGFILE = None
|
|
|
|
|
|
def get_config_file():
|
|
directory = os.path.dirname(__file__)
|
|
return {
|
|
"development": "{}/../config/development.cfg".format(directory),
|
|
"staging": "{}/../config/staging.cfg".format(directory),
|
|
"production": "{}/../config/production.cfg".format(directory)
|
|
}.get(ENVIRONMENT, None)
|
|
|
|
CONFIGFILE = get_config_file()
|
|
|
|
if CONFIGFILE is None:
|
|
sys.exit("Configuration error! Unknown environment set. \
|
|
Edit config.py and set appropriate environment")
|
|
print("Config file: {}".format(CONFIGFILE))
|
|
if not os.path.exists(CONFIGFILE):
|
|
sys.exit("Configuration error! Config file does not exist")
|
|
print("Config ok ....")
|