ulysia/szurubooru
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix config.py

Browse source
This commit is contained in:
nothink 2018-07-26 01:45:00 +09:00
parent 0b1029251d
commit 8837fe6b92
1 changed files with 30 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
server/szurubooru/config.py
View file

@ -16,33 +16,33 @@ def merge(left: Dict, right: Dict) -> Dict:
return left
# def docker_config() -> Dict:
# for key in [
# 'POSTGRES_USER',
# 'POSTGRES_PASSWORD',
# 'POSTGRES_HOST',
# 'ESEARCH_HOST'
# ]:
# if not os.getenv(key, False):
# raise errors.ConfigError(f'Environment variable "{key}" not set')
# return {
# 'debug': True,
# 'show_sql': int(os.getenv('LOG_SQL', 0)),
# 'data_url': os.getenv('DATA_URL', '/data/'),
# 'data_dir': '/data/',
# 'database': 'postgres://%(user)s:%(pass)s@%(host)s:%(port)d/%(db)s' % {
# 'user': os.getenv('POSTGRES_USER'),
# 'pass': os.getenv('POSTGRES_PASSWORD'),
# 'host': os.getenv('POSTGRES_HOST'),
# 'port': int(os.getenv('POSTGRES_PORT', 5432)),
# 'db': os.getenv('POSTGRES_DB', os.getenv('POSTGRES_USER'))
# },
# 'elasticsearch': {
# 'host': os.getenv('ESEARCH_HOST'),
# 'port': int(os.getenv('ESEARCH_PORT', 9200)),
# 'index': os.getenv('ESEARCH_INDEX', 'szurubooru')
# }
# }
def docker_config() -> Dict:
for key in [
'POSTGRES_USER',
'POSTGRES_PASSWORD',
'POSTGRES_HOST',
'ESEARCH_HOST'
]:
if not os.getenv(key, False):
raise errors.ConfigError(f'Environment variable "{key}" not set')
return {
'debug': True,
'show_sql': int(os.getenv('LOG_SQL', 0)),
'data_url': os.getenv('DATA_URL', '/data/'),
'data_dir': '/data/',
'database': 'postgres://%(user)s:%(pass)s@%(host)s:%(port)d/%(db)s' % {
'user': os.getenv('POSTGRES_USER'),
'pass': os.getenv('POSTGRES_PASSWORD'),
'host': os.getenv('POSTGRES_HOST'),
'port': int(os.getenv('POSTGRES_PORT', 5432)),
'db': os.getenv('POSTGRES_DB', os.getenv('POSTGRES_USER'))
},
'elasticsearch': {
'host': os.getenv('ESEARCH_HOST'),
'port': int(os.getenv('ESEARCH_PORT', 9200)),
'index': os.getenv('ESEARCH_INDEX', 'szurubooru')
}
}
def read_config() -> Dict:
@ -52,9 +52,9 @@ def read_config() -> Dict:
with open('../config.yaml') as handle:
ret = merge(ret, yaml.load(handle.read()))
if os.getenv('CI') == 'true':
print('CI')
# if os.path.exists('/.dockerenv'):
# ret = merge(ret, docker_config())
pass
elif os.path.exists('/.dockerenv'):
ret = merge(ret, docker_config())
return ret