From a175dd574f3880edbc9e8b7a6b8de35f34f1fac6 Mon Sep 17 00:00:00 2001 From: ache Date: Fri, 21 Apr 2023 03:40:06 +0200 Subject: Add types to function --- autoDHCP.py | 15 ++++++++------- 1 file 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 -- cgit v1.2.3