65 lines
1.7 KiB
Python
Executable File
65 lines
1.7 KiB
Python
Executable File
#!/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.rstrip("/")
|
|
|
|
def backup():
|
|
fd = open(backup_file,"w+")
|
|
result=[]
|
|
for rd,dr,fs in os.walk(directory):
|
|
for d in dr:
|
|
res = os.path.join(rd, d)
|
|
result.append(res)
|
|
for f in fs:
|
|
res = os.path.join(rd, f)
|
|
result.append(res)
|
|
for f_name_tmp in result:
|
|
try:
|
|
f_name = f_name_tmp.replace(directory,'').lstrip("/")
|
|
f_perm = str(oct(os.stat(f_name_tmp).st_mode))
|
|
line = "%s|%s\n" % (f_name,f_perm)
|
|
fd.write(line)
|
|
except OSError:
|
|
pass
|
|
fd.close()
|
|
|
|
def restore():
|
|
fd = open(backup_file,"r+")
|
|
for f in fd:
|
|
line = f.strip("\n")
|
|
f_name_tmp = line.split('|')[0]
|
|
f_name = "%s/%s" % (directory,f_name_tmp)
|
|
f_perm = line.split('|')[1]
|
|
try:
|
|
os.chmod(f_name,int(f_perm,8))
|
|
except OSError:
|
|
pass
|
|
|
|
if mode == "backup":
|
|
backup()
|
|
elif mode == "restore":
|
|
restore()
|
|
|