style: 💄 format code using yapf

Author: ZhaoQi99

pyencrypt/decrypt.py

@@ -16,10 +16,7 @@ def _decrypt_file(data: bytes, key: str) -> bytes:
     return aes_decrypt(data, key)
 
 
-def decrypt_file(path: Path,
-                 key: str,
-                 delete_origin: bool = False,
-                 new_path: Path = None) -> bytes:
+def decrypt_file(path: Path, key: str, delete_origin: bool = False, new_path: Path = None) -> bytes:
     if path.suffix != '.pye':
         raise Exception(f"{path.name} can't be decrypted.")
     data = _decrypt_file(path.read_bytes(), key)

pyencrypt/encrypt.py

@@ -52,15 +52,15 @@ def generate_so_file(cipher_key: str, d: int, n: int, base_dir: Path = None):
     need_import_files = ['ntt.py', 'aes.py', 'decrypt.py']
     for file in need_import_files:
         file_path = path / file
-        decrypt_source_ls.append(file_path.read_text().replace(
-            'from pyencrypt.ntt import intt',
-            '').replace('from pyencrypt.aes import aes_decrypt', ''))
+        decrypt_source_ls.append(
+            file_path.read_text().replace('from pyencrypt.ntt import intt',
+                                          '').replace('from pyencrypt.aes import aes_decrypt', '')
+        )
 
     loader_source_path = path / 'loader.py'
     loader_source = loader_source_path.read_text().replace(
-        "__private_key = ''", f"__private_key = '{private_key}'",
-        1).replace("__cipher_key = ''", f"__cipher_key = '{cipher_key}'",
-                   1).replace("from pyencrypt.decrypt import *", '')
+        "__private_key = ''", f"__private_key = '{private_key}'", 1
+    ).replace("__cipher_key = ''", f"__cipher_key = '{cipher_key}'", 1).replace("from pyencrypt.decrypt import *", '')
 
     if base_dir is None:
         base_dir = Path(os.getcwd())
@@ -79,10 +79,14 @@ def generate_so_file(cipher_key: str, d: int, n: int, base_dir: Path = None):
     loader_origin_file_path.write_text(f"{decrypt_source}\n{loader_source}")
 
     args = [
-        'pyminifier', '--obfuscate-classes', '--obfuscate-import-methods',
-        '--replacement-length', '20', '-o',
+        'pyminifier',
+        '--obfuscate-classes',
+        '--obfuscate-import-methods',
+        '--replacement-length',
+        '20',
+        '-o',
+        loader_file_path.as_posix(),
         loader_file_path.as_posix(),
-        loader_file_path.as_posix()
     ]
     ret = subprocess.run(args, shell=False, encoding='utf-8')
     if ret.returncode == 0:
@@ -100,10 +104,7 @@ def generate_so_file(cipher_key: str, d: int, n: int, base_dir: Path = None):
     return True
 
 
-def encrypt_file(path: Path,
-                 key: str,
-                 delete_origin: bool = False,
-                 new_path: Path = None):
+def encrypt_file(path: Path, key: str, delete_origin: bool = False, new_path: Path = None):
     if not can_encrypt(path):
         raise Exception(f"{path.name} can't be encrypted.")
     encrypted_data = _encrypt_file(path.read_bytes(), key)

pyencrypt/generate.py

@@ -8,7 +8,6 @@ def generate_aes_key(size: int = 32) -> bytes:
     return base64.b64encode(os.urandom(size))
 
 
-def generate_rsa_number(bits: int)-> dict:
+def generate_rsa_number(bits: int) -> dict:
     r = RSA.generate(bits)
     return {'p': r.p, 'q': r.q, 'n': r.n, 'e': r.e, 'd': r.d}
-

pyencrypt/loader.py

@@ -32,22 +32,17 @@ def get_filename(self, fullname: str) -> str:
     def get_data(self, path: _Path) -> bytes:
         try:
             __n, __d = self.__private_key.split('O', 1)
-            return decrypt_file(
-                Path(path), decrypt_key(self.__cipher_key, int(__d), int(__n)))
+            return decrypt_file(Path(path), decrypt_key(self.__cipher_key, int(__d), int(__n)))
         except Exception:
             traceback.print_exc()
             return b''
 
 
 class EncryptFileFinder(abc.MetaPathFinder, Base):
-    def find_spec(self,
-                  fullname: str,
-                  path: Sequence[_Path],
-                  target: types.ModuleType = None) -> ModuleSpec:
+    def find_spec(self, fullname: str, path: Sequence[_Path], target: types.ModuleType = None) -> ModuleSpec:
         if path:
             if isinstance(path, _NamespacePath):
-                file_path = Path(
-                    path._path[0]) / f'{fullname.rsplit(".",1)[-1]}.pye'
+                file_path = Path(path._path[0]) / f'{fullname.rsplit(".",1)[-1]}.pye'
             else:
                 file_path = Path(path[0]) / f'{fullname.rsplit(".",1)[-1]}.pye'
         else:
@@ -59,9 +54,7 @@ def find_spec(self,
         if not os.path.exists(file_path):
             return None
         loader = EncryptFileLoader(file_path)
-        return spec_from_loader(name=fullname,
-                                loader=loader,
-                                origin='origin-encrypt')
+        return spec_from_loader(name=fullname, loader=loader, origin='origin-encrypt')
 
 
 # TODO: generate randomly AES Class

tests/test_aes.py

@@ -14,27 +14,30 @@
 PLAIN_3 = b'abcdefghijklmnop'
 CIPHER_3 = b'r\x14\xa7\x92\xd6\x1f\x0c\xf4\x10g\x99\t/\xf0z\xfc' + b'&\x80\xdb\x94\xd1\xf7\x9f\xe0Qo\x05\x98\x7f\xe6j\x8c'
 
-class TestAES:
 
-    @pytest.mark.parametrize('key, plain, cipher', [
-        (AES_KEY, PLAIN_1, CIPHER_1),
-        (AES_KEY, PLAIN_2, CIPHER_2),
-        (AES_KEY, PLAIN_3, CIPHER_3),
-    ])
+class TestAES:
+    @pytest.mark.parametrize(
+        'key, plain, cipher', [
+            (AES_KEY, PLAIN_1, CIPHER_1),
+            (AES_KEY, PLAIN_2, CIPHER_2),
+            (AES_KEY, PLAIN_3, CIPHER_3),
+        ]
+    )
     def test_aes_encrypt(self, plain, cipher, key):
         assert aes_encrypt(plain, key) == cipher
 
-    @pytest.mark.parametrize('key, plain, cipher', [
-        (AES_KEY, PLAIN_1, CIPHER_1),
-        (AES_KEY, PLAIN_2, CIPHER_2),
-        (AES_KEY, PLAIN_3, CIPHER_3),
-    ])
+    @pytest.mark.parametrize(
+        'key, plain, cipher', [
+            (AES_KEY, PLAIN_1, CIPHER_1),
+            (AES_KEY, PLAIN_2, CIPHER_2),
+            (AES_KEY, PLAIN_3, CIPHER_3),
+        ]
+    )
     def test_aes_decrypt(self, plain, cipher, key):
         assert aes_decrypt(cipher, key) == plain
 
 
 class TestAESModeOfOperationECB:
-
     def setup_class(self):
         self.cipher = AESModeOfOperationECB(AES_KEY)
 
@@ -54,12 +57,12 @@ def test_decrypt_exception(self, length):
         (PLAIN_PADDING_1, CIPHER_1),
         (PLAIN_PADDING_2, CIPHER_2),
     ])
-    def test_encrypt(self,plain, cipher):
+    def test_encrypt(self, plain, cipher):
         assert self.cipher.encrypt(plain) == cipher
 
     @pytest.mark.parametrize('plain, cipher', [
         (PLAIN_PADDING_1, CIPHER_1),
         (PLAIN_PADDING_2, CIPHER_2),
     ])
-    def test_decrypt(self,plain, cipher):
-        assert self.cipher.decrypt(cipher) == plain
+    def test_decrypt(self, plain, cipher):
+        assert self.cipher.decrypt(cipher) == plain

tests/test_encrypt.py

@@ -8,7 +8,6 @@
 from pyencrypt.generate import generate_aes_key
 
 
-
 @pytest.mark.parametrize('key', [
     AES_KEY,
     generate_aes_key(),
@@ -34,7 +33,6 @@ def test_can_encrypt(path, expected):
 
 
 class TestGenarateSoFile:
-
     def setup_method(self, method):
         if method.__name__ == 'test_generate_so_file_default_path':
             shutil.rmtree((Path(os.getcwd()) / 'encrypted').as_posix(), ignore_errors=True)
@@ -100,8 +98,9 @@ def test_encrypt_file_new_path(python_file_path):
     assert new_path.exists() == True
     assert python_file_path.exists() == True
 
+
 def test_encrypt_file_new_path_exception(python_file_path):
     new_path = python_file_path.parent / 'test.py'
     with pytest.raises(Exception) as excinfo:
         encrypt_file(python_file_path, AES_KEY, new_path=new_path)
-    assert str(excinfo.value) == 'Encrypted file path must be pye suffix.'
+    assert str(excinfo.value) == 'Encrypted file path must be pye suffix.'

tests/test_ntt.py

@@ -5,7 +5,6 @@
 
 
 class TestNtt:
-
     @pytest.mark.parametrize(
         "input,expected", [
             ([1, 2, 3, 4], [10, 173167434, 998244351, 825076915]),
@@ -29,7 +28,7 @@ def test_ntt_inverse(self, input, expected):
         random_list(6),
         random_list(7),
     ])
-    def test_ntt_exception(self,input):
+    def test_ntt_exception(self, input):
         with pytest.raises(ValueError) as excinfo:
             ntt(input)
         assert str(excinfo.value) == "The length of input must be a power of 2."
@@ -44,13 +43,11 @@ def test_intt_exception(self, input):
             intt(input)
         assert str(excinfo.value) == "The length of input must be a power of 2."
 
-    @pytest.mark.parametrize(
-        "input,expected", [
-            (random_list(4), 4),
-            (random_list(8), 8),
-            (random_list(16), 16),
-        ]
-    )
+    @pytest.mark.parametrize("input,expected", [
+        (random_list(4), 4),
+        (random_list(8), 8),
+        (random_list(16), 16),
+    ])
     def test_ntt_result_length(self, input, expected):
         assert len(ntt(input)) == expected