update setup.py from:

https://gist.github.com/jedie/e356a0c17ac1a2fcb716

Author: jedie

setup.py

@@ -21,6 +21,22 @@
 
 
 if "publish" in sys.argv:
+    """
+    Build and upload to PyPi, if...
+        ... __version__ doesn't contains "dev"
+        ... we are on git 'master' branch
+        ... git repository is 'clean' (no changed files)
+
+    Upload with "twine", git tag the current version and git push --tag
+
+    The cli arguments will be pass to 'twine'. So this is possible:
+     * Display 'twine' help page...: ./setup.py publish --help
+     * use testpypi................: ./setup.py publish --repository=test
+    """
+    # Imports here, so it's easier to copy&paste this complete code block ;)
+    import subprocess
+    import shutil
+
     try:
         # Test if wheel is installed, otherwise the user will only see:
         #   error: invalid command 'bdist_wheel'
@@ -33,23 +49,39 @@
         print("    ~/your/env/$ pip install wheel")
         sys.exit(-1)
 
-    import subprocess
+    try:
+        import twine
+    except ImportError as err:
+        print("\nError: %s" % err)
+        print("\nMaybe https://pypi.python.org/pypi/twine is not installed or virtualenv not activated?!?")
+        print("e.g.:")
+        print("    ~/your/env/$ source bin/activate")
+        print("    ~/your/env/$ pip install twine")
+        sys.exit(-1)
 
     def verbose_check_output(*args):
-        print("\nCall: %r\n" %  " ".join(args))
+        """ 'verbose' version of subprocess.check_output() """
+        call_info = "Call: %r" % " ".join(args)
         try:
-            return subprocess.check_output(args, universal_newlines=True)
+            output = subprocess.check_output(args, universal_newlines=True, stderr=subprocess.STDOUT)
         except subprocess.CalledProcessError as err:
             print("\n***ERROR:")
             print(err.output)
             raise
+        return call_info, output
 
     def verbose_check_call(*args):
-        print("\nCall: %r\n" %  " ".join(args))
+        """ 'verbose' version of subprocess.check_call() """
+        print("\tCall: %r\n" % " ".join(args))
         subprocess.check_call(args, universal_newlines=True)
 
-    # Check if we are on 'master' branch:
-    output = verbose_check_output("git", "branch", "--no-color")
+    if "dev" in __version__:
+        print("\nERROR: Version contains 'dev': v%s\n" % __version__)
+        sys.exit(-1)
+
+    print("\nCheck if we are on 'master' branch:")
+    call_info, output = verbose_check_output("git", "branch", "--no-color")
+    print("\t%s" % call_info)
     if "* master" in output:
         print("OK")
     else:
@@ -59,22 +91,48 @@ def verbose_check_call(*args):
             print("Bye.")
             sys.exit(-1)
 
-    # publish only if git repro is clean:
-    output = verbose_check_output("git", "status", "--porcelain")
+    print("\ncheck if if git repro is clean:")
+    call_info, output = verbose_check_output("git", "status", "--porcelain")
+    print("\t%s" % call_info)
     if output == "":
         print("OK")
     else:
-        print("\n***ERROR: git repro not clean:")
+        print("\n *** ERROR: git repro not clean:")
         print(output)
         sys.exit(-1)
 
-    # tag first (will raise a error of tag already exists)
+    print("\ngit tag version (will raise a error of tag already exists)")
     verbose_check_call("git", "tag", "v%s" % __version__)
 
-    # build and upload to PyPi:
-    verbose_check_call(sys.executable or "python", "setup.py", "sdist", "bdist_wheel", "upload")
-
-    # push
+    print("\nCleanup old builds:")
+    def rmtree(path):
+        path = os.path.abspath(path)
+        print("\tremove tree:", path)
+        shutil.rmtree(path)
+    rmtree("./dist")
+    rmtree("./build")
+
+    print("\nbuild but don't upload...")
+    log_filename="build.log"
+    with open(log_filename, "a") as log:
+        call_info, output = verbose_check_output(
+            sys.executable or "python",
+            "setup.py", "sdist", "bdist_wheel", "bdist_egg"
+        )
+        print("\t%s" % call_info)
+        log.write(call_info)
+        log.write(output)
+    print("Build output is in log file: %r" % log_filename)
+
+    print("\nUpload with twine:")
+    twine_args = sys.argv[1:]
+    twine_args.remove("publish")
+    twine_args.insert(1, "dist/*")
+    print("\ttwine upload command args: %r" % " ".join(twine_args))
+    from twine.commands.upload import main as twine_upload
+    twine_upload(twine_args)
+
+    print("\ngit push to server")
     verbose_check_call("git", "push")
     verbose_check_call("git", "push", "--tags")