Correct schema problem when adding URL with -u option (issues #2)

t4d · Sep 2, 2018 · b1901a1 · b1901a1
1 parent 97803ce
commit b1901a1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/stalkphish/tools/addurl.py b/stalkphish/tools/addurl.py
@@ -20,10 +20,11 @@
 def AddUniqueURL(URLadd,LOG,SQL,TABLEname,PROXY,UAFILE):
 	UAG = UAgent()
 	# add schema 
-	if not URLadd.startswith("http(|s)://") :
-		URLadd="http://{}".format(URLadd)
-	else:
+	if URLadd.startswith("http://") or URLadd.startswith("https://"):
 		pass
+	else:
+		URLadd="http://{}".format(URLadd)
+
 	# remove URL containing UID-style strings
 	siteURL = re.split("(?:[0-9a-fA-F]:?){32}", URLadd.rstrip())[0]
 	## Test if entry still exist in DB