lionel/cryptAES
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

upload/download in class

Browse Source
This commit is contained in:
toto 2021-11-02 14:43:45 +01:00
parent 1cc0e98de3
commit f84e226770
1 changed files with 20 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
backup.py
View File

@ -59,16 +59,16 @@ class Backup:
c_date = datetime.fromtimestamp(os.path.getctime(uri)).strftime("%Y-%m-%d %H:%M:%S.%f") c_date = datetime.fromtimestamp(os.path.getctime(uri)).strftime("%Y-%m-%d %H:%M:%S.%f")
if size > tarball_size: if size > tarball_size:
crypt_id = self.bdd.add([{'name': f, crypt_id = self.bdd.add([{'name': f,
'path': pathlib.Path(uri).as_posix(), 'path': pathlib.Path(uri).as_posix(),
'size': size, 'size': size,
'm_date': m_date, 'm_date': m_date,
'c_date': c_date}]) 'c_date': c_date}])
if crypt_id is not None: if crypt_id is not None:
print("Proceed", uri, ' ==> ', crypt_id) print("Proceed", uri, ' ==> ', crypt_id)
enc = crypt(compress(uri), self.key) enc = crypt(compress(uri), self.key)
self.bdd.set_crypt_attr(crypt_id, compress_mode="gz", sha1sum=get_hash(enc)) self.bdd.set_crypt_attr(crypt_id, compress_mode="gz", sha1sum=get_hash(enc))
print(" Size : ", get_size(enc)) print(" Size : ", get_size(enc))
upload_file(enc, self.save_mode, file_name=os.path.join(self.save_location, str(crypt_id).zfill(ZFILL)), bucket=self.buk) self.upload_file(enc, file_name=str(crypt_id).zfill(ZFILL))
else: else:
files.append({'name': f, files.append({'name': f,
'path': pathlib.Path(uri).as_posix(), 'path': pathlib.Path(uri).as_posix(),
@ -85,13 +85,13 @@ class Backup:
enc = crypt(compress(tarball), self.key) enc = crypt(compress(tarball), self.key)
self.bdd.set_crypt_attr(crypt_id, compress_mode="tar.gz", sha1sum=get_hash(enc)) self.bdd.set_crypt_attr(crypt_id, compress_mode="tar.gz", sha1sum=get_hash(enc))
print(" Size : ", get_size(enc)) print(" Size : ", get_size(enc))
upload_file(enc, self.save_mode, file_name=os.path.join(self.save_location, str(crypt_id).zfill(ZFILL)), bucket=self.buk) self.upload_file(enc, file_name=str(crypt_id).zfill(ZFILL))
def recover_file(self, paths, parents=False, save_path=os.getcwd()): def recover_file(self, paths, parents=False, save_path=os.getcwd()):
files = self.bdd.get_crypt_name(paths) files = self.bdd.get_crypt_name(paths)
for file in files: for file in files:
if file['crypt_id'] is not None: if file['crypt_id'] is not None:
encrypted_file = download_file(self.save_mode, os.path.join(self.save_location, str(file['crypt_id']).zfill(ZFILL)), bucket=self.buk) encrypted_file = self.download_file(str(file['crypt_id']).zfill(ZFILL))
file_hash = get_hash(encrypted_file) file_hash = get_hash(encrypted_file)
if file['sha1sum'] == file_hash: if file['sha1sum'] == file_hash:
if parents: if parents:
@ -103,22 +103,20 @@ class Backup:
print("{} {}".format(file_hash, "File")) print("{} {}".format(file_hash, "File"))
print() print()
def upload_file(self, file, file_name):
if self.save_mode == BACKBLAZE:
self.buk.upload_bytes(file.read(), file_name)
elif self.save_mode == LOCAL:
save(file, os.path.join(self.save_location, file_name))
def upload_file(file, mode, file_name, bucket=None): def download_file(self, file):
if mode == BACKBLAZE: dl = tempfile.SpooledTemporaryFile()
bucket.upload_bytes(file.read(), file_name) if self.save_mode == BACKBLAZE:
elif mode == LOCAL: self.buk.download_file_by_name(file).save(dl)
save(file, file_name) elif self.save_mode == LOCAL:
with open(os.path.join(self.save_location, file), 'rb') as infile:
dl.write(infile.read())
def download_file(mode, file, bucket=None): return dl
dl = tempfile.SpooledTemporaryFile()
if mode == BACKBLAZE:
bucket.download_file_by_name(file).save(dl)
elif mode == LOCAL:
with open(file, 'rb') as infile:
dl.write(infile.read())
return dl
def get_size(in_file): def get_size(in_file):