mirror of
https://github.com/PyMySQL/mysqlclient.git
synced 2025-08-15 11:10:58 +08:00
148 lines
4.1 KiB
Python
148 lines
4.1 KiB
Python
import os
|
|
import sys
|
|
|
|
# This dequote() business is required for some older versions
|
|
# of mysql_config
|
|
|
|
|
|
def dequote(s):
|
|
if not s:
|
|
raise Exception(
|
|
"Wrong MySQL configuration: maybe https://bugs.mysql.com/bug.php?id=86971 ?"
|
|
)
|
|
if s[0] in "\"'" and s[0] == s[-1]:
|
|
s = s[1:-1]
|
|
return s
|
|
|
|
|
|
_mysql_config_path = "mysql_config"
|
|
|
|
|
|
def mysql_config(what):
|
|
from os import popen
|
|
|
|
f = popen("{} --{}".format(_mysql_config_path, what))
|
|
data = f.read().strip().split()
|
|
ret = f.close()
|
|
if ret:
|
|
if ret / 256:
|
|
data = []
|
|
if ret / 256 > 1:
|
|
raise OSError("{} not found".format(_mysql_config_path))
|
|
return data
|
|
|
|
|
|
def get_config():
|
|
from setup_common import get_metadata_and_options, enabled, create_release_file
|
|
|
|
global _mysql_config_path
|
|
|
|
metadata, options = get_metadata_and_options()
|
|
|
|
if "mysql_config" in options:
|
|
_mysql_config_path = options["mysql_config"]
|
|
else:
|
|
try:
|
|
mysql_config("version")
|
|
except OSError:
|
|
# try mariadb_config
|
|
_mysql_config_path = "mariadb_config"
|
|
try:
|
|
mysql_config("version")
|
|
except OSError:
|
|
_mysql_config_path = "mysql_config"
|
|
|
|
extra_objects = []
|
|
static = enabled(options, "static")
|
|
|
|
# allow a command-line option to override the base config file to permit
|
|
# a static build to be created via requirements.txt
|
|
#
|
|
if "--static" in sys.argv:
|
|
static = True
|
|
sys.argv.remove("--static")
|
|
|
|
libs = mysql_config("libs")
|
|
library_dirs = [dequote(i[2:]) for i in libs if i.startswith("-L")]
|
|
libraries = [dequote(i[2:]) for i in libs if i.startswith("-l")]
|
|
extra_link_args = [x for x in libs if not x.startswith(("-l", "-L"))]
|
|
|
|
removable_compile_args = ("-I", "-L", "-l")
|
|
extra_compile_args = [
|
|
i.replace("%", "%%")
|
|
for i in mysql_config("cflags")
|
|
if i[:2] not in removable_compile_args
|
|
]
|
|
|
|
# Copy the arch flags for linking as well
|
|
for i in range(len(extra_compile_args)):
|
|
if extra_compile_args[i] == "-arch":
|
|
extra_link_args += ["-arch", extra_compile_args[i + 1]]
|
|
|
|
include_dirs = [
|
|
dequote(i[2:]) for i in mysql_config("include") if i.startswith("-I")
|
|
]
|
|
|
|
if static:
|
|
# properly handle mysql client libraries that are not called libmysqlclient
|
|
client = None
|
|
CLIENT_LIST = [
|
|
"mysqlclient",
|
|
"mysqlclient_r",
|
|
"mysqld",
|
|
"mariadb",
|
|
"mariadbclient",
|
|
"perconaserverclient",
|
|
"perconaserverclient_r",
|
|
]
|
|
for c in CLIENT_LIST:
|
|
if c in libraries:
|
|
client = c
|
|
break
|
|
|
|
if client == "mariadb":
|
|
client = "mariadbclient"
|
|
if client is None:
|
|
raise ValueError("Couldn't identify mysql client library")
|
|
|
|
extra_objects.append(os.path.join(library_dirs[0], "lib%s.a" % client))
|
|
if client in libraries:
|
|
libraries.remove(client)
|
|
else:
|
|
# mysql_config may have "-lmysqlclient -lz -lssl -lcrypto", but zlib and
|
|
# ssl is not used by _mysql. They are needed only for static build.
|
|
for L in ("crypto", "ssl", "z"):
|
|
if L in libraries:
|
|
libraries.remove(L)
|
|
|
|
name = "mysqlclient"
|
|
metadata["name"] = name
|
|
|
|
define_macros = [
|
|
("version_info", metadata["version_info"]),
|
|
("__version__", metadata["version"]),
|
|
]
|
|
create_release_file(metadata)
|
|
del metadata["version_info"]
|
|
ext_options = dict(
|
|
library_dirs=library_dirs,
|
|
libraries=libraries,
|
|
extra_compile_args=extra_compile_args,
|
|
extra_link_args=extra_link_args,
|
|
include_dirs=include_dirs,
|
|
extra_objects=extra_objects,
|
|
define_macros=define_macros,
|
|
)
|
|
|
|
# newer versions of gcc require libstdc++ if doing a static build
|
|
if static:
|
|
ext_options["language"] = "c++"
|
|
|
|
return metadata, ext_options
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.stderr.write(
|
|
"""You shouldn't be running this directly; it is used by setup.py."""
|
|
)
|