from galaxy.structured_app import BasicSharedApp
[docs]class ApiKeyManager:
[docs] def __init__(self, app: BasicSharedApp):
self.app = app
[docs] def create_api_key(self, user) -> str:
guid = self.app.security.get_new_guid()
new_key = self.app.model.APIKeys()
new_key.user_id = user.id
new_key.key = guid
sa_session = self.app.model.context
sa_session.add(new_key)
sa_session.flush()
return guid
[docs] def get_or_create_api_key(self, user) -> str:
# Logic Galaxy has always used - but it would appear to have a race
# condition. Worth fixing? Would kind of need a message queue to fix
# in multiple process mode.
if user.api_keys:
key = user.api_keys[0].key
else:
key = self.create_api_key(user)
return key