setup_posix.py 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. import os, sys
  2. from ConfigParser import SafeConfigParser
  3. # This dequote() business is required for some older versions
  4. # of mysql_config
  5. def dequote(s):
  6. if s[0] in "\"'" and s[0] == s[-1]:
  7. s = s[1:-1]
  8. return s
  9. def compiler_flag(f):
  10. return "-%s" % f
  11. def mysql_config(what):
  12. from os import popen
  13. f = popen("%s --%s" % (mysql_config.path, what))
  14. data = f.read().strip().split()
  15. ret = f.close()
  16. if ret:
  17. if ret/256:
  18. data = []
  19. if ret/256 > 1:
  20. raise EnvironmentError("%s not found" % (mysql_config.path,))
  21. return data
  22. mysql_config.path = "mysql_config"
  23. def get_config():
  24. from setup_common import get_metadata_and_options, enabled, create_release_file
  25. metadata, options = get_metadata_and_options()
  26. if 'mysql_config' in options:
  27. mysql_config.path = options['mysql_config']
  28. extra_objects = []
  29. static = enabled(options, 'static')
  30. if enabled(options, 'embedded'):
  31. libs = mysql_config("libmysqld-libs")
  32. client = "mysqld"
  33. elif enabled(options, 'threadsafe'):
  34. libs = mysql_config("libs_r")
  35. client = "mysqlclient_r"
  36. if not libs:
  37. libs = mysql_config("libs")
  38. client = "mysqlclient"
  39. else:
  40. libs = mysql_config("libs")
  41. client = "mysqlclient"
  42. library_dirs = [ dequote(i[2:]) for i in libs if i.startswith(compiler_flag("L")) ]
  43. libraries = [ dequote(i[2:]) for i in libs if i.startswith(compiler_flag("l")) ]
  44. removable_compile_args = [ compiler_flag(f) for f in "ILl" ]
  45. extra_compile_args = [ i.replace("%", "%%") for i in mysql_config("cflags")
  46. if i[:2] not in removable_compile_args ]
  47. # Copy the arch flags for linking as well
  48. extra_link_args = list()
  49. for i in range(len(extra_compile_args)):
  50. if extra_compile_args[i] == '-arch':
  51. extra_link_args += ['-arch', extra_compile_args[i + 1]]
  52. include_dirs = [ dequote(i[2:])
  53. for i in mysql_config('include')
  54. if i.startswith(compiler_flag('I')) ]
  55. if not include_dirs: # fix for MySQL-3.23
  56. include_dirs = [ dequote(i[2:])
  57. for i in mysql_config('cflags')
  58. if i.startswith(compiler_flag('I')) ]
  59. if static:
  60. extra_objects.append(os.path.join(library_dirs[0],'lib%s.a' % client))
  61. if client in libraries:
  62. libraries.remove(client)
  63. name = "MySQL-python"
  64. if enabled(options, 'embedded'):
  65. name = name + "-embedded"
  66. metadata['name'] = name
  67. define_macros = [
  68. ('version_info', metadata['version_info']),
  69. ('__version__', metadata['version']),
  70. ]
  71. create_release_file(metadata)
  72. del metadata['version_info']
  73. ext_options = dict(
  74. name = "_mysql",
  75. library_dirs = library_dirs,
  76. libraries = libraries,
  77. extra_compile_args = extra_compile_args,
  78. extra_link_args = extra_link_args,
  79. include_dirs = include_dirs,
  80. extra_objects = extra_objects,
  81. define_macros = define_macros,
  82. )
  83. return metadata, ext_options
  84. if __name__ == "__main__":
  85. sys.stderr.write("""You shouldn't be running this directly; it is used by setup.py.""")