用python实现文件加密功能

生活中，有时候咱们须要对一些重要的文件进行加密，Python 提供了诸如 hashlib，base64 等便于使用的加密库。

但对于平常学习而言，咱们能够借助异或操做，实现一个简单的文件加密程序，从而强化自身的编程能力。

基础知识

在 Python 中异或操做符为：^，也能够记做 XOR。按位异或的意思是：相同值异或为 0，不一样值异或为 1。具体来说，有四种可能：0 ^ 0 = 0，0 ^ 1 = 1, 1 ^ 0 = 1, 1 ^ 1 = 0。咱们还可总结出规律（A 为 0 或 1）：0 和 A 异或为 A自己；1 和 A 异或为 A 反。

让咱们想看看一位二进制数知足的性质：

• 一位二进制数与自身的异或值为 0

b ^ b = 0

• 异或操做知足交换律

a ^ b ^ c = a ^ (b ^ c) = (a ^ b) ^ c

• 0 与 a 的异或为 a

(a ^ b) ^ b = a ^ (b ^ b) = a ^ 0 = a

易知，对任意长二进制数都知足上述性质。

原理

经过了解异或操做的性质，加密原理就很是清晰了。

加密操做：

首先将文件转换成二进制数，再生成与该二进制数等长的随机密钥，将二进制数与密钥进行异或操做，获得加密后的二进制数。

解密操做：

将加密后的二进制程序与密钥进行异或操做，就获得原二进制数，最后将原二进制数恢复成文本文件。

生成随机密钥：

secrets 库是 Python 3.6 引入的伪随机数模块，适合生成随机密钥。token_bytes 函数接受一个 int 参数，用于指定随机字节串的长度。int.from_bytes 把字节串转换为 int，也就是咱们须要的二进制数。

from secrets import token_bytes

def random_key(length):
    key = token_bytes(nbytes=length)
    key_int = int.from_bytes(key, 'big')
    return key_int

 

加密单元:

encrypt 函数接受一个 str 对象，返回元组 (int, int)。经过 encode 方法，咱们将字符串编码成字节串。int.from_bytes 函数将字节串转换为 int 对象。最后对二进制对象和随机密钥进行异或操做，就获得了加密文本。

def encrypt(raw):
    raw_bytes = raw.encode()
    raw_int = int.from_bytes(raw_bytes, 'big')
    key_int = random_key(len(raw_bytes))
    return raw_int ^ key_int, key_int

 

解密单元：

decrypt 接受两个 int 对象，分别为加密文本和随机密钥。首先对二者进行异或操做，计算解密出来的 int 对象所占比特数。decrypted.bit_length 函数获得的是二进制数的位数，除以 8 能够获得所占比特大小。为了防止，1 ~ 7 位的二进制数整除 8 获得 0，因此要加上 7，而后再进行整除 8 的操做。使用 int.to_bytes 函数将解密以后的 int 的对象转换成 bytes 对象。最后经过 decode 方法，将字节串转换成字符串。

def decrypt(encrypted, key_int):
    decrypted = encrypted ^ key_int
    length = (decrypted.bit_length() + 7) // 8
    decrypted_bytes = int.to_bytes(decrypted, length, 'big') 
    return decrypted_bytes.decode()

 

利用上述函数，咱们能够很轻松对文本文件进行加密、解密操做。

>>> raw = '画图省识春风面，环珮空归夜月魂'
>>> encrypted = encrypt(raw)
>>> encrypted
(217447100157746604585...,
 9697901906831571319...)
>>> decrypt(*encrypted)
'画图省识春风面，环珮空归夜月魂'

加密文本文件

path 为待加密文件的地址，若是不指定密钥地址，则在该目录下新建目录和文件。

import json
from pathlib import Path

def encrypt_file(path, key_path=None, *, encoding='utf-8'):
    path = Path(path)
    cwd = path.cwd() / path.name.split('.')[0]
    path_encrypted = cwd / path.name 
    if key_path is None:
        key_path = cwd / 'key'
    if not cwd.exists():
        cwd.mkdir()
        path_encrypted.touch()
        key_path.touch()

    with path.open('rt', encoding=encoding) as f1, \
        path_encrypted.open('wt', encoding=encoding) as f2, \
            key_path.open('wt', encoding=encoding) as f3:
        encrypted, key = encrypt(f1.read())
        json.dump(encrypted, f2)
        json.dump(key, f3)

 

解密文件

#Python学习群592539176
def decrypt_file(path_encrypted, key_path=None, *, encoding='utf-8'):
    path_encrypted = Path(path_encrypted)
    cwd = path_encrypted.cwd()
    path_decrypted = cwd / 'decrypted' 
    if not path_decrypted.exists():
        path_decrypted.mkdir()
        path_decrypted /= path_encrypted.name
        path_decrypted.touch()
    if key_path is None:
        key_path = cwd / 'key'
    with path_encrypted.open('rt', encoding=encoding) as f1, \
        key_path.open('rt', encoding=encoding) as f2, \
        path_decrypted.open('wt', encoding=encoding) as f3:
        decrypted = decrypt(json.load(f1), json.load(f2))
        f3.write(decrypted)

 

执行完加密、解密文件操做，获得的解密文件与原文件相同，示意图以下：

以上就是 Python 加密文件的所有内容，还请你们多多转发支持。