paulbsd/matrix
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updated matrix program

Browse Source
This commit is contained in:
Paul 2020-04-25 22:57:20 +02:00
parent 2e09efb8d9
commit 583ee40431
1 changed files with 39 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
matrix.py
View File

@ -23,42 +23,50 @@ args = parser.parse_args()
mode = args.mode
backup_file = args.backup_file
directory = args.directory.rstrip("/")
directory = args.directory
def backup():
fd = open(backup_file,"w+")
i = 0
with open(backup_file, "w+") as file_desc:
result = []
for rd,dr,fs in os.walk(directory):
for d in dr:
res = os.path.join(rd, d)
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 fs:
res = os.path.join(rd, f)
for f in files:
res = os.path.join(root_dir, f)
result.append(res)
for f_name_tmp in result:
for f_name 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)
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
fd.close()
print("Backup {} files permissions, storing to {}".format(i, backup_file))
def restore():
fd = open(backup_file,"r+")
for f in fd:
i = 0
with open(backup_file, "r+") as file_desc:
for f in file_desc:
line = f.strip("\n")
f_name_tmp = line.split('|')[0]
f_name = "%s/%s" % (directory,f_name_tmp)
f_perm = line.split('|')[1]
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()