Improve Travis deploy scripts

.travis.yml

@@ -45,14 +45,12 @@ script:
   - python ./azure_nosetests.py
 after_success:
   - bash <(curl -s https://codecov.io/bash)
-before_deploy:
-  - python ./before_deploy.py
 deploy:
   provider: pypi
   user: Laurent.Mazuel
   # password: use $PYPI_PASSWORD
   # server: $PYPI_SERVER to override PyPI (e.g. https://testpypi.python.org/)
-  distributions: check # Hack, packages will be created by "before deploy" script
+  distributions: travis_deploy # Personal step
   skip_upload_docs: true
   on:
     tags: true

build_package.py

@@ -8,6 +8,9 @@
 
 import argparse
 import os
+import sys
+import re
+from pathlib import Path
 from subprocess import check_call
 
 DEFAULT_DEST_FOLDER = "./dist"
@@ -17,11 +20,40 @@ def create_package(name, dest_folder=DEFAULT_DEST_FOLDER):
     check_call(['python', 'setup.py', 'bdist_wheel', '-d', dest_folder], cwd=absdirpath)
     check_call(['python', 'setup.py', "sdist", "--format", "zip", '-d', dest_folder], cwd=absdirpath)
 
+def travis_build_package():
+    """Assumed called on Travis, to prepare a package to be deployed
+
+    This method prints on stdout for Travis.
+    Return is obj to pass to sys.exit() directly
+    """
+    travis_tag = os.environ.get('TRAVIS_TAG')
+    if not travis_tag:
+        print("TRAVIS_TAG environment variable is not present")
+        return
+
+    matching = re.match(r"^(?P<name>azure[a-z\-]*)_(?P<version>[.0-9]+[rc0-9]*)$", travis_tag)
+    if not matching:
+        print("TRAVIS_TAG is not '<package_name> <version>' (tag is: {})".format(travis_tag))
+        return
+
+    name, version = matching.groups()
+    create_package(name, '../dist')
+
+    print("Produced:\n{}".format(list(Path('./dist').glob('*'))))
+
+    pattern = "*{}*".format(version)
+    packages = list(Path('./dist').glob(pattern))
+    if not packages:
+        return "Package version does not match tag {}, abort".format(version)
+    pypi_server = os.environ.get("PYPI_SERVER", "default PyPI server")
+    print("Package created as expected and will be pushed to {}".format(pypi_server))
+
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Build Azure package.')
     parser.add_argument('name', help='The package name')
     parser.add_argument('--dest', '-d', default=DEFAULT_DEST_FOLDER,
                         help='Destination folder. Relative to the package dir. [default: %(default)s]')
 
     args = parser.parse_args()
-    create_package(args.name, args.dest)
+    create_package(args.name, args.dest)

setup.py

@@ -12,6 +12,7 @@
 import copy
 import sys
 import runpy
+from build_package import travis_build_package
 
 root_folder = os.path.abspath(os.path.dirname(__file__))
 
@@ -36,6 +37,8 @@
 # Package final:
 if "install" in sys.argv:
     packages = content_package
+elif "travis_deploy" in sys.argv:
+    sys.exit(travis_build_package())
 else:
     packages = nspkg_packages + content_package + meta_package