setup.py

   1 #!/usr/bin/env python
   2
   3 import distutils.core
   4 import optparse
   5 import sys
   6 import os
   7 from glob import glob
   8 from os.path import join
   9
  10 def add_flag(option, opt, value, parser):
  11         dist_args.append(opt)
  12
  13 def add_value(option, opt, value, parser):
  14         dist_args.append(opt)
  15         dist_args.append(value)
  16
  17 def add_prefix_value(option, opt, value, parser):
  18         global dist_prefix
  19         dist_prefix = value
  20         add_value(option, opt, value, parser)
  21
  22 parser = optparse.OptionParser()
  23 parser.add_option(
  24         '--prefix',
  25         help='installation prefix',
  26         action='callback', callback=add_prefix_value, type=str)
  27 parser.add_option(
  28         '--compile',
  29         help='compile .py to .pyc [default]',
  30         action='callback', callback=add_flag)
  31 parser.add_option(
  32         '--no-compile',
  33         help="don't compile .py files",
  34         action='callback', callback=add_flag)
  35 parser.add_option(
  36         '-O', '--optimize',
  37         help='optimize',
  38         action='callback', callback=add_value, type=str)
  39 parser.add_option(
  40         '-f', '--force',
  41         help='force installation (overwrite any existing files)',
  42         action='callback', callback=add_flag)
  43
  44 parse_args = None
  45 dist_args = None
  46 dist_prefix = '/usr/local'
  47
  48 for i in xrange(1, len(sys.argv)):
  49         arg = sys.argv[i]
  50         if arg.startswith('-'):
  51                 continue
  52         if arg.startswith('install'):
  53                 dist_args = sys.argv[:i+1]
  54                 parse_args = sys.argv[i+1:]
  55         break
  56
  57 if parse_args:
  58         parser.usage = '%s [options]' % ' '.join(dist_args)
  59         parser.parse_args(args=parse_args)
  60
  61         sys.argv = dist_args
  62
  63 binfiles = [
  64         'bin/matrix',
  65         'bin/matrix-graph',
  66         'bin/matrix-upload',
  67         'bin/matrixify',
  68 ]
  69
  70 scriptdir = 'share/matrix/scripts'
  71
  72 distutils.core.setup(
  73         name='matrix',
  74         version='1.0-pre',
  75         license='GPL',
  76         packages=['matrix'],
  77         scripts=binfiles,
  78         data_files=[
  79                 (scriptdir, glob('scripts/*.sh') + glob('scripts/*.mk')),
  80                 (join(scriptdir, 'styles'), glob('scripts/styles/*.mk')),
  81         ],
  82 )
  83
  84 python_version = 'python%d.%d' % tuple(sys.version_info[0:2])
  85
  86 dist_moduledir = join(dist_prefix, 'lib', python_version, 'site-packages')
  87 dist_scriptdir = join(dist_prefix, scriptdir)
  88
  89 dist_binfiles = [join(dist_prefix, file) for file in binfiles]
  90
  91 try:
  92         while dist_binfiles:
  93                 path = dist_binfiles[0]
  94
  95                 file = open(path, 'rU')
  96                 lines = file.readlines()
  97                 file.close()
  98
  99                 file = open(path, 'w')
 100                 for line in lines:
 101                         line = line.replace('@MODULEDIR@', dist_moduledir)
 102                         line = line.replace('@SCRIPTDIR@', dist_scriptdir)
 103                         file.write(line)
 104                 file.close()
 105
 106                 del dist_binfiles[0]
 107 finally:
 108         for path in dist_binfiles:
 109                 try:
 110                         os.unlink(path)
 111                 except:
 112                         pass