58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
from django.db.backends.base.client import BaseDatabaseClient
|
|
|
|
|
|
class DatabaseClient(BaseDatabaseClient):
|
|
executable_name = 'mysql'
|
|
|
|
@classmethod
|
|
def settings_to_cmd_args_env(cls, settings_dict, parameters):
|
|
args = [cls.executable_name]
|
|
env = None
|
|
db = settings_dict['OPTIONS'].get('db', settings_dict['NAME'])
|
|
user = settings_dict['OPTIONS'].get('user', settings_dict['USER'])
|
|
password = settings_dict['OPTIONS'].get(
|
|
'password',
|
|
settings_dict['OPTIONS'].get('passwd', settings_dict['PASSWORD'])
|
|
)
|
|
host = settings_dict['OPTIONS'].get('host', settings_dict['HOST'])
|
|
port = settings_dict['OPTIONS'].get('port', settings_dict['PORT'])
|
|
server_ca = settings_dict['OPTIONS'].get('ssl', {}).get('ca')
|
|
client_cert = settings_dict['OPTIONS'].get('ssl', {}).get('cert')
|
|
client_key = settings_dict['OPTIONS'].get('ssl', {}).get('key')
|
|
defaults_file = settings_dict['OPTIONS'].get('read_default_file')
|
|
charset = settings_dict['OPTIONS'].get('charset')
|
|
# Seems to be no good way to set sql_mode with CLI.
|
|
|
|
if defaults_file:
|
|
args += ["--defaults-file=%s" % defaults_file]
|
|
if user:
|
|
args += ["--user=%s" % user]
|
|
if password:
|
|
# The MYSQL_PWD environment variable usage is discouraged per
|
|
# MySQL's documentation due to the possibility of exposure through
|
|
# `ps` on old Unix flavors but --password suffers from the same
|
|
# flaw on even more systems. Usage of an environment variable also
|
|
# prevents password exposure if the subprocess.run(check=True) call
|
|
# raises a CalledProcessError since the string representation of
|
|
# the latter includes all of the provided `args`.
|
|
env = {'MYSQL_PWD': password}
|
|
if host:
|
|
if '/' in host:
|
|
args += ["--socket=%s" % host]
|
|
else:
|
|
args += ["--host=%s" % host]
|
|
if port:
|
|
args += ["--port=%s" % port]
|
|
if server_ca:
|
|
args += ["--ssl-ca=%s" % server_ca]
|
|
if client_cert:
|
|
args += ["--ssl-cert=%s" % client_cert]
|
|
if client_key:
|
|
args += ["--ssl-key=%s" % client_key]
|
|
if charset:
|
|
args += ['--default-character-set=%s' % charset]
|
|
if db:
|
|
args += [db]
|
|
args.extend(parameters)
|
|
return args, env
|