python编写程序实现移位密码和文件读写。加密时:读取明文文件(小写字母),键盘输入密钥(一个参数),结果保存为密文文件(大写字母)
时间: 2024-01-23 17:04:03 浏览: 72
好的,我可以帮你编写这个程序。首先,我们来实现移位密码加密的函数:
```python
def shift_encrypt(plaintext, key):
ciphertext = ""
for char in plaintext:
if char.islower():
ciphertext += chr((ord(char) - 97 + key) % 26 + 65)
else:
ciphertext += char
return ciphertext
```
这个函数接受两个参数,明文和密钥,返回密文。它会遍历明文中的每个字符,如果是小写字母,则将其转换成大写字母后再进行移位加密,否则直接保留原字符。加密的方式是将字符的 ASCII 码值减去 97(小写字母 a 的 ASCII 码值),加上密钥,再对 26 取模,最后加上 65(大写字母 A 的 ASCII 码值),转换成对应的大写字母。注意,这里我们假设明文中只包含小写字母和其他字符,如果有其他情况需要额外进行处理。
接下来,我们来实现文件读写的功能。这里我们使用 Python 中的文件对象,打开明文文件和密文文件,分别读取明文并加密,将密文写入到密文文件中:
```python
def shift_file_encrypt(input_file, output_file, key):
with open(input_file, "r") as f:
plaintext = f.read()
ciphertext = shift_encrypt(plaintext, key)
with open(output_file, "w") as f:
f.write(ciphertext)
```
这个函数接受三个参数,明文文件名、密文文件名和密钥。它使用 `with` 语句打开文件,读取明文并调用上面的 `shift_encrypt` 函数加密,最后将密文写入到密文文件中。注意,这里我们使用的是文本模式,即将文件内容视为字符串处理,如果需要处理二进制文件需要使用二进制模式。
最后,我们可以编写一个测试函数,来测试我们的程序是否正常工作:
```python
def test_shift_file_encrypt():
input_file = "plaintext.txt"
output_file = "ciphertext.txt"
key = 3
with open(input_file, "w") as f:
f.write("hello world")
shift_file_encrypt(input_file, output_file, key)
with open(output_file, "r") as f:
ciphertext = f.read()
assert ciphertext == "KHOOR ZRUOG"
print("All tests passed")
```
这个函数先创建一个明文文件,然后调用我们的加密函数进行加密,最后读取密文文件并检查是否正确。如果一切正常,会输出 "All tests passed"。
完整代码如下:
阅读全文