mirror of
https://github.com/esphome/home-assistant-addon.git
synced 2024-11-07 09:44:39 +01:00
96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import re
|
||
|
|
import subprocess
|
||
|
|
from dataclasses import dataclass
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Version:
|
||
|
|
major: int
|
||
|
|
minor: int
|
||
|
|
patch: int
|
||
|
|
beta: int = 0
|
||
|
|
dev: bool = False
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return f'{self.major}.{self.minor}.{self.full_patch}'
|
||
|
|
|
||
|
|
@property
|
||
|
|
def full_patch(self):
|
||
|
|
res = f'{self.patch}'
|
||
|
|
if self.beta > 0:
|
||
|
|
res += f'b{self.beta}'
|
||
|
|
if self.dev:
|
||
|
|
res += '-dev'
|
||
|
|
return res
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def parse(cls, value):
|
||
|
|
match = re.match(r'(\d+).(\d+).(\d+)(b\d+)?(-dev)?', value)
|
||
|
|
assert match is not None
|
||
|
|
major = int(match[1])
|
||
|
|
minor = int(match[2])
|
||
|
|
patch = int(match[3])
|
||
|
|
beta = int(match[4][1:]) if match[4] else 0
|
||
|
|
dev = bool(match[5])
|
||
|
|
return Version(
|
||
|
|
major=major, minor=minor, patch=patch,
|
||
|
|
beta=beta, dev=dev
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def sub(path, pattern, repl, expected_count=1):
|
||
|
|
with open(path) as fh:
|
||
|
|
content = fh.read()
|
||
|
|
content, count = re.subn(pattern, repl, content, re.MULTILINE)
|
||
|
|
if expected_count is not None:
|
||
|
|
assert count == expected_count
|
||
|
|
with open(path, "wt") as fh:
|
||
|
|
fh.write(content)
|
||
|
|
|
||
|
|
|
||
|
|
def write_version(target: str, version: Version):
|
||
|
|
# version: '1.14.5' # BETA
|
||
|
|
# version: '1.14.5' # STABLE
|
||
|
|
sub(
|
||
|
|
'template/config.yaml',
|
||
|
|
r" version: '[^']+' # {}".format(target.upper()),
|
||
|
|
f" version: '{version}' # {target.upper()}"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('new_version', type=str)
|
||
|
|
parser.add_argument('--beta', action='store_true', help="Update the beta addon")
|
||
|
|
parser.add_argument('--stable', action='store_true', help="Update the stable addon")
|
||
|
|
parser.add_argument('--commit', action='store_true')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if not any([args.beta, args.stable]):
|
||
|
|
print("At least one of beta or stable has to be bumped")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
if args.commit and subprocess.call(["git", "diff", "--quiet"]) == 1:
|
||
|
|
print("Cannot use --commit because git is dirty.")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
version = Version.parse(args.new_version)
|
||
|
|
print(f"Bumping to {version}")
|
||
|
|
if args.beta:
|
||
|
|
write_version('beta', version)
|
||
|
|
if args.stable:
|
||
|
|
assert not version.beta
|
||
|
|
write_version('stable', version)
|
||
|
|
|
||
|
|
if args.commit:
|
||
|
|
subprocess.check_call(["git", "commit", "-nam", f"Bump version to v{version}"])
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main() or 0)
|