Add types

Author: myakove

app/login.py

@@ -31,7 +31,7 @@
 """
 
 
-def vault_login():
+def vault_login() -> hvac.Client | None:
     client = hvac.Client(verify=False)
 
     auth_url_response = client.auth.oidc.oidc_authorization_url_request(
@@ -64,7 +64,7 @@ def vault_login():
 
 
 # handles the callback
-def login_oidc_get_token():
+def login_oidc_get_token() -> str:
     from http.server import BaseHTTPRequestHandler, HTTPServer
 
     class HttpServ(HTTPServer):

app/utils.py

@@ -6,14 +6,14 @@
 import dmenu
 
 
-def copy_to_clipboard(text):
+def copy_to_clipboard(text: bytes):
     p = Popen(["xsel", "-i"], stdin=PIPE)
     p.communicate(input=text)
 
 
-def which(program):
-    def is_exe(fpath):
-        return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
+def which(program: str) -> str | None:
+    def is_exe(_fpath):
+        return os.path.isfile(_fpath) and os.access(_fpath, os.X_OK)
 
     fpath, _ = os.path.split(program)
     if fpath:
@@ -28,7 +28,7 @@ def is_exe(fpath):
     return None
 
 
-def call_rofi_dmenu(options, abort=True, prompt=None):
+def call_rofi_dmenu(options: list, abort: bool = True, prompt: str = None) -> str:
     if which("rofi"):
         _rofi = Rofi()
         index, key = _rofi.select(prompt or "Select:", options)
@@ -43,7 +43,7 @@ def call_rofi_dmenu(options, abort=True, prompt=None):
         return user_select
 
 
-def parse_user_config():
+def parse_user_config() -> tuple:
     config_file = os.path.join(os.path.expanduser("~"), ".config", "vaultrun", "config")
     _config_parser = ConfigParser()
     # Open the file with the correct encoding
@@ -52,7 +52,7 @@ def parse_user_config():
     if len(sections_from_config) == 1:
         config_section = sections_from_config[0]
     else:
-        config_section = call_rofi_dmenu(options=sections_from_config, abort=True, prompt=None)
+        config_section = call_rofi_dmenu(options=sections_from_config)
 
     _mount_point = _config_parser[config_section]["mount_point"]
     _secret_path = _config_parser[config_section]["secret_path"]

app/vaultrun.py

@@ -1,9 +1,11 @@
+import hvac
+
 from app.login import vault_login
 from app.utils import call_rofi_dmenu, parse_user_config, which, copy_to_clipboard
 from rich import print, pretty
 
 
-def parse_vault_path(client, mount_point, secret_path):
+def parse_vault_path(client: hvac.Client, mount_point: str, secret_path: str) -> str:
     all_secrets = client.secrets.kv.v2.list_secrets(mount_point=mount_point, path=secret_path)
     _selected = call_rofi_dmenu(options=all_secrets.get("data", {}).get("keys"), abort=True, prompt=None)
     if _selected.endswith("/"):