added _get_version() method
[sfa.git] / 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/plc/sfa-import-plc.py', 
16     'sfa/plc/sfa-nuke-plc.py', 
17     'sfa/server/sfa-ca.py', 
18     'sfa/server/sfa-server.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/client',
40     'sfa/methods',
41     'sfa/plc',
42     'sfa/server',
43     'sfa/trust',
44     'sfa/util', 
45     'sfa/managers',
46     'sfa/rspecs',
47     'sfa/rspecs/elements',
48     'sfatables',
49     'sfatables/commands',
50     'sfatables/processors',
51     'flashpolicy',
52     ]
53
54
55 data_files = [('/etc/sfa/', [ 'config/aggregates.xml',
56                               'config/registries.xml',
57                               'config/default_config.xml',
58                               'config/sfi_config',
59                               'sfa/managers/pl/pl.rng',
60                               'sfa/trust/credential.xsd',
61                               'sfa/trust/top.xsd',
62                               'sfa/trust/sig.xsd',
63                               'sfa/trust/xml.xsd',
64                               'sfa/trust/protogeni-rspec-common.xsd',
65                               'flashpolicy/sfa_flashpolicy_config.xml',
66                             ]),
67               ('/etc/sfatables/matches/', glob('sfatables/matches/*.xml')),
68               ('/etc/sfatables/targets/', glob('sfatables/targets/*.xml')),
69               ('/etc/init.d/', ['sfa/init.d/sfa', 'sfa/init.d/sfa-cm'])]
70
71 # add sfatables processors as data_files
72 processor_files = [f for f in glob('sfatables/processors/*') if os.path.isfile(f)]
73 data_files.append(('/etc/sfatables/processors/', processor_files))
74 processor_subdirs = [d for d in glob('sfatables/processors/*') if os.path.isdir(d)]
75 for d in processor_subdirs:
76     etc_dir = os.path.join("/etc/sfatables/processors", os.path.basename(d))
77     d_files = [f for f in glob(d + '/*') if os.path.isfile(f)]
78     data_files.append((etc_dir, processor_files))
79
80 initscripts = [ '/etc/init.d/sfa', '/etc/init.d/sfa-cm' ]
81
82 if sys.argv[1] in ['uninstall', 'remove', 'delete', 'clean']:
83     python_path = sys.path
84     site_packages_path = [ os.path.join(p,'sfa') for p in python_path if p.endswith('site-packages')]
85     site_packages_path += [ os.path.join(p,'sfatables') for p in python_path if p.endswith('site-packages')]
86     remove_dirs = ['/etc/sfa/', '/etc/sfatables'] + site_packages_path
87     remove_bins = [ '/usr/bin/' + os.path.basename(bin) for bin in bins ]
88     remove_files = remove_bins + initscripts
89
90     # remove files   
91     for filepath in remove_files:
92         print "removing", filepath, "...",
93         try: 
94             os.remove(filepath)
95             print "success"
96         except: print "failed"
97     # remove directories 
98     for directory in remove_dirs: 
99         print "removing", directory, "...",
100         try: 
101             shutil.rmtree(directory)
102             print "success"
103         except: print "failed"
104 else:
105     # avoid repeating what's in the specfile already
106     setup(name='sfa',
107           packages = package_dirs, 
108           data_files = data_files,
109           ext_modules = [],
110           py_modules = [],
111           scripts = bins)
112