Skip to content

Commit

Permalink
Keep requirements clean.
Browse files Browse the repository at this point in the history
Also Black formatting changes.
  • Loading branch information
forestmonster committed Feb 13, 2020
1 parent 2e36fe1 commit 9d8b299
Showing 1 changed file with 45 additions and 47 deletions.
92 changes: 45 additions & 47 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,86 +12,84 @@

from setuptools import find_packages

if tuple(map(int, pip.__version__.split('.'))) >= (10, 0, 0):
if tuple(map(int, pip.__version__.split("."))) >= (10, 0, 0):
from pip._internal.download import PipSession
from pip._internal.req import parse_requirements
else:
from pip.download import PipSession
from pip.req import parse_requirements


with io.open('README.rst', encoding='utf-8') as readme:
with io.open("README.rst", encoding="utf-8") as readme:
long_description = readme.read()


def load_requirements(filename):
with io.open(filename, encoding='utf-8') as reqfile:
return [line.split()[0] for line in reqfile if not line.startswith(('#', '-'))]
with io.open(filename, encoding="utf-8") as reqfile:
return [line.split()[0] for line in reqfile if not line.startswith(("#", "-"))]


# Disabled the associated tests.
#def moto_broken():
# """Until https://github.com/spulec/moto/pull/1589 is resolved.
#
# Then we will no longer need to fork moto, roll our own release, and rely either on
# this hack, or the dependency_links declaration.
# """
# reqts = load_requirements('dev-requirements.txt')
# return reqts.append('moto==1.3.5')
def moto_broken():
"""Until https://github.com/spulec/moto/pull/1589 is resolved.
Then we will no longer need to fork moto, roll our own release, and rely either on
this hack, or the dependency_links declaration.
"""
reqts = load_requirements("dev-requirements.txt")
# return reqts.append('moto==1.3.5')
return reqts


# Populates __version__ without importing the package
__version__ = None
with io.open('diffy/_version.py', encoding='utf-8') as ver_file:
with io.open("diffy/_version.py", encoding="utf-8") as ver_file:
exec(ver_file.read()) # nosec: config file safe
if not __version__:
print('Could not find __version__ from diffy/_version.py')
print("Could not find __version__ from diffy/_version.py")
exit(1)


setup(
name='diffy',
name="diffy",
version=__version__,
author='Netflix',
author_email='[email protected]',
url='https://github.com/Netflix-Skunkworks/diffy',
description='Forensic differentiator',
author="Netflix",
author_email="[email protected]",
url="https://github.com/Netflix-Skunkworks/diffy",
description="Forensic differentiator",
long_description=long_description,
packages=find_packages(exclude=['diffy.tests']),
packages=find_packages(exclude=["diffy.tests"]),
include_package_data=True,
#dependency_links=['git+https://github.com/forestmonster/moto.git@master#egg=moto-1.3.5'],
install_requires=load_requirements('requirements.txt'),
tests_require=['pytest'],
# dependency_links=['git+https://github.com/forestmonster/moto.git@master#egg=moto-1.3.5'],
install_requires=load_requirements("requirements.txt"),
tests_require=["pytest"],
extras_require={
'dev': load_requirements('dev-requirements.txt'),
'web': load_requirements('web-requirements.txt'),
"dev": load_requirements("dev-requirements.txt"),
"web": load_requirements("web-requirements.txt"),
},
entry_points={
'console_scripts': [
'diffy = diffy_cli.core:entry_point'
],
'diffy.plugins': [
'aws_persistence_s3 = diffy.plugins.diffy_aws.plugin:S3PersistencePlugin',
'aws_collection_ssm = diffy.plugins.diffy_aws.plugin:SSMCollectionPlugin',
'aws_target_auto_scaling_group = diffy.plugins.diffy_aws.plugin:AutoScalingTargetPlugin',
'local_analysis_simple = diffy.plugins.diffy_local.plugin:SimpleAnalysisPlugin',
'local_analysis_cluster = diffy.plugins.diffy_local.plugin:ClusterAnalysisPlugin',
'local_persistence_file = diffy.plugins.diffy_local.plugin:FilePersistencePlugin',
'local_payload_command = diffy.plugins.diffy_local.plugin:CommandPayloadPlugin',
'local_shell_collection = diffy.plugins.diffy_local.plugin:LocalShellCollectionPlugin',
'local_target = diffy.plugins.diffy_local.plugin:LocalTargetPlugin',
'osquery_payload = diffy.plugins.diffy_osquery.plugin:OSQueryPayloadPlugin'
]
"console_scripts": ["diffy = diffy_cli.core:entry_point"],
"diffy.plugins": [
"aws_persistence_s3 = diffy.plugins.diffy_aws.plugin:S3PersistencePlugin",
"aws_collection_ssm = diffy.plugins.diffy_aws.plugin:SSMCollectionPlugin",
"aws_target_auto_scaling_group = diffy.plugins.diffy_aws.plugin:AutoScalingTargetPlugin",
"local_analysis_simple = diffy.plugins.diffy_local.plugin:SimpleAnalysisPlugin",
"local_analysis_cluster = diffy.plugins.diffy_local.plugin:ClusterAnalysisPlugin",
"local_persistence_file = diffy.plugins.diffy_local.plugin:FilePersistencePlugin",
"local_payload_command = diffy.plugins.diffy_local.plugin:CommandPayloadPlugin",
"local_shell_collection = diffy.plugins.diffy_local.plugin:LocalShellCollectionPlugin",
"local_target = diffy.plugins.diffy_local.plugin:LocalTargetPlugin",
"osquery_payload = diffy.plugins.diffy_osquery.plugin:OSQueryPayloadPlugin",
],
},
classifiers=[
'Intended Audience :: Developers',
'Intended Audience :: System Administrators',
'Operating System :: OS Independent',
'Topic :: Software Development',
"Intended Audience :: Developers",
"Intended Audience :: System Administrators",
"Operating System :: OS Independent",
"Topic :: Software Development",
"Programming Language :: Python",
"Programming Language :: Python :: 3.6",
"Natural Language :: English",
"License :: OSI Approved :: Apache Software License"
"License :: OSI Approved :: Apache Software License",
],
python_requires='>=3.6'
python_requires=">=3.6",
)

0 comments on commit 9d8b299

Please sign in to comment.