def test_encrypt_decrypt(self): path = "testdata//test.mkv" with open(path, "rb") as in_f, open(path+".tmp", "wb") as out_f: encrypt.encrypt(in_f, out_f, "Password01!") with open(path+".tmp", "rb") as in_f, open(path+".tmp2", "wb") as out_f: encrypt.decrypt(in_f, out_f, "Password01!") with open(path+".tmp", "rb") as in_f, open(path+".tmp3", "wb") as out_f: encrypt.decrypt(in_f, out_f, "Password02!") p1 = fileprocessor.FileProcessor(path) p2 = fileprocessor.FileProcessor(path+".tmp2") p3 = fileprocessor.FileProcessor(path+".tmp3") self.assertEqual(p1.md5, p2.md5) self.assertNotEqual(p1.md5, p3.md5)
def chuncks(self):
if self.__chuncks == None:
cur_size = 0
if config.is_encrypt():
tmp_path = self.__file_path + ".tmp"
with open(self.__file_path, "rb") as in_f, open(tmp_path, "wb") as out_f:
encrypt.encrypt(in_f, out_f, config.config["pwd"])
cur_size = os.path.getsize(tmp_path)
else:
cur_size = self.size
if cur_size % FileProcessor.__CHUNCK_SIZE == 0:
self.__chuncks = cur_size // FileProcessor.__CHUNCK_SIZE
else:
self.__chuncks = cur_size // FileProcessor.__CHUNCK_SIZE + 1
return self.__chuncks