compute_checksums.py

import os
import subprocess
import sys

def run_checksum(basedir, filename, output):
  args = ['xxhsum', filename]
  process = subprocess.Popen(args, stdout=output, close_fds=True, cwd=basedir)
  if process.wait() != 0:
    raise subprocess.CalledProcessError(process.returncode, args)

def main():
  # TODO: Replace with the actual basedir of the backup directory
  basedir = os.getcwd()
  checksums_path = os.path.join(basedir, 'CHECKSUMS')

  with open(checksums_path, 'wb') as output:
    for root, dirs, files in os.walk(basedir):
      for fname in files:
        cpath = os.path.join(root, fname)
        if cpath == checksums_path:
          continue

        rpath = os.path.relpath(cpath, basedir)
        run_checksum(basedir, rpath, output)

if __name__ == '__main__':
  sys.exit(main())