100 lines
3.5 KiB
Python
Executable File
100 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# SPDX-FileCopyrightText: 2014 MicroPython & CircuitPython contributors (https://github.com/adafruit/circuitpython/graphs/contributors)
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import argparse
|
|
import os
|
|
import os.path
|
|
from pathlib import Path
|
|
import semver
|
|
import subprocess
|
|
|
|
# Compatible with Python 3.4 due to travis using trusty as default.
|
|
|
|
|
|
def version_string(path=None, *, valid_semver=False):
|
|
version = None
|
|
try:
|
|
tag = subprocess.check_output("git describe --tags --exact-match", shell=True, cwd=path)
|
|
version = tag.strip().decode("utf-8", "strict")
|
|
except subprocess.CalledProcessError:
|
|
describe = subprocess.check_output("git describe --tags", shell=True, cwd=path)
|
|
tag, additional_commits, commit_ish = (
|
|
describe.strip().decode("utf-8", "strict").rsplit("-", maxsplit=2)
|
|
)
|
|
commit_ish = commit_ish[1:]
|
|
if valid_semver:
|
|
version_info = semver.parse_version_info(tag)
|
|
if not version_info.prerelease:
|
|
version = (
|
|
semver.bump_patch(tag)
|
|
+ "-alpha.0.plus."
|
|
+ additional_commits
|
|
+ "+"
|
|
+ commit_ish
|
|
)
|
|
else:
|
|
version = tag + ".plus." + additional_commits + "+" + commit_ish
|
|
else:
|
|
version = commit_ish
|
|
return version
|
|
|
|
|
|
# Visit all the .py files in topdir. Replace any __version__ = "0.0.0-auto.0" type of info
|
|
# with actual version info derived from git.
|
|
def copy_and_process(in_dir, out_dir):
|
|
for root, subdirs, files in os.walk(in_dir):
|
|
# Skip library examples directory and subfolders.
|
|
relative_path_parts = Path(root).relative_to(in_dir).parts
|
|
if relative_path_parts and relative_path_parts[0] in [
|
|
"examples",
|
|
"docs",
|
|
"tests",
|
|
"utils",
|
|
]:
|
|
del subdirs[:]
|
|
continue
|
|
|
|
for file in files:
|
|
# Skip top-level setup.py (module install info) and conf.py (sphinx config),
|
|
# which are not part of the library
|
|
if (root == in_dir) and file in ("conf.py", "setup.py"):
|
|
continue
|
|
|
|
input_file_path = Path(root, file)
|
|
output_file_path = Path(out_dir, input_file_path.relative_to(in_dir))
|
|
|
|
if file.endswith(".py"):
|
|
if not output_file_path.parent.exists():
|
|
output_file_path.parent.mkdir(parents=True)
|
|
with input_file_path.open("r") as input, output_file_path.open("w") as output:
|
|
for line in input:
|
|
if line.startswith("__version__"):
|
|
module_version = version_string(root, valid_semver=True)
|
|
line = line.replace("0.0.0-auto.0", module_version)
|
|
output.write(line)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
argparser = argparse.ArgumentParser(
|
|
description="""\
|
|
Copy and pre-process .py files into output directory, before freezing.
|
|
1. Remove top-level repo directory.
|
|
2. Update __version__ info.
|
|
3. Remove examples.
|
|
4. Remove non-library setup.py and conf.py"""
|
|
)
|
|
argparser.add_argument(
|
|
"in_dirs",
|
|
metavar="input-dir",
|
|
nargs="+",
|
|
help="top-level code dirs (may be git repo dirs)",
|
|
)
|
|
argparser.add_argument("-o", "--out_dir", help="output directory")
|
|
args = argparser.parse_args()
|
|
|
|
for in_dir in args.in_dirs:
|
|
copy_and_process(in_dir, args.out_dir)
|