setup.py

   1 #!/usr/bin/python
   2
   3 from distutils.core import setup, Extension, Command
   4 from distutils.command.build import build
   5 from distutils.command.clean import clean
   6 from distutils.command.sdist import sdist
   7 from distutils.dir_util import remove_tree
   8 from distutils.util import get_platform
   9 from distutils.spawn import spawn
  10 from distutils.errors import DistutilsExecError
  11 import distutils
  12
  13 import sys
  14 import datetime
  15 import os
  16 import os.path
  17 import re
  18 import time
  19
  20 MIN_LIBVIRT = "0.9.11"
  21 MIN_LIBVIRT_LXC = "1.0.2"
  22
  23 # Hack to stop 'pip install' failing with error
  24 # about missing 'build' dir.
  25 if not os.path.exists("build"):
  26     os.mkdir("build")
  27
  28 pkgcfg = distutils.spawn.find_executable("pkg-config")
  29
  30 if pkgcfg is None:
  31     raise Exception("pkg-config binary is required to compile libvirt-python")
  32
  33 spawn([pkgcfg,
  34        "--print-errors",
  35        "--atleast-version=%s" % MIN_LIBVIRT,
  36        "libvirt"])
  37
  38 have_libvirt_lxc=True
  39 try:
  40     spawn([pkgcfg,
  41            "--atleast-version=%s" % MIN_LIBVIRT_LXC,
  42          "libvirt"])
  43 except DistutilsExecError:
  44     have_libvirt_lxc=False
  45
  46 def get_pkgconfig_data(args, mod, required=True):
  47     """Run pkg-config to and return content associated with it"""
  48     f = os.popen("%s %s %s" % (pkgcfg, " ".join(args), mod))
  49
  50     line = f.readline()
  51     if line is not None:
  52         line = line.strip()
  53
  54     if line is None or line == "":
  55         if required:
  56             raise Exception("Cannot determine '%s' from libvirt pkg-config file" % " ".join(args))
  57         else:
  58             return ""
  59
  60     return line
  61
  62 def get_api_xml_files():
  63     """Check with pkg-config that libvirt is present and extract
  64     the API XML file paths we need from it"""
  65
  66     libvirt_api = get_pkgconfig_data(["--variable", "libvirt_api"], "libvirt")
  67
  68     offset = libvirt_api.index("-api.xml")
  69     libvirt_qemu_api = libvirt_api[0:offset] + "-qemu-api.xml"
  70
  71     offset = libvirt_api.index("-api.xml")
  72     libvirt_lxc_api = libvirt_api[0:offset] + "-lxc-api.xml"
  73
  74     return (libvirt_api, libvirt_qemu_api, libvirt_lxc_api)
  75
  76 ldflags = get_pkgconfig_data(["--libs-only-L"], "libvirt", False)
  77 cflags = get_pkgconfig_data(["--cflags"], "libvirt", False)
  78
  79 c_modules = []
  80 py_modules = []
  81
  82 module = Extension('libvirtmod',
  83                    sources = ['libvirt-override.c', 'build/libvirt.c', 'typewrappers.c', 'libvirt-utils.c'],
  84                    libraries = [ "virt" ],
  85                    include_dirs = [ "." ])
  86 if cflags != "":
  87     module.extra_compile_args.append(cflags)
  88 if ldflags != "":
  89     module.extra_link_args.append(ldflags)
  90
  91 c_modules.append(module)
  92 py_modules.append("libvirt")
  93
  94 moduleqemu = Extension('libvirtmod_qemu',
  95                        sources = ['libvirt-qemu-override.c', 'build/libvirt-qemu.c', 'typewrappers.c', 'libvirt-utils.c'],
  96                        libraries = [ "virt-qemu" ],
  97                        include_dirs = [ "." ])
  98 if cflags != "":
  99     moduleqemu.extra_compile_args.append(cflags)
 100 if ldflags != "":
 101     moduleqemu.extra_link_args.append(ldflags)
 102
 103 c_modules.append(moduleqemu)
 104 py_modules.append("libvirt_qemu")
 105
 106 if have_libvirt_lxc:
 107     modulelxc = Extension('libvirtmod_lxc',
 108                           sources = ['libvirt-lxc-override.c', 'build/libvirt-lxc.c', 'typewrappers.c', 'libvirt-utils.c'],
 109                           libraries = [ "virt-lxc" ],
 110                           include_dirs = [ "." ])
 111     if cflags != "":
 112         modulelxc.extra_compile_args.append(cflags)
 113     if ldflags != "":
 114         modulelxc.extra_link_args.append(ldflags)
 115
 116     c_modules.append(modulelxc)
 117     py_modules.append("libvirt_lxc")
 118
 119
 120 class my_build(build):
 121
 122     def run(self):
 123         apis = get_api_xml_files()
 124
 125         self.spawn([sys.executable, "generator.py", "libvirt", apis[0]])
 126         self.spawn([sys.executable, "generator.py", "libvirt-qemu", apis[1]])
 127         if have_libvirt_lxc:
 128             self.spawn([sys.executable, "generator.py", "libvirt-lxc", apis[2]])
 129
 130         build.run(self)
 131
 132 class my_sdist(sdist):
 133     user_options = sdist.user_options
 134
 135     description = "Update libvirt-python.spec; build sdist-tarball."
 136
 137     def initialize_options(self):
 138         self.snapshot = None
 139         sdist.initialize_options(self)
 140
 141     def finalize_options(self):
 142         if self.snapshot is not None:
 143             self.snapshot = 1
 144         sdist.finalize_options(self)
 145
 146     def gen_rpm_spec(self):
 147         f1 = open('libvirt-python.spec.in', 'r')
 148         f2 = open('libvirt-python.spec', 'w')
 149         for line in f1:
 150             f2.write(line
 151                      .replace('@PY_VERSION@', self.distribution.get_version())
 152                      .replace('@C_VERSION@', MIN_LIBVIRT))
 153         f1.close()
 154         f2.close()
 155
 156     def gen_authors(self):
 157         f = os.popen("git log --pretty=format:'%aN <%aE>'")
 158         authors = []
 159         for line in f:
 160             authors.append("   " + line.strip())
 161
 162         authors.sort(key=str.lower)
 163
 164         f1 = open('AUTHORS.in', 'r')
 165         f2 = open('AUTHORS', 'w')
 166         for line in f1:
 167             f2.write(line.replace('@AUTHORS@', "\n".join(authors)))
 168         f1.close()
 169         f2.close()
 170
 171
 172     def gen_changelog(self):
 173         f1 = os.popen("git log '--pretty=format:%H:%ct %an  <%ae>%n%n%s%n%b%n'")
 174         f2 = open("ChangeLog", 'w')
 175
 176         for line in f1:
 177             m = re.match(r'([a-f0-9]+):(\d+)\s(.*)', line)
 178             if m:
 179                 t = time.gmtime(int(m.group(2)))
 180                 f2.write("%04d-%02d-%02d %s\n" % (t.tm_year, t.tm_mon, t.tm_mday, m.group(3)))
 181             else:
 182                 if re.match(r'Signed-off-by', line):
 183                     continue
 184                 f2.write("    " + line.strip() + "\n")
 185
 186         f1.close()
 187         f2.close()
 188
 189
 190     def run(self):
 191         if not os.path.exists("build"):
 192             os.mkdir("build")
 193
 194         if os.path.exists(".git"):
 195             try:
 196                 self.gen_rpm_spec()
 197                 self.gen_authors()
 198                 self.gen_changelog()
 199
 200                 sdist.run(self)
 201
 202             finally:
 203                 files = ["libvirt-python.spec",
 204                          "AUTHORS",
 205                          "ChangeLog"]
 206                 for f in files:
 207                     if os.path.exists(f):
 208                         os.unlink(f)
 209         else:
 210             sdist.run(self)
 211
 212 class my_rpm(Command):
 213     user_options = []
 214
 215     description = "Build src and noarch rpms."
 216
 217     def initialize_options(self):
 218         pass
 219
 220     def finalize_options(self):
 221         pass
 222
 223     def run(self):
 224         """
 225         Run sdist, then 'rpmbuild' the tar.gz
 226         """
 227
 228         self.run_command('sdist')
 229         os.system('rpmbuild -ta --clean dist/libvirt-python-%s.tar.gz' %
 230                   self.distribution.get_version())
 231
 232 class my_test(Command):
 233     user_options = [
 234         ('build-base=', 'b',
 235          "base directory for build library"),
 236         ('build-platlib=', None,
 237          "build directory for platform-specific distributions"),
 238         ('plat-name=', 'p',
 239          "platform name to build for, if supported "
 240          "(default: %s)" % get_platform()),
 241     ]
 242
 243     description = "Run test suite."
 244
 245     def initialize_options(self):
 246         self.build_base = 'build'
 247         self.build_platlib = None
 248         self.plat_name = None
 249
 250     def finalize_options(self):
 251         if self.plat_name is None:
 252             self.plat_name = get_platform()
 253
 254         plat_specifier = ".%s-%s" % (self.plat_name, sys.version[0:3])
 255
 256         if hasattr(sys, 'gettotalrefcount'):
 257             plat_specifier += '-pydebug'
 258
 259         if self.build_platlib is None:
 260             self.build_platlib = os.path.join(self.build_base,
 261                                               'lib' + plat_specifier)
 262
 263     def run(self):
 264         """
 265         Run test suite
 266         """
 267
 268         apis = get_api_xml_files()
 269
 270         self.spawn([sys.executable, "sanitytest.py", self.build_platlib, apis[0]])
 271
 272
 273 class my_clean(clean):
 274     def run(self):
 275         clean.run(self)
 276
 277         if os.path.exists("build"):
 278             remove_tree("build")
 279
 280 setup(name = 'libvirt-python',
 281       version = '1.2.1',
 282       url = 'http://www.libvirt.org',
 283       maintainer = 'Libvirt Maintainers',
 284       maintainer_email = 'libvir-list@redhat.com',
 285       description = 'The libvirt virtualization API',
 286       ext_modules = c_modules,
 287       py_modules = py_modules,
 288       package_dir = {
 289           '': 'build'
 290       },
 291       cmdclass = {
 292           'build': my_build,
 293           'clean': my_clean,
 294           'sdist': my_sdist,
 295           'rpm': my_rpm,
 296           'test': my_test
 297       })