|
| 1 | +#!/usr/bin/env python3 |
| 2 | +importargparse |
| 3 | +importos |
| 4 | +importos.path |
| 5 | +frompathlibimportPath |
| 6 | +importsemver |
| 7 | +importsubprocess |
| 8 | + |
| 9 | +defversion_string(path=None,*,valid_semver=False): |
| 10 | +version=None |
| 11 | +tag=subprocess.run('git describe --tags --exact-match',shell=True, |
| 12 | +stdout=subprocess.PIPE,stderr=subprocess.PIPE,cwd=path) |
| 13 | +iftag.returncode==0: |
| 14 | +version=tag.stdout.strip().decode("utf-8","strict") |
| 15 | +else: |
| 16 | +describe=subprocess.run("git describe --tags",shell=True,stdout=subprocess.PIPE,cwd=path) |
| 17 | +tag,additional_commits,commitish=describe.stdout.strip().decode("utf-8","strict").rsplit("-",maxsplit=2) |
| 18 | +commitish=commitish[1:] |
| 19 | +ifvalid_semver: |
| 20 | +version_info=semver.parse_version_info(tag) |
| 21 | +ifnotversion_info.prerelease: |
| 22 | +version=semver.bump_patch(tag)+"-alpha.0.plus."+additional_commits+"+"+commitish |
| 23 | +else: |
| 24 | +version=tag+".plus."+additional_commits+"+"+commitish |
| 25 | +else: |
| 26 | +version=commitish |
| 27 | +returnversion |
| 28 | + |
| 29 | +# Visit all the .py files in topdir. Replace any __version__ = "0.0.0-auto.0" type of info |
| 30 | +# with actual version info derived from git. |
| 31 | +defcopy_and_process(in_dir,out_dir): |
| 32 | +forroot,subdirs,filesinos.walk(in_dir): |
| 33 | + |
| 34 | +# Skip library examples directories. |
| 35 | +ifPath(root).name=='examples': |
| 36 | +continue |
| 37 | + |
| 38 | +forfileinfiles: |
| 39 | +# Skip top-level setup.py (module install info) and conf.py (sphinx config), |
| 40 | +# which are not part of the library |
| 41 | +if (root==in_dir)andfilein ('conf.py','setup.py'): |
| 42 | +continue |
| 43 | + |
| 44 | +input_file_path=Path(root,file) |
| 45 | +output_file_path=Path(out_dir,input_file_path.relative_to(in_dir)) |
| 46 | + |
| 47 | +iffile.endswith(".py"): |
| 48 | +output_file_path.parent.mkdir(parents=True,exist_ok=True) |
| 49 | +withinput_file_path.open("r")asinput,output_file_path.open("w")asoutput: |
| 50 | +forlineininput: |
| 51 | +ifline.startswith("__version__"): |
| 52 | +module_version=version_string(root,valid_semver=True) |
| 53 | +line=line.replace("0.0.0-auto.0",module_version) |
| 54 | +output.write(line) |
| 55 | + |
| 56 | +if__name__=='__main__': |
| 57 | +argparser=argparse.ArgumentParser(description="""\ |
| 58 | + Copy and pre-process .py files into output directory, before freezing. |
| 59 | + 1. Remove top-level repo directory. |
| 60 | + 2. Update __version__ info. |
| 61 | + 3. Remove examples. |
| 62 | + 4. Remove non-library setup.py and conf.py""") |
| 63 | +argparser.add_argument("in_dirs",metavar="input-dir",nargs="+", |
| 64 | +help="top-level code dirs (may be git repo dirs)") |
| 65 | +argparser.add_argument("-o","--out_dir",help="output directory") |
| 66 | +args=argparser.parse_args() |
| 67 | + |
| 68 | +forin_dirinargs.in_dirs: |
| 69 | +copy_and_process(in_dir,args.out_dir) |