docker-unifi/.updater/updategit.py

#!/usr/bin/python3

import fileinput
import re
import subprocess
import sys
from os import close, remove, path
from shutil import move
from tempfile import mkstemp

assert path.isdir('.git'), "No git dir found"


def replace(file_path, pattern, subst):
    # From http://stackoverflow.com/a/39110
    fh, abs_path = mkstemp()
    with open(abs_path, 'w') as new_file:
        with open(file_path) as old_file:
            for line in old_file:
                new_file.write(re.sub(pattern, subst, line))
    close(fh)
    remove(file_path)
    move(abs_path, file_path)


def call(args):
    print('>>> %s' % ' '.join(args))
    return subprocess.call(args)

call(['git', 'fetch'])

updated = False
for line in fileinput.input():
    branch, version, changelog, url = line.strip().split(' ')
    branch = 'master' if branch == 'latest' else branch
    call(['git', 'branch', branch, 'master']) # Will fail if the branch exists
    call(['git', 'checkout', branch])
    call(['git', 'reset', '--hard', 'origin/%s' % branch]) # May fail as well
    replace('Dockerfile', r'^ARG url=.*', 'ARG url=%s' % url)
    c = call(['git', 'commit', '-am', 'Auto update to version %s.\n\nChangelog: %s' % (version, changelog)])
    if c == 0:
      call(['git', 'push', 'origin', branch])
      updated = True

call(['git', 'checkout', 'master'])

sys.exit(0 if updated else 1)