写一个脚本,使用指定秘钥为文本文件内容加密和解密
时间: 2024-04-04 12:35:43 浏览: 14
您可以使用Python语言编写脚本来使用指定秘钥为文本文件内容进行加密和解密。下面是一个示例脚本:
```python
import cryptography.fernet
import argparse
import os
# 定义命令行参数
parser = argparse.ArgumentParser(description='Encrypt/Decrypt a file with a given key.')
parser.add_argument('filename', help='Name of the file to encrypt/decrypt.')
parser.add_argument('key', help='Encryption/Decryption key.')
# 解析命令行参数
args = parser.parse_args()
# 获取文件名和密钥
filename = args.filename
key = args.key.encode()
# 随机生成加密密钥
cipher_suite = cryptography.fernet.Fernet(key)
# 判断文件是否存在
if not os.path.isfile(filename):
print(f"Error: {filename} does not exist.")
exit()
# 加密文件内容
with open(filename, 'rb') as f:
plaintext = f.read()
ciphertext = cipher_suite.encrypt(plaintext)
# 将加密后的内容写入文件
with open(filename + '.enc', 'wb') as f:
f.write(ciphertext)
print(f"{filename} encrypted successfully as {filename}.enc.")
# 解密文件内容
with open(filename + '.enc', 'rb') as f:
ciphertext = f.read()
plaintext = cipher_suite.decrypt(ciphertext)
# 将解密后的内容写入文件
with open(filename + '.decrypted', 'wb') as f:
f.write(plaintext)
print(f"{filename}.enc decrypted successfully as {filename}.decrypted.")
```
在上述示例中,使用`argparse`模块定义了两个命令行参数:`filename`和`key`。`filename`参数表示要加密/解密的文件名,`key`参数表示加密/解密使用的密钥。然后使用`cryptography.fernet.Fernet()`函数生成一个加密密钥,将密钥保存在变量`cipher_suite`中。接下来,判断文件是否存在,如果文件不存在,输出错误信息并退出程序。然后,使用`cipher_suite.encrypt()`函数对文件内容进行加密得到密文,使用`open()`函数创建一个新的文件,将加密后的内容写入其中。最后,使用`cipher_suite.decrypt()`函数解密文件内容得到明文,并将明文写入新的文件中。
注意:该示例脚本使用`cryptography`库中的Fernet加密算法进行加密和解密,具有较高的安全性和保密性。但是请注意保护好密钥,不要将密钥泄露给他人。