#!/usr/bin/python3

# coding:utf-8

# Auther:AlphaPanda

# Description:使用hashlib模块，对文件内容进行校验。以此来判断文件传输过程中是否发生变化，或者有损坏

# Version:1

# Date:Fri Dec  6 22:15:16 EST 2019

import hashlib

# 针对小文件的内容校验

def check_md5(file):

    with open(file,mode="rb") as fp:

        res = fp.read()

        hs = hashlib.md5()

        hs.update(res)

    return hs.hexdigest()

res = check_md5("ceshi1.txt")

print(res)

# 针对大文件的内容校验

hs = hashlib.md5()

hs.update("你好\n世界".encode())

print(hs.hexdigest())

hs = hashlib.md5()

hs.update("你好".encode())

hs.update("\n世界".encode())

print(hs.hexdigest())

# 通过以上实验，证明一次加密文件的一行，依次加密完所有的文件内容和一次加密所有的文件内容，所获得的hash值一样。

# 大文件校验方法1：

def check2_md5(file):

    hs = hashlib.md5()

    with open(file,mode="rb") as fp:

        while True:

            content = fp.read(3)

            if content:

                hs.update(content)

            else:

                break

        return hs.hexdigest()

print(check2_md5("ceshi3.txt"))

# 方法二：

import os

def check4_md5(file):

    # 计算文件大小

    file_size = os.path.getsize(file)

    hs = hashlib.md5()

    with open(file,mode="rb") as fp:

        while file_size:

            # 一次最多读取三个字节

            content = fp.read(3)

            hs.update(content)

            file_size -= len(content)

        return hs.hexdigest()

print(check4_md5("ceshi4.txt"))