Fix spansh url check

seldonlabs · Jun 27, 2018 · de6cac5 · de6cac5
1 parent 866d8e0
commit de6cac5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spansh.py b/spansh.py
@@ -1,6 +1,7 @@
 import json
 import os
 import pyperclip
+import re
 import requests
 import threading
 import time
@@ -106,7 +107,8 @@ def button_callback(self):
         if DEBUG:
             url = "https://spansh.co.uk/api/results/zuppa"
 
-        if not url.startswith("https://spansh.co.uk/api/results/"):
+        # if not url.startswith("https://spansh.co.uk/api/results/"):
+        if not re.match(r"^https://(www\.)?spansh.co.uk/api/results/", url):
             print "Wrong URL, not spansh!"
             self.show_error("The URL you pasted is not a spansh.co.uk URL."
                             " Please copy a plot URL from https://spansh.co.uk"