diff --git a/MANIFEST.in b/MANIFEST.in index f9bd145..fafba2f 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -1 +1,4 @@ include requirements.txt +include requirements_test.txt + + diff --git a/setup.py b/setup.py index 52c8fd1..21e40ee 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,6 @@ -from setuptools import setup, find_packages +from setuptools import setup, find_packages, Command from codecs import open -from os import path +from os import path, system import re package_name = "pytchat" @@ -14,7 +14,6 @@ def _test_requirements(): return [name.rstrip() for name in open(path.join(root_dir, 'requirements_test.txt')).readlines()] - with open(path.join(root_dir, package_name, '__init__.py')) as f: init_text = f.read() version = re.search(r'__version__\s*=\s*[\'\"](.+?)[\'\"]', init_text).group(1) @@ -32,6 +31,18 @@ assert url with open('README.md', encoding='utf-8') as f: long_description = f.read() +class CleanCommand(Command): + """Custom clean command to tidy up the project root.""" + user_options = [] + def initialize_options(self): + pass + def finalize_options(self): + pass + def run(self): + #system('rm -vrf ./build ./dist ./*.pyc ./*.tgz ./*.egg-info') + system('rmdir /Q pytchat.egg-info') + + setup( name=package_name, packages=find_packages(), @@ -57,5 +68,7 @@ setup( 'License :: OSI Approved :: MIT License', ], keywords='youtube livechat asyncio', - + cmdclass={ + 'clean': CleanCommand, + } ) \ No newline at end of file