Skip to content

Commit 69d749d

Browse files
Add files via upload
1 parent 7979a36 commit 69d749d

File tree

4 files changed

+10
-9
lines changed

4 files changed

+10
-9
lines changed

utils/conversor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def h2b(hex:str)-> bytes:
1818

1919

2020
# Convert a number to an array of numbers
21-
def numberToArr(n, base) -> list:
21+
def numberToArr(n:int, base:int) -> list:
2222
if n == 0: return [0]
2323
digits = []
2424
while n:
@@ -28,7 +28,7 @@ def numberToArr(n, base) -> list:
2828

2929

3030
# Convert from base 10 to base 4
31-
def quat(n) -> int:
31+
def quat(n:int) -> int:
3232
b = bin(n)[2:]
3333

3434
if len(b) % 2 != 0:
@@ -41,7 +41,7 @@ def quat(n) -> int:
4141
return quaternary_n
4242

4343
# Convert from int to base 2,4,8 or 16
44-
def conv(x, base):
44+
def conv(x:int, base:int) -> str:
4545
if base == 2:
4646
return bin(x)[2:]
4747
elif base == 4:

utils/cryptography.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212

1313
# Generate a key with Fernet.generate_key(), save it to a file and return the path
14-
def gen_key(new_filename=None) -> str:
14+
def gen_key(new_filename:str=None) -> str:
1515
key = Fernet.generate_key()
1616

1717
if new_filename:
@@ -35,7 +35,7 @@ def gen_key(new_filename=None) -> str:
3535

3636

3737
# Load the key from the path
38-
def load_key(path) -> bytes:
38+
def load_key(path:str) -> bytes:
3939
stem = Path(path).stem
4040

4141
with open(path, 'rb') as f:
@@ -80,7 +80,7 @@ def get_key_paths() -> list:
8080

8181

8282
# Decrypt the content of the file and filename trying with all the key files
83-
def decrypt_content(file, filename) -> dict:
83+
def decrypt_content(file:bytes, filename:bytes) -> dict:
8484
for key_path in get_key_paths():
8585
fernet = Fernet(load_key(key_path))
8686
try:

utils/folders.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,15 @@
55
from utils.ctxt import *
66

77
# Create folders if they don't exist
8-
def create_folders(*args) -> None:
8+
def create_folders(*args:str) -> None:
99
folders = args
1010

1111
for folder in folders:
1212
# Create folder if it doesn't exist
1313
os.makedirs(folder, exist_ok=True)
1414

1515

16-
def folder2png(folder) -> None:
16+
def folder2png(folder:str) -> None:
1717
for file in os.listdir(folder):
1818
ext = Path(file).suffix
1919
stem = Path(file).stem

utils/input.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
import os
22
from pathlib import Path
3+
from xmlrpc.client import boolean
34

45
from utils.ctxt import *
56

67
def check_path(path:str) -> bool:
78
return os.path.isfile(path)
89

9-
def get_valid_filename(msg:str, exts:list=None, allow_blank=False) -> str:
10+
def get_valid_filename(msg:str, exts:list=None, allow_blank:bool=False) -> str:
1011
while True:
1112
filename = input(msg).strip()
1213
ext = Path(filename).suffix

0 commit comments

Comments
 (0)