cutthecord/patches/branding/addpatch.py

#!/bin/env python3
import os
import sys

fname = sys.argv[1]
persistdir = "/home/ave/distokrepos/versionlogs/"


def counterup():
    # HACKY
    cfname = persistdir + fname.split("/")[-1].replace(".patch", "")
    if os.path.isfile(fname):
        with open(cfname, 'r') as file:
            countdata = file.read()
    else:
        countdata = 0

    countdata = int(countdata) + 1

    with open(cfname, 'w') as file:
        file.write(str(countdata))

    return countdata


# Read in the file
with open(fname, 'r') as file:
    filedata = file.read()

buildnum = counterup()

name = "CutTheCord"
branch = "base"

if "customtheme" in sys.argv:
    branch = "themed"
    # name += " Themed"

# Replace the target string
filedata = filedata.replace("CTCBRANCH", branch)\
    .replace("CTCBUILD", buildnum)\
    .replace("CTCNAME", name)

# Write the file out again
with open(fname.replace(".patch", "-custom.patch"), 'w') as file:
    file.write(filedata)