flt/tools
2
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

encrypt files

Browse Source
This commit is contained in:
flt6
2025-05-06 22:37:22 +08:00
parent 10d7a55454
commit 71c3e30267
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
encryptFiles/new_1.py
View File

@ -125,6 +125,8 @@ def decrypt_files():
with open(PASSWORD_FILE, "r") as f: with open(PASSWORD_FILE, "r") as f:
password_data = json.load(f) password_data = json.load(f)
suc=True
for encrypted_key_hex, encrypted_files in password_data.items(): for encrypted_key_hex, encrypted_files in password_data.items():
encrypted_aes_key = bytes.fromhex(encrypted_key_hex) encrypted_aes_key = bytes.fromhex(encrypted_key_hex)
aes_key = rsa_cipher.decrypt(encrypted_aes_key) aes_key = rsa_cipher.decrypt(encrypted_aes_key)
@ -136,6 +138,7 @@ def decrypt_files():
iv, ciphertext = f.read(16), f.read() iv, ciphertext = f.read(16), f.read()
except OSError as e: except OSError as e:
print(f"Cannot read file {file_path}: {e}") print(f"Cannot read file {file_path}: {e}")
suc=False
continue continue
except Exception as e: except Exception as e:
from traceback import print_exc from traceback import print_exc
@ -152,9 +155,11 @@ def decrypt_files():
f.write(plaintext) f.write(plaintext)
os.remove(file_path) os.remove(file_path)
if suc:
os.remove(PASSWORD_FILE) os.remove(PASSWORD_FILE)
print("Files decrypted.") print("Files decrypted.")
else:
print("Some files failed to decrypt.")
def main(): def main():
if not ENCRYPT_DIR.exists(): if not ENCRYPT_DIR.exists():