Update setup.py

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,
+    }
 )