🗓 Dec 29, 2021 5:51:16 PM

🔍 added Literal types of some functions
securisec · Dec 29, 2021 · 3840240 · 3840240
1 parent 70eba2c
commit 3840240
Show file tree

Hide file tree

Showing 7 changed files with 26 additions and 26 deletions.
diff --git a/chepy/core.pyi b/chepy/core.pyi
@@ -1,4 +1,4 @@
-from typing import Any, List, Mapping, Tuple, Union, TypeVar
+from typing import Any, List, Mapping, Tuple, Union, TypeVar, Literal
 
 jsonpickle: Any
 
@@ -64,5 +64,5 @@ class ChepyCore:
     def reset(self: ChepyCoreT): ...
     def load_command(self: ChepyCoreT): ...
     def pretty(self: ChepyCoreT, indent: int=...) -> ChepyCoreT: ...
-    def plugins(self: ChepyCoreT, enable: str) -> None: ...
+    def plugins(self: ChepyCoreT, enable: Literal['true', 'false']) -> None: ...
     def set_plugin_path(self: ChepyCoreT, path: str) -> None: ...
diff --git a/chepy/modules/codetidy.pyi b/chepy/modules/codetidy.pyi
@@ -1,5 +1,5 @@
 from ..core import ChepyCore
-from typing import Any, TypeVar
+from typing import Any, TypeVar, Literal
 
 pydash: Any
 CodeTidyT = TypeVar('CodeTidyT', bound='CodeTidy')
@@ -12,7 +12,7 @@ class CodeTidy(ChepyCore):
     def minify_xml(self: CodeTidyT) -> CodeTidyT: ...
     def beautify_xml(self: CodeTidyT) -> CodeTidyT: ...
     def php_deserialize(self: CodeTidyT) -> CodeTidyT: ...
-    def to_upper_case(self: CodeTidyT, by: str=...) -> CodeTidyT: ...
+    def to_upper_case(self: CodeTidyT, by: Literal['all', 'word', 'sentence']=...) -> CodeTidyT: ...
     def to_lower_case(self: CodeTidyT) -> CodeTidyT: ...
     def to_snake_case(self: CodeTidyT) -> CodeTidyT: ...
     def to_camel_case(self: CodeTidyT, ignore_space: bool=...) -> CodeTidyT: ...

diff --git a/chepy/modules/compression.pyi b/chepy/modules/compression.pyi
@@ -1,5 +1,5 @@
 from ..core import ChepyCore
-from typing import Any, TypeVar
+from typing import Any, TypeVar, Literal
 
 CompressionT = TypeVar('CompressionT', bound='Compression')
 
@@ -12,10 +12,10 @@ class Compression(ChepyCore):
     def unzip_one(self: CompressionT, file: str, password: str=...) -> CompressionT: ...
     def unzip_all(self: CompressionT, password: str=...) -> CompressionT: ...
     def create_zip_file(self: CompressionT, file_name: str) -> CompressionT: ...
-    def tar_list_files(self: CompressionT, mode: str=...) -> CompressionT: ...
-    def tar_extract_one(self: CompressionT, filename: str, mode: str=...) -> CompressionT: ...
-    def tar_extract_all(self: CompressionT, mode: str=...) -> CompressionT: ...
-    def tar_compress(self: CompressionT, filename: str, mode: str=...) -> CompressionT: ...
+    def tar_list_files(self: CompressionT, mode: Literal["gz", "bz2", "xz", ""]=...) -> CompressionT: ...
+    def tar_extract_one(self: CompressionT, filename: str, mode: Literal["gz", "bz2", "xz", ""]=...) -> CompressionT: ...
+    def tar_extract_all(self: CompressionT, mode: Literal["gz", "bz2", "xz", ""]=...) -> CompressionT: ...
+    def tar_compress(self: CompressionT, filename: str, mode: Literal["gz", "bz2", "xz", ""]=...) -> CompressionT: ...
     def gzip_compress(self: CompressionT, file_name: str=...) -> CompressionT: ...
     def gzip_decompress(self: CompressionT) -> CompressionT: ...
     def bzip_compress(self: CompressionT) -> CompressionT: ...

diff --git a/chepy/modules/encryptionencoding.py b/chepy/modules/encryptionencoding.py
@@ -3,7 +3,7 @@
 import codecs
 import itertools
 import string
-from typing import TypeVar
+from typing import Literal, TypeVar
 
 import lazy_import
 
@@ -159,7 +159,7 @@ def rot_47(self) -> EncryptionEncodingT:
 
     @ChepyDecorators.call_stack
     def xor(
-        self, key: str, key_type: str = "hex", ascii: bool = False
+        self, key: str, key_type: Literal['hex', 'utf', 'base64'] = "hex", ascii: bool = False
     ) -> EncryptionEncodingT:
         """XOR state with a key
 

diff --git a/chepy/modules/encryptionencoding.pyi b/chepy/modules/encryptionencoding.pyi
@@ -1,5 +1,5 @@
 from ..core import ChepyCore
-from typing import Any, TypeVar
+from typing import Any, TypeVar, Literal
 
 jwt: Any
 AES: Any
@@ -16,22 +16,22 @@ class EncryptionEncoding(ChepyCore):
     def rotate_bruteforce(self: EncryptionEncodingT) -> EncryptionEncodingT: ...
     def rot_13(self: EncryptionEncodingT) -> EncryptionEncodingT: ...
     def rot_47(self: EncryptionEncodingT) -> EncryptionEncodingT: ...
-    def xor(self: EncryptionEncodingT, key: str, key_type: str=..., ascii: bool=...) -> EncryptionEncodingT: ...
+    def xor(self: EncryptionEncodingT, key: str, key_type: Literal['hex', 'utf', 'base64']=..., ascii: bool=...) -> EncryptionEncodingT: ...
     def xor_bruteforce(self: EncryptionEncodingT, length: int=...) -> EncryptionEncodingT: ...
     def jwt_decode(self: EncryptionEncodingT) -> EncryptionEncodingT: ...
     def jwt_verify(self: EncryptionEncodingT, secret: str, algorithm: list=...) -> EncryptionEncodingT: ...
     def jwt_sign(self: EncryptionEncodingT, secret: str, algorithms: str=...) -> EncryptionEncodingT: ...
     def jwt_bruteforce(self: EncryptionEncodingT, wordlist: str, b64_encode: bool=..., algorithm: list=...) -> EncryptionEncodingT: ...
     def rc4_encrypt(self: EncryptionEncodingT, key: str, hex_key: bool=...) -> EncryptionEncodingT: ...
     def rc4_decrypt(self: EncryptionEncodingT, key: str, hex_key: bool=...) -> EncryptionEncodingT: ...
-    def des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def des_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def triple_des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def triple_des_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def aes_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def aes_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def blowfish_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
-    def blowfish_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: str=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def des_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def triple_des_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def triple_des_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def aes_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB", "GCM"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def aes_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB", "GCM"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def blowfish_encrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
+    def blowfish_decrypt(self: EncryptionEncodingT, key: str, iv: str=..., mode: Literal["CBC", "OFB", "CTR", "ECB"]=..., hex_key: bool=..., hex_iv: bool=...) -> EncryptionEncodingT: ...
     def vigenere_encode(self: EncryptionEncodingT, key: str) -> EncryptionEncodingT: ...
     def vigenere_decode(self: EncryptionEncodingT, key: str) -> EncryptionEncodingT: ...
     def affine_encode(self: EncryptionEncodingT, a: int=..., b: int=...) -> EncryptionEncodingT: ...

diff --git a/chepy/modules/hashing.pyi b/chepy/modules/hashing.pyi
@@ -26,7 +26,7 @@ class Hashing(ChepyCore):
     def sha1(self: HashingT) -> HashingT: ...
     def sha2_256(self: HashingT) -> HashingT: ...
     def sha2_512(self: HashingT) -> HashingT: ...
-    def sha2_512_truncate(self: HashingT, truncate: int=...) -> HashingT: ...
+    def sha2_512_truncate(self: HashingT, truncate: Literal[256, 224]=...) -> HashingT: ...
     def sha2_384(self: HashingT) -> HashingT: ...
     def sha2_224(self: HashingT) -> HashingT: ...
     def sha3_512(self: HashingT) -> HashingT: ...
@@ -43,8 +43,8 @@ class Hashing(ChepyCore):
     def shake_256(self: HashingT, size: int=...) -> HashingT: ...
     def shake_128(self: HashingT, size: int=...) -> HashingT: ...
     def ripemd_160(self: HashingT) -> HashingT: ...
-    def blake_2b(self: HashingT, bits: int=..., key: bytes=...) -> HashingT: ...
-    def blake_2s(self: HashingT, bits: int=..., key: bytes=...) -> HashingT: ...
+    def blake_2b(self: HashingT, bits: Literal[512, 384, 256, 160, 128]=..., key: bytes=...) -> HashingT: ...
+    def blake_2s(self: HashingT, bits: Literal[256, 160, 128]=..., key: bytes=...) -> HashingT: ...
     def crc8_checksum(self: HashingT) -> HashingT: ...
     def crc16_checksum(self: HashingT) -> HashingT: ...
     def crc32_checksum(self: HashingT) -> HashingT: ...

diff --git a/chepy/modules/utils.pyi b/chepy/modules/utils.pyi
@@ -1,5 +1,5 @@
 from ..core import ChepyCore
-from typing import Any, TypeVar, Union
+from typing import Any, Literal, TypeVar, Union
 
 UtilsT = TypeVar('UtilsT', bound='Utils')
 
@@ -32,7 +32,7 @@ class Utils(ChepyCore):
     def unescape_string(self: UtilsT) -> UtilsT: ...
     def color_hex_to_rgb(self: UtilsT) -> UtilsT: ...
     def diff(self: UtilsT, state: int=..., buffer: int=..., colors: bool=..., swap: bool=...) -> Any: ...
-    def pad(self: UtilsT, width: int, direction: str=..., char: str=...) -> UtilsT: ...
+    def pad(self: UtilsT, width: int, direction: Literal['left', 'right']=..., char: str=...) -> UtilsT: ...
     def count(self: UtilsT) -> UtilsT: ...
     def set(self: UtilsT) -> UtilsT: ...
     def regex_to_str(self: UtilsT, all_combo: bool=...) -> UtilsT: ...