mysqlclient/setup_posix.py

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):
    cmd = "{} --{}".format(_mysql_config_path, what)
    print(cmd)
    f = os.popen(cmd)
    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))
    print(data)
    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 = os.environ.get("MYSQLCLIENT_LDFLAGS")
    if libs:
        libs = libs.strip().split()
    else:
        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"))]

    cflags = os.environ.get("MYSQLCLIENT_CFLAGS")
    if cflags:
        use_mysqlconfig_cflags = False
        cflags = cflags.strip().split()
    else:
        use_mysqlconfig_cflags = True
        cflags = mysql_config("cflags")

    include_dirs = []
    extra_compile_args = ["-std=c99"]

    for a in cflags:
        if a.startswith("-I"):
            include_dirs.append(dequote(a[2:]))
        elif a.startswith(("-L", "-l")):  # This should be LIBS.
            pass
        else:
            extra_compile_args.append(a.replace("%", "%%"))

    # Copy the arch flags for linking as well
    try:
        i = extra_compile_args.index("-arch")
        if "-arch" not in extra_link_args:
            extra_link_args += ["-arch", extra_compile_args[i + 1]]
    except ValueError:
        pass

    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:
        if use_mysqlconfig_cflags:
            # 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", "zstd"):
                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++"

    print("ext_options:")
    for k, v in ext_options.items():
        print("  {}: {}".format(k, v))

    return metadata, ext_options


if __name__ == "__main__":
    sys.stderr.write(
        """You shouldn't be running this directly; it is used by setup.py."""
    )