json_backup.py

import click
import os
import os.path
import pathlib as purepath
import shutil

@click.command(help="Clone the json files *only* out of a directory tree")
@click.argument('indir', type=click.STRING)
@click.argument('outdir', type=click.STRING)
@click.option('-X', '--exclude', type=click.STRING, help="Skip this file/directory", multiple=True)
@click.option('--extension', type=click.STRING, help="Use this extension instead of .json", default="json")
def json_backup(indir, outdir, exclude, extension):
	if not os.path.isdir(indir):
		raise click.ClickException(f"Input directory not a directory: '{indir}'")

	os.mkdir(outdir)
	exclude = [purepath.Path(x).resolve() for x in exclude]

	# Fixup if outdir is inside indir and not excluded
	outpath = purepath.Path(outdir).resolve()
	if outpath.is_relative_to(purepath.Path(indir).resolve()):
		exclude.append(outpath)

	for root, dirs, files in os.walk(indir, topdown=True):
		#print("VISIT", root)
		excluded = False
		for path in exclude:
			if purepath.Path(root).resolve().is_relative_to(path):
				#print("HIT EXCLUDE", path, root)
				dirs.clear() # Do not descend
				excluded = True
				break

		if not excluded:
			newroot = None
			for file in files:
				if file.endswith(extension):
					path = os.path.join(root, file)
					if not newroot:
						try:
							newroot = os.path.join(outdir, root)

							purepath.Path(newroot).mkdir(parents=True)
						except FileExistsError:
							pass
					newpath = os.path.join(newroot, file)
					print(f"...Copy '{path} to '{newpath}'")
					shutil.copy2(path, newpath)

json_backup()