setup.py

#!/usr/bin/env python
importos
importsubprocess
importsys

importsetuptools
fromconfigparserimportConfigParser


release_info= {}
withopen("src/MySQLdb/release.py", encoding="utf-8") asf:
exec(f.read(), None, release_info)


deffind_package_name():
"""Get available pkg-config package name"""
# Ubuntu uses mariadb.pc, but CentOS uses libmariadb.pc
packages= ["mysqlclient", "mariadb", "libmariadb", "perconaserverclient"]
forpkginpackages:
try:
cmd=f"pkg-config --exists {pkg}"
print(f"Trying {cmd}")
subprocess.check_call(cmd, shell=True)
exceptsubprocess.CalledProcessErroraserr:
print(err)
else:
returnpkg
raiseException(
"Can not find valid pkg-config name.\n"
"Specify MYSQLCLIENT_CFLAGS and MYSQLCLIENT_LDFLAGS env vars manually"
 )


defget_config_posix(options=None):
# allow a command-line option to override the base config file to permit
# a static build to be created via requirements.txt
# TODO: find a better way for
static=False
if"--static"insys.argv:
static=True
sys.argv.remove("--static")

ldflags=os.environ.get("MYSQLCLIENT_LDFLAGS")
cflags=os.environ.get("MYSQLCLIENT_CFLAGS")

pkg_name=None
static_opt=" --static"ifstaticelse""
ifnot (cflagsandldflags):
pkg_name=find_package_name()
ifnotcflags:
cflags=subprocess.check_output(
f"pkg-config{static_opt} --cflags {pkg_name}", encoding="utf-8", shell=True
 )
ifnotldflags:
ldflags=subprocess.check_output(
f"pkg-config{static_opt} --libs {pkg_name}", encoding="utf-8", shell=True
 )

cflags=cflags.split()
forfincflags:
iff.startswith("-std="):
break
else:
cflags+= ["-std=c99"]

ldflags=ldflags.split()

define_macros= [
 ("version_info", release_info["version_info"]),
 ("__version__", release_info["__version__"]),
 ]

ext_options=dict(
extra_compile_args=cflags,
extra_link_args=ldflags,
define_macros=define_macros,
 )
# newer versions of gcc require libstdc++ if doing a static build
ifstatic:
ext_options["language"] ="c++"

returnext_options


defget_config_win32(options):
client="mariadbclient"
connector=os.environ.get("MYSQLCLIENT_CONNECTOR", options.get("connector"))
ifnotconnector:
connector=os.path.join(
os.environ["ProgramFiles"], "MariaDB", "MariaDB Connector C"
 )

extra_objects= []

library_dirs= [
os.path.join(connector, "lib", "mariadb"),
os.path.join(connector, "lib"),
 ]
libraries= [
"kernel32",
"advapi32",
"wsock32",
"shlwapi",
"Ws2_32",
"crypt32",
"secur32",
"bcrypt",
client,
 ]
include_dirs= [
os.path.join(connector, "include", "mariadb"),
os.path.join(connector, "include", "mysql"),
os.path.join(connector, "include"),
 ]

extra_link_args= ["/MANIFEST"]

define_macros= [
 ("version_info", release_info["version_info"]),
 ("__version__", release_info["__version__"]),
 ]

ext_options=dict(
library_dirs=library_dirs,
libraries=libraries,
extra_link_args=extra_link_args,
include_dirs=include_dirs,
extra_objects=extra_objects,
define_macros=define_macros,
 )
returnext_options


defenabled(options, option):
value=options[option]
s=value.lower()
ifsin ("yes", "true", "1", "y"):
returnTrue
elifsin ("no", "false", "0", "n"):
returnFalse
else:
raiseValueError(f"Unknown value {value} for option {option}")


defget_options():
config=ConfigParser()
config.read(["site.cfg"])
options=dict(config.items("options"))
options["static"] =enabled(options, "static")
returnoptions


ifsys.platform=="win32":
ext_options=get_config_win32(get_options())
else:
ext_options=get_config_posix(get_options())

print("# Options for building extension module:")
fork, vinext_options.items():
print(f" {k}: {v}")

ext_modules= [
setuptools.Extension(
"MySQLdb._mysql",
sources=["src/MySQLdb/_mysql.c"],
**ext_options,
 )
]
setuptools.setup(ext_modules=ext_modules)