mirror of
https://github.com/ansible/awx.git
synced 2026-02-24 14:36:00 -03:30
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import os
|
|
import sys
|
|
|
|
|
|
def get_service_name(argv):
|
|
'''
|
|
Return best-effort guess as to the name of this service
|
|
'''
|
|
for arg in argv:
|
|
if arg == '-m':
|
|
continue
|
|
if 'python' in arg:
|
|
continue
|
|
if 'manage' in arg:
|
|
continue
|
|
if arg.startswith('run_'):
|
|
return arg[len('run_') :]
|
|
return arg
|
|
|
|
|
|
def get_application_name(CLUSTER_HOST_ID, function=''):
|
|
if function:
|
|
function = f'_{function}'
|
|
return f'awx-{os.getpid()}-{get_service_name(sys.argv)}{function}-{CLUSTER_HOST_ID}'[:63]
|
|
|
|
|
|
def set_application_name(DATABASES, CLUSTER_HOST_ID, function=''):
|
|
"""In place modification of DATABASES to set the application name for the connection."""
|
|
# If settings files were not properly passed DATABASES could be {} at which point we don't need to set the app name.
|
|
if not DATABASES or 'default' not in DATABASES:
|
|
return
|
|
|
|
if 'sqlite3' in DATABASES['default']['ENGINE']:
|
|
return
|
|
options_dict = DATABASES['default'].setdefault('OPTIONS', dict())
|
|
options_dict['application_name'] = get_application_name(CLUSTER_HOST_ID, function)
|