forked from sensepost/dwn
/
setup.py
70 lines (53 loc) · 1.75 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
import os
from setuptools import setup, find_packages
from dwn import __version__
def _package_files(directory: str, suffix: str = None) -> list:
"""
Get all of the file paths in the directory specified by suffix.
:param directory:
:return:
"""
paths = []
for (path, directories, filenames) in os.walk(directory):
for filename in filenames:
if suffix is not None:
if filename.endswith(suffix):
paths.append(os.path.join('..', path, filename))
else:
paths.append(os.path.join('..', path, filename))
return paths
# here - where we are.
here = os.path.abspath(os.path.dirname(__file__))
# read the package requirements for install_requires
with open(os.path.join(here, 'requirements.txt'), 'r') as f:
requirements = f.readlines()
# setup!
setup(
name='dwn',
author='Leon Jacobs',
author_email='leon@sensepost.com',
description='dwn, a docker pwn tool manager',
license='GPL v3',
packages=find_packages(),
install_requires=requirements,
python_requires='>=3.8',
url='https://github.com/sensepost/dwn',
download_url='https://github.com/sensepost/dwn/archive/' + __version__ + '.tar.gz',
keywords=['docker', 'tool', 'pentest', 'framework'],
version=__version__,
# include other files
package_data={
'': _package_files(os.path.join(here, 'dwn/assets')) +
_package_files(os.path.join(here, 'plans/'))
},
classifiers=[
'Operating System :: OS Independent',
'Natural Language :: English',
'Programming Language :: Python :: 3 :: Only',
],
entry_points={
'console_scripts': [
'dwn = dwn.cli.cli:cli',
],
},
)