Add types to function

1 files changed, 8 insertions, 7 deletions

diff --git a/autoDHCP.py b/autoDHCP.py
index ce699c4..247d2f2 100755
--- a/autoDHCP.py
+++ b/autoDHCP.py
@@ -9,21 +9,21 @@ import subprocess
 import tempfile
 
 
-def guess_wifi():
+def guess_wifi() -> list[str]:
     return [interface for interface in os.listdir('/sys/class/net/') if
             exists(f'/sys/class/net/{interface}/wireless')]
 
 
-def guess_nowifi():
+def guess_nowifi() -> list[str]:
     return [interface for interface in os.listdir('/sys/class/net/') if
             not exists(f'/sys/class/net/{interface}/wireless')]
 
 
-def guess_interface():
+def guess_interface() -> list[str]:
     return [interface for interface in os.listdir('/sys/class/net/')]
 
 
-def guess_internet():
+def guess_internet() -> list[str]:
     com = subprocess.Popen(['ip', 'route', 'show'], stdout=subprocess.PIPE,
                            stderr=subprocess.STDOUT)
     stdout, stderr = com.communicate()
@@ -39,7 +39,7 @@ def guess_internet():
     return list(set(internet_i))
 
 
-def stripedLine(line):
+def stripedLine(line) -> str:
     words = line.split(' ')
     if 'dev' in words and words.index('dev'):
         return words[words.index('dev') + 1]
@@ -47,10 +47,11 @@ def stripedLine(line):
         return ''
 
 
-def menu(listEntry, strInput, refresh=None):
+def menu(listEntry, strInput, refresh=None) -> int:
     if listEntry == []:
         print(f"Error menu: Empty list\n({strInput})", file=sys.stderr)
-        exit(1)
+        sys.exit(1)
+
     c = -1
     if len(listEntry) == 1:
         c = 0