Merge pull request #2 from r0ei/patch-1

Fixed index out of range error
mxrch · Mar 26, 2021 · 00a2fbb · 00a2fbb
2 parents 44acebc + 21b4e9b
commit 00a2fbb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/webwrap.py b/webwrap.py
@@ -15,8 +15,11 @@
 	reg = """\]LEDEBUT\]([\s\S]*)\]LAFIN\]"""
 
 	req = httpx.get(host.replace("WRAP", "echo -n ]LEDEBUT]$(whoami)[$(hostname)[$(pwd)]LAFIN]"))
-	prefixes = re.compile(reg).findall(req.text)[0].split("[")
-	path = prefixes[2]
+	prefixes = re.compile(reg).findall(req.text)
+	if not prefixes:
+		print("Req.text not found!\n")
+		exit(-1)
+	path = prefixes[0].split("[")[2]
 	prefix = colored(prefixes[0] + "@" + prefixes[1], "red") + ":" + colored(prefixes[2], "cyan") + "$ "
 	print("")