Merge pull request #11 from dipcode-software/feat/md-to-rst

refactor to setup.py to accept a publish

Author: srtab

setup.py

@@ -1,17 +1,85 @@
-from setuptools import find_packages, setup
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+import io
+import os
+from shutil import rmtree
+import sys
+
+from setuptools import Command, find_packages, setup
+
+
+# Package meta-data.
+VERSION = __import__("obfuscator").__version__
+NAME = 'django-obfuscate'
+DESCRIPTION = 'Django app to obfuscate data. '
+URL = 'https://github.com/dipcode-software/django-obfuscator'
+EMAIL = '[email protected]'
+AUTHOR = 'Dipcode'
+
+
+here = os.path.abspath(os.path.dirname(__file__))
+with io.open(os.path.join(here, 'README.rst'), encoding='utf-8') as f:
+    long_description = '\n' + f.read()
+
+
+class PublishCommand(Command):
+    """Support setup.py publish."""
+
+    description = 'Build and publish the package.'
+    user_options = []
+    environment = None
+
+    @staticmethod
+    def status(s):
+        """Prints things in bold."""
+        print('\033[1m{0}\033[0m'.format(s))
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        try:
+            self.status('Removing previous builds…')
+            rmtree(os.path.join(here, 'dist'))
+            rmtree(os.path.join(here, 'build'))
+        except OSError:
+            pass
+
+        self.status('Building Source and Wheel distribution…')
+        os.system(
+            '{0} setup.py sdist bdist_wheel '.format(sys.executable)
+        )
+
+        self.status('Uploading the package to {env} via Twine…'\
+            .format(env=self.environment))
+        os.system('twine upload --repository {env} dist/*'.format(env=self.environment))
+
+        sys.exit()
+
+
+class ProductionPublishCommand(PublishCommand):
+    environment = 'pypi'
+
+
+class DevelopmentPublishCommand(PublishCommand):
+    environment = 'testpypi'
 
 
 setup(
-    name='django-obfuscate',
-    version=__import__("obfuscator").__version__,
+    name=NAME,
+    version=VERSION,
+    description=DESCRIPTION,
+    long_description=long_description,
+    author=AUTHOR,
+    author_email=EMAIL,
+    url=URL,
     packages=find_packages(),
     include_package_data=True,
     license='MIT',
-    description='Django app to obfuscate data.',
-    url='https://github.com/dipcode-software/django-obfuscator',
-    author='Dipcode',
-    author_email='[email protected]',
-    keywords=['django', 'django-models', 'models', 'obfuscator'],
     classifiers=[
         'Environment :: Web Environment',
         'Framework :: Django',
@@ -22,6 +90,10 @@
         'Programming Language :: Python',
         'Programming Language :: Python :: 2.7',
     ],
+    cmdclass={
+        'publish_dev': DevelopmentPublishCommand,
+        'publish_prod': ProductionPublishCommand,
+    },
     install_requires=[
         'setuptools-git >= 1.2'
     ]