added install_entry to fix setup cache data bug

Author: lakshmanaram

ACedIt/install_entry.py

@@ -0,0 +1,25 @@
+import json
+import os
+from setuptools.command.install import install
+
+class InstallEntry(install):
+
+    def run(self):
+
+		default_site = 'codeforces'
+		workdir = os.path.join(os.path.expanduser('~'), 'ACedIt')
+		cache_dir = os.path.join(os.path.expanduser('~'), '.cache', 'ACedIt')
+
+		from main import supported_sites
+
+		for site in supported_sites:
+			# create cache directory and working directory structure
+		    if not os.path.isdir(os.path.join(cache_dir, site)):
+		        os.makedirs(os.path.join(cache_dir, site))
+		    if not os.path.isdir(os.path.join(workdir, site)):
+		    	os.makedirs(os.path.join(workdir, site))
+
+		data = {'default_site': default_site.strip(), 'workdir': os.path.expanduser('~'),
+		        'cachedir': cache_dir}
+		with open(os.path.join(cache_dir, 'constants.json'), 'w') as f:
+		    f.write(json.dumps(data, indent=2))

setup.py

@@ -4,8 +4,8 @@
     from distutils.core import setup
 
 import sys
-import json
-import os
+
+from ACedIt.install_entry import InstallEntry
 
 with open('requirements.txt', 'r') as f:
     requirements = [line.strip() for line in f.readlines()]
@@ -14,24 +14,6 @@
 if sys.version_info >= (3,):
     extra['use_2to3'] = True
 
-default_site = 'codeforces'
-workdir = os.path.join(os.path.expanduser('~'), 'ACedIt')
-cache_dir = os.path.join(os.path.expanduser('~'), '.cache', 'ACedIt')
-
-from ACedIt.main import supported_sites
-
-for site in supported_sites:
-	# create cache directory and working directory structure
-    if not os.path.isdir(os.path.join(cache_dir, site)):
-        os.makedirs(os.path.join(cache_dir, site))
-    if not os.path.isdir(os.path.join(workdir, site)):
-    	os.makedirs(os.path.join(workdir, site))
-
-data = {'default_site': default_site.strip(), 'workdir': os.path.expanduser('~'),
-        'cachedir': cache_dir}
-with open(os.path.join(cache_dir, 'constants.json'), 'w') as f:
-    f.write(json.dumps(data, indent=2))
-
 setup(
     name='ACedIt',
 
@@ -57,6 +39,11 @@
 
     keywords=['sample test cases', 'downloader', 'competitive programming'],
 
+    cmdclass={
+            'install': InstallEntry,
+            'develop': InstallEntry,
+    },
+
     classifiers=[
         'Operating System :: POSIX :: Linux',
         'License :: OSI Approved :: MIT License',