forked from ForensicArtifacts/artifacts
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
executable file
·134 lines (106 loc) · 3.86 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#!/usr/bin/python
# -*- coding: utf-8 -*-
"""Installation and deployment script."""
import glob
import os
import sys
try:
from setuptools import find_packages, setup, Command
except ImportError:
from distutils.core import find_packages, setup, Command
try:
from setuptools.commands.bdist_rpm import bdist_rpm
except ImportError:
from distutils.command.bdist_rpm import bdist_rpm
# Change PYTHONPATH to include artifacts so that we can get the version.
sys.path.insert(0, '.')
import artifacts
class BdistRPMCommand(bdist_rpm):
"""Custom handler for the bdist_rpm command."""
def _make_spec_file(self):
"""Generates the text of an RPM spec file.
Returns:
A list of strings containing the lines of text.
"""
# Note that bdist_rpm can be an old style class.
if issubclass(BdistRPMCommand, object):
spec_file = super(BdistRPMCommand, self)._make_spec_file()
else:
spec_file = bdist_rpm._make_spec_file(self)
if sys.version_info[0] < 3:
python_package = 'python'
else:
python_package = 'python3'
description = []
summary = ''
in_description = False
python_spec_file = []
for index, line in enumerate(spec_file):
if line.startswith('Summary: '):
summary = line
elif line.startswith('BuildRequires: '):
line = 'BuildRequires: {0:s}-setuptools'.format(python_package)
elif line.startswith('Requires: '):
if python_package == 'python3':
line = line.replace('python', 'python3')
elif line.startswith('%description'):
in_description = True
elif line.startswith('%files'):
line = '%files -f INSTALLED_FILES -n {0:s}-%{{name}}'.format(
python_package)
elif line.startswith('%prep'):
in_description = False
python_spec_file.append(
'%package -n {0:s}-%{{name}}'.format(python_package))
python_spec_file.append('{0:s}'.format(summary))
python_spec_file.append('')
python_spec_file.append(
'%description -n {0:s}-%{{name}}'.format(python_package))
python_spec_file.extend(description)
elif in_description:
# Ignore leading white lines in the description.
if not description and not line:
continue
description.append(line)
python_spec_file.append(line)
return python_spec_file
artifacts_version = artifacts.__version__
# Command bdist_msi does not support the library version, neither a date
# as a version but if we suffix it with .1 everything is fine.
if 'bdist_msi' in sys.argv:
artifacts_version = '{0}.1'.format(artifacts_version)
artifacts_description = (
'ForensicArtifacts.com Artifact Repository.')
artifacts_long_description = (
'A free, community-sourced, machine-readable knowledge base of forensic '
'artifacts that the world can use both as an information source and '
'within other tools.')
setup(
name='artifacts',
version=artifacts_version,
description=artifacts_description,
long_description=artifacts_long_description,
license='Apache License, Version 2.0',
url='https://github.com/ForensicArtifacts/artifacts',
maintainer='The ForensicArtifacts.com Artifact Repository project',
scripts=[
os.path.join('tools', 'stats.py'),
os.path.join('tools', 'validator.py'),
],
cmdclass={'bdist_rpm': BdistRPMCommand},
classifiers=[
'Development Status :: 3 - Alpha',
'Environment :: Console',
'Operating System :: OS Independent',
'Programming Language :: Python',
],
packages=find_packages('.', exclude=[u'tests', u'tests.*', u'tools']),
package_dir={'artifacts': 'artifacts'},
data_files=[
('share/artifacts', glob.glob(os.path.join('definitions', '*'))),
],
install_requires=[
'PyYAML >= 3.11',
],
)