setup.py

   1 #!/usr/bin/python
   2
   3 """
   4 Installation script for the sfa module
   5 """
   6
   7 import sys, os, os.path
   8 from glob import glob
   9 import shutil
  10 from distutils.core import setup
  11
  12 bins = [
  13     'config/sfa-config-tty',
  14     'config/gen-sfa-cm-config.py',
  15     'sfa/importer/sfa-import-plc.py',
  16     'sfa/importer/sfa-nuke-plc.py',
  17     'sfa/server/sfa-ca.py',
  18     'sfa/server/sfa-start.py',
  19     'sfa/server/sfa-clean-peer-records.py',
  20     'sfa/server/sfa_component_setup.py',
  21     'sfa/client/sfi.py',
  22     'sfa/client/getNodes.py',
  23     'sfa/client/getRecord.py',
  24     'sfa/client/setRecord.py',
  25     'sfa/client/sfadump.py',
  26     'sfa/client/sfiListNodes.py',
  27     'sfa/client/sfiListSlivers.py',
  28     'sfa/client/sfiAddSliver.py',
  29     'sfa/client/sfiDeleteSliver.py',
  30     'sfa/client/sfiAddAttribute.py',
  31     'sfa/client/sfiDeleteAttribute.py',
  32     'sfatables/sfatables',
  33     'keyconvert/keyconvert.py',
  34     'flashpolicy/sfa_flashpolicy.py',
  35     ]
  36
  37 package_dirs = [
  38     'sfa',
  39     'sfa/trust',
  40     'sfa/util',
  41     'sfa/client',
  42     'sfa/server',
  43     'sfa/methods',
  44     'sfa/generic',
  45     'sfa/managers',
  46     'sfa/managers/vini',
  47     'sfa/importer',
  48     'sfa/plc',
  49     'sfa/rspecs',
  50     'sfa/rspecs/elements',
  51     'sfa/rspecs/elements/versions',
  52     'sfa/rspecs/versions',
  53     'sfatables',
  54     'sfatables/commands',
  55     'sfatables/processors',
  56     'flashpolicy',
  57     ]
  58
  59
  60 data_files = [('/etc/sfa/', [ 'config/aggregates.xml',
  61                               'config/registries.xml',
  62                               'config/default_config.xml',
  63                               'config/sfi_config',
  64                               'config/topology',
  65                               'sfa/managers/pl/pl.rng',
  66                               'sfa/trust/credential.xsd',
  67                               'sfa/trust/top.xsd',
  68                               'sfa/trust/sig.xsd',
  69                               'sfa/trust/xml.xsd',
  70                               'sfa/trust/protogeni-rspec-common.xsd',
  71                               'flashpolicy/sfa_flashpolicy_config.xml',
  72                             ]),
  73               ('/etc/sfatables/matches/', glob('sfatables/matches/*.xml')),
  74               ('/etc/sfatables/targets/', glob('sfatables/targets/*.xml')),
  75               ('/etc/init.d/', ['sfa/init.d/sfa', 'sfa/init.d/sfa-cm'])]
  76
  77 # add sfatables processors as data_files
  78 processor_files = [f for f in glob('sfatables/processors/*') if os.path.isfile(f)]
  79 data_files.append(('/etc/sfatables/processors/', processor_files))
  80 processor_subdirs = [d for d in glob('sfatables/processors/*') if os.path.isdir(d)]
  81 for d in processor_subdirs:
  82     etc_dir = os.path.join("/etc/sfatables/processors", os.path.basename(d))
  83     d_files = [f for f in glob(d + '/*') if os.path.isfile(f)]
  84     data_files.append((etc_dir, processor_files))
  85
  86 initscripts = [ '/etc/init.d/sfa', '/etc/init.d/sfa-cm' ]
  87
  88 if sys.argv[1] in ['uninstall', 'remove', 'delete', 'clean']:
  89     python_path = sys.path
  90     site_packages_path = [ os.path.join(p,'sfa') for p in python_path if p.endswith('site-packages')]
  91     site_packages_path += [ os.path.join(p,'sfatables') for p in python_path if p.endswith('site-packages')]
  92     remove_dirs = ['/etc/sfa/', '/etc/sfatables'] + site_packages_path
  93     remove_bins = [ '/usr/bin/' + os.path.basename(bin) for bin in bins ]
  94     remove_files = remove_bins + initscripts
  95
  96     # remove files
  97     for filepath in remove_files:
  98         print "removing", filepath, "...",
  99         try:
 100             os.remove(filepath)
 101             print "success"
 102         except: print "failed"
 103     # remove directories
 104     for directory in remove_dirs:
 105         print "removing", directory, "...",
 106         try:
 107             shutil.rmtree(directory)
 108             print "success"
 109         except: print "failed"
 110 else:
 111     # avoid repeating what's in the specfile already
 112     setup(name='sfa',
 113           packages = package_dirs,
 114           data_files = data_files,
 115           ext_modules = [],
 116           py_modules = [],
 117           scripts = bins)
 118