1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
#!/usr/bin/env python -O
# -*- coding: utf-8 -*-
import os
import mariadb
from .conf_test import conf
def is_skysql():
if conf()["host"][-13:] == "db.skysql.net":
return True
return False
def is_maxscale():
return (os.environ.get('srv') == "maxscale" or
os.environ.get('srv') == 'skysql-ha')
def is_mysql():
mysql_server = 1
conn = create_connection()
cursor = conn.cursor()
cursor.execute("select version()")
row = cursor.fetchone()
if "MARIADB" in row[0].upper():
mysql_server = 0
conn.close()
del cursor, conn
return mysql_server
def get_host_suffix():
return "@'localhost'" if os.getenv("LOCAL_DB", "container") == "local" else "@'%'"
def create_connection(additional_conf=None):
default_conf = conf()
if additional_conf is None:
c = {key: value for (key, value) in (default_conf.items())}
else:
c = {key: value for (key, value) in (list(default_conf.items()) + list(
additional_conf.items()))}
return mariadb.connect(**c)
|
