matrix/matrix.py

#!/usr/bin/env python3

# Copyright (c) 2018, Paul Lecuq
# All rights reserved.
# BSD Licensed

# Matrix is a small program that saves permissions on directory to a file and can restore it
# on another directory (or another machine)

import os
import argparse

mode = ""
backup_file = ""
rep = ""

parser = argparse.ArgumentParser(description='Parse arguments')
parser.add_argument('mode', metavar='mode', type=str, choices=['backup', 'restore'], help='Mode')
parser.add_argument('backup_file', metavar='backup_file', type=str, help='Backup file')
parser.add_argument('directory', metavar='directory', type=str, help='Directory')

args = parser.parse_args()

mode = args.mode
backup_file = args.backup_file
directory = args.directory

def backup():
    i = 0
    with open(backup_file, "w+") as file_desc:
        result = []
        for root_dir, dirs, files in os.walk(directory):
            for d in dirs:
                res = os.path.join(root_dir, d)
                result.append(res)
            for f in files:
                res = os.path.join(root_dir, f)
                result.append(res)
        for f_name in result:
            try:
                i += 1
                f_uid = os.stat(f_name).st_uid
                f_gid = os.stat(f_name).st_gid
                f_perm = str(oct(os.stat(f_name).st_mode))
                line = "{}|{}|{}|{}\n".format(f_name, f_uid, f_gid, f_perm)
                file_desc.write(line)
            except OSError:
                pass
        print("Backup {} files permissions, storing to {}".format(i, backup_file))

def restore():
    i = 0
    with open(backup_file, "r+") as file_desc:
        for f in file_desc:
            line = f.strip("\n")
            f_name = line.split('|')[0]
            f_uid = line.split('|')[1]
            f_gid = line.split('|')[2]
            f_perm = line.split('|')[3]
            try:
                os.chmod(f_name, int(f_perm, 8))
                os.chown(f_name, int(f_uid), int(f_gid))
            except OSError:
                pass
            i += 1

    print("Restored {} files permissions".format(i))

if mode == "backup":
    backup()
elif mode == "restore":
    restore()