ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-06-26 23:15:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for fix user list credentials

Browse Source
This commit is contained in:
allegroai
2019-07-17 18:15:58 +03:00
parent b7614622fc
commit 1a00f29415
3 changed files with 46 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
server/database/__init__.py
View File

@@ -1,3 +1,6 @@
from os import getenv
from furl import furl
from jsonmodels import models
from jsonmodels.errors import ValidationError
from jsonmodels.fields import StringField
@@ -8,9 +11,11 @@ from config import config
from .defs import Database
from .utils import get_items
log = config.logger(__file__)
log = config.logger("database")
strict = config.get('apiserver.mongo.strict', True)
strict = config.get("apiserver.mongo.strict", True)
OVERRIDE_HOST_ENV_KEY = "MONGODB_SERVICE_SERVICE_HOST"
_entries = []
@@ -21,28 +26,39 @@ class DatabaseEntry(models.Base):
@property
def health_alias(self):
return '__health__' + self.alias
return "__health__" + self.alias
def initialize():
db_entries = config.get('hosts.mongo', {})
db_entries = config.get("hosts.mongo", {})
missing = []
log.info('Initializing database connections')
log.info("Initializing database connections")
override_hostname = getenv(OVERRIDE_HOST_ENV_KEY)
if override_hostname:
log.info(f"Using override mongodb host {override_hostname}")
for key, alias in get_items(Database).items():
if key not in db_entries:
missing.append(key)
continue
entry = DatabaseEntry(alias=alias, **db_entries.get(key))
if override_hostname:
entry.host = furl(entry.host).set(host=override_hostname).url
try:
entry.validate()
log.info('Registering connection to %(alias)s (%(host)s)' % entry.to_struct())
log.info(
"Registering connection to %(alias)s (%(host)s)" % entry.to_struct()
)
register_connection(alias=alias, host=entry.host)
_entries.append(entry)
except ValidationError as ex:
raise Exception('Invalid database entry `%s`: %s' % (key, ex.args[0]))
raise Exception("Invalid database entry `%s`: %s" % (key, ex.args[0]))
if missing:
raise ValueError('Missing database configuration for %s' % ', '.join(missing))
raise ValueError("Missing database configuration for %s" % ", ".join(missing))
def get_entries():