contrib/client-side/svnpatch.py

#!/usr/bin/env python

# svnpatch.py - svnpatch helper script
# Author: Arfrever Frehtes Taifersar Arahesis
# License: GPL-3

import base64
import sys
import textwrap
import zlib

svnpatch1_block_start = b"========================= SVNPATCH1 BLOCK =========================\n"

def encode(input, output_file = sys.stdout):
	output = svnpatch1_block_start.decode()
	output += "\n".join(textwrap.wrap(base64.encodestring(zlib.compress(b"".join([x for x in input]).rstrip(b"\n"))).decode(), 76))
	output_file.write(output)

def decode(input, output_file = sys.stdout):
	svnpatch1_block_start_index = input.index(svnpatch1_block_start)
	svnpatch1_block = input[svnpatch1_block_start_index+1:]
	output = zlib.decompress(base64.decodestring(b"".join([x.rstrip(b"\n") for x in svnpatch1_block])))
	if sys.version_info[0] >= 3:
		output_file.buffer.write(output)
	else:
		output_file.write(output)

def help():
	print("svnpatch.py - svnpatch helper script")
	print("Usage: svnpatch.py [-e | --encode | -d | --decode] FILE")
	print("       svnpatch.py [-e | --encode | -d | --decode] -")
	print("       svnpatch.py [-h | --help]")
	print("")
	print("Author: Arfrever Frehtes Taifersar Arahesis")
	print("License: GPL-3")
	exit(0)

if len(sys.argv) == 2 and sys.argv[1] in ("-h", "--help"):
	help()

elif len(sys.argv) < 3:
	sys.stderr.write("svnpatch.py: Missing arguments\n")
	exit(1)

elif len(sys.argv) > 3:
	sys.stderr.write("svnpatch.py: Excessive argument(s)\n")
	exit(1)

if sys.argv[1] in ("-e", "--encode"):
	func = encode
elif sys.argv[1] in ("-d", "--decode"):
	func = decode
else:
	sys.stderr.write("Incorrect option\n")
	exit(1)

if sys.argv[2] == "-":
	if sys.version_info[0] >= 3:
		lines = sys.stdin.buffer.readlines()
	else:
		lines = sys.stdin.readlines()
else:
	lines = open(sys.argv[2], "rb").readlines()

func(lines)
print("")