X-Git-Url: https://git.rmz.io/dotfiles.git/blobdiff_plain/db4fd9f943db71a5baa7c86580b86d0d809c15dd..e081ae2be04bf7224eadc202f3b9e821d65a440b:/qutebrowser/userscripts/config-cycle-tld

diff --git a/qutebrowser/userscripts/config-cycle-tld b/qutebrowser/userscripts/config-cycle-tld
index d1751a3..b5801d3 100755
--- a/qutebrowser/userscripts/config-cycle-tld
+++ b/qutebrowser/userscripts/config-cycle-tld
@@ -6,7 +6,15 @@ import tldextract
 
 url = os.getenv('QUTE_URL')
 
-pattern = '*://*.{}/*'.format(tldextract.extract(url).registered_domain)
+ext = tldextract.extract(url)
+pattern = None
+if ext.registered_domain:
+    pattern = '*://*.{}/*'.format(tldextract.extract(url).registered_domain)
+elif ext.domain:
+    pattern = '*://{}/*'.format(tldextract.extract(url).domain)
 
 with open(os.getenv('QUTE_FIFO'), 'w') as fifo:
+    if not pattern:
+        fifo.write('message-error Cannot extract tld or domain from `{}`', url)
+        sys.exit()
     fifo.write('config-cycle -u {} {}\n'.format(pattern, ' '.join(sys.argv[1:])))