Encode and decode Themis config files

themisCfg.py

import zlib
from sys import argv


def decode(s):
    dec = zlib.decompress(s)
    res = dec.decode("utf-8")
    return res


def encode(s):
    enc = zlib.compress(s)
    return enc


if __name__ == "__main__":

    with open(argv[1], "rb") as fi:
        s = fi.read()

    if argv[1][-4:] == ".xml":
        out = encode(s)

        with open(argv[1][:-4], "wb") as fo:
            fo.write(out)

    else:
        out = decode(s)

        with open(argv[1] + ".xml", "w", encoding="utf-8") as fo:
            fo.write(out)

Credit to @TrungNguyen1909 for the idea.