Add setup.py commands for building and publishing (#31)

Author: cancan101

setup.py

@@ -1,14 +1,95 @@
-from setuptools import find_packages, setup
+import os
+import sys
+from pathlib import Path
+from shutil import rmtree
+from typing import List, Tuple
+
+from setuptools import Command, find_packages, setup
+
+# -----------------------------------------------------------------------------
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+# read the contents of your README file
+this_directory = Path(__file__).parent
+long_description = (this_directory / "README.md").read_text()
+long_description_content_type = "text/markdown; charset=UTF-8; variant=GFM"
+
+# -----------------------------------------------------------------------------
+
+
+class BaseCommand(Command):
+    user_options: List[Tuple[str, str, str]] = []
+
+    @staticmethod
+    def status(s: str) -> None:
+        """Prints things in bold."""
+        print("\033[1m{0}\033[0m".format(s))  # noqa: T001
+
+    def system(self, command: str) -> None:
+        os.system(command)  # noqa: S605
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+
+class BuildCommand(BaseCommand):
+    """Support setup.py building."""
+
+    description = "Build the package."
+
+    def run(self):
+        try:
+            self.status("Removing previous builds…")
+            rmtree(os.path.join(here, "dist"))
+        except OSError:
+            pass
+
+        self.status("Building Source and Wheel (universal) distribution…")
+        self.system("{0} -m build --sdist --wheel .".format(sys.executable))
+
+        self.status("Checking wheel contents…")
+        self.system("check-wheel-contents dist/*.whl")
+
+        self.status("Running twine check…")
+        self.system("{0} -m twine check dist/*".format(sys.executable))
+
+
+class UploadTestCommand(BaseCommand):
+    """Support uploading to test PyPI."""
+
+    description = "Upload the package to the test PyPI."
+
+    def run(self):
+        self.status("Uploading the package to PyPi via Twine…")
+        self.system(
+            "twine upload --repository-url https://test.pypi.org/legacy/ dist/*"
+        )
+
+
+class UploadCommand(BaseCommand):
+    """Support uploading to PyPI."""
+
+    description = "Upload the package to PyPI."
+
+    def run(self):
+        self.status("Uploading the package to PyPi via Twine…")
+        self.system("twine upload dist/*")
+
 
 # -----------------------------------------------------------------------------
 
 DESCRIPTION = "Python DB-API and SQLAlchemy interface for Airtable."
 
 setup(
     name="sqlalchemy-airtable",
-    version="0.0.1.dev0",
+    version="0.0.1.dev1",
     description=DESCRIPTION,
-    long_description=DESCRIPTION,
+    long_description=long_description,
+    long_description_content_type=long_description_content_type,
     author="Alex Rothberg",
     author_email="[email protected]",
     url="https://github.com/cancan101/airtable-db-api",
@@ -45,6 +126,8 @@
     ],
     # $ setup.py publish support.
     cmdclass={
-        # 'upload': UploadCommand,
+        "buildit": BuildCommand,
+        "uploadtest": UploadTestCommand,
+        "upload": UploadCommand,
     },
 )