Only load rarely used imports when needed

This commit is contained in:
2022-11-06 11:03:06 -05:00
parent 8916f9b2e2
commit 1c6150afe4
+3 -3
View File
@@ -1,18 +1,17 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from base64 import b32decode from base64 import b32decode
from hmac import new as new_mac
from os import environ, listdir, path, system, uname from os import environ, listdir, path, system, uname
from pathlib import Path from pathlib import Path
from shutil import rmtree
from sshync import get_profile from sshync import get_profile
from sshyp import decrypt, entry_list_gen, shm_gen, whitelist_verify from sshyp import decrypt, entry_list_gen, shm_gen, whitelist_verify
from struct import pack, unpack
from sys import argv, exit as s_exit from sys import argv, exit as s_exit
from time import sleep, strftime, time from time import sleep, strftime, time
def totp(_secret, _algo, _digits, _period): # uses provided information to generate a standard totp key def totp(_secret, _algo, _digits, _period): # uses provided information to generate a standard totp key
from hmac import new as new_mac
from struct import pack, unpack
_secret = b32decode(_secret.upper() + '=' * ((8 - len(_secret)) % 8)) _secret = b32decode(_secret.upper() + '=' * ((8 - len(_secret)) % 8))
_counter = pack('>Q', int(time() / _period)) _counter = pack('>Q', int(time() / _period))
_mac = new_mac(_secret, _counter, _algo).digest() _mac = new_mac(_secret, _counter, _algo).digest()
@@ -22,6 +21,7 @@ def totp(_secret, _algo, _digits, _period): # uses provided information to gene
def mfa_read_shortcut(): # reads and extracts MFA info from the user-specified sshyp entry def mfa_read_shortcut(): # reads and extracts MFA info from the user-specified sshyp entry
from shutil import rmtree
if not Path(f"{directory}{argument}.gpg").exists(): if not Path(f"{directory}{argument}.gpg").exists():
print(f"\n\u001b[38;5;9merror: entry ({argument}) does not exist\u001b[0m\n") print(f"\n\u001b[38;5;9merror: entry ({argument}) does not exist\u001b[0m\n")
s_exit(1) s_exit(1)