added updated_conda_forge_package function

alixdamman · alixdamman · commit c7edf0ead7c3 · 2017-10-18T17:34:37.000+02:00
diff --git a/make_release.py b/make_release.py
@@ -14,6 +14,8 @@
 import subprocess
 import sys
 import zipfile
+import hashlib
+import urllib.request as request
 
 from datetime import date
 from os import chdir, makedirs
@@ -24,7 +26,9 @@
 
 PY2 = sys.version_info[0] < 3
 TMP_PATH = r"c:\tmp\larray_new_release"
-
+TMP_PATH_CONDA = r"c:\tmp\larray_conda_new_release"
+LARRAY_REP = 'https://github.com/liam2/larray'
+CONDA_LARRAY_FEEDSTOCK_REP = 'https://github.com/larray-project/larray-feedstock.git'
 
 try:
     input = raw_input
@@ -411,6 +415,47 @@ def update_changelog(context):
     call('git commit -m "update release date in changes.rst" {}'.format(fpath))
 
 
+def update_conda_forge_package(context):
+    create_tmp_directory(context)
+
+    try:
+        # compute sha256
+        url = LARRAY_REP + '/archive/{version}.tar.gz'.format(version=short(context['release_name']))
+        print('Computing SHA256 from archive {}'.format(url), end=' ')
+        with request.urlopen(url) as response:
+            sha256 = hashlib.sha256(response.read()).hexdigest()
+            print('done.')
+            print('SHA256: ', sha256)
+
+        # clone repository
+        clone_repository(context)
+
+        chdir(context['build_dir'])
+
+        # set version and sha256 in meta.yml file
+        meta_file = r'recipe\meta.yaml'
+        with open(meta_file, 'r') as f:
+            lines = open(meta_file, 'r').readlines()
+        lines[1] = '{{% set version = "{release_name}" %}}\n'.format(**context)
+        lines[2] = '{{% set sha256 = "{sha256}" %}}\n'.format(sha256=sha256)
+        with open(meta_file, 'w') as f:
+            f.writelines(lines)
+        # add, commit and push
+        print(call('git status -s'))
+        print(call('git diff {meta_file}'.format(meta_file=meta_file)))
+        if no('Does that last changes look right?'):
+            exit(1)
+        do('Adding', call, 'git add {}'.format(meta_file))
+        do('Commiting', call, 'git commit -m "bump to version {release_name}"'.format(**context))
+        print(call('git log -1'))
+        do('Pushing to GitHub', call, 'git push origin {branch}'.format(**context))
+        # delete temporary directory
+        cleanup(context)
+    except Exception as e:
+        cleanup(context)
+        raise e
+
+
 def build_doc(context):
     chdir(context['build_dir'])
     chdir('doc')
@@ -480,6 +525,7 @@ def cleanup(context):
     (create_archives, 'Creating archives'),
     (final_confirmation, ''),
     (tag_release, 'Tagging release'),
+    (update_conda_forge_package, ''),
     # We used to push from /tmp to the local repository but you cannot push
     # to the currently checked out branch of a repository, so we need to
     # pull changes instead. However pull (or merge) add changes to the
@@ -530,20 +576,27 @@ def make_release(release_name='dev', steps=':', branch='master'):
                'tmp_dir': TMP_PATH,
                'build_dir': os.path.join(TMP_PATH, 'build'),
                'public_release': public_release}
+    context_conda = {'branch': branch,
+                     'release_name': release_name,
+                     'rev': rev,
+                     'repository': CONDA_LARRAY_FEEDSTOCK_REP,
+                     'tmp_dir': TMP_PATH_CONDA,
+                     'build_dir': os.path.join(TMP_PATH, 'build'),
+                     'public_release': public_release}
+
     for step_func, step_desc in steps_funcs[start:stop]:
+        ctx = context_conda if step_func == update_conda_forge_package else context
         if step_desc:
-            do(step_desc, step_func, context)
+            do(step_desc, step_func, ctx)
         else:
-            step_func(context)
+            step_func(ctx)
 
 
 if __name__ == '__main__':
     argv = sys.argv
-    # if len(argv) < 2:
-    #     print("Usage: {} release_name|dev [step|startstep:stopstep] [branch]".format(argv[0]))
-    #     print("steps:", ', '.join(f.__name__ for f, _ in steps_funcs))
-    #     sys.exit()
-    #
-    # make_release(*argv[1:])
-
-    run_tests()
+    if len(argv) < 2:
+        print("Usage: {} release_name|dev [step|startstep:stopstep] [branch]".format(argv[0]))
+        print("steps:", ', '.join(f.__name__ for f, _ in steps_funcs))
+        sys.exit()
+
+    make_release(*argv[1:])
