Add support to protect file name

This commit is contained in:
2023-02-16 13:31:31 +08:00
parent a779ca4b8d
commit ea645a8ec9
5 changed files with 100 additions and 8 deletions

View File

@@ -39,6 +39,8 @@ class RestoreTask(Thread):
nam = r.real_name
if f.encrypted:
src = join(self.cfg.dest, '.encrypt', prog, fn)
if not exists(src):
src = join(self.cfg.dest, '.encrypt', '.id', str(f.id)) # noqa: E501
else:
src = join(self.cfg.dest, prog, fn)
c = r.compress_config
@@ -78,6 +80,8 @@ class RestoreTask(Thread):
nam = r.real_name
if f.encrypted:
src = join(self.cfg.dest, '.encrypt', prog, fn + '.db')
if not exists(src):
src = join(self.cfg.dest, '.encrypt', '.id', str(f.id)) # noqa: E501
else:
src = join(self.cfg.dest, prog, fn + '.db')
c = r.compress_config