From e1bf8348772da89a645a5359fb7b7625e4061cd7 Mon Sep 17 00:00:00 2001
From: James Lu <GLolol1@hotmail.com>
Date: Sun, 17 May 2015 21:51:36 -0700
Subject: [PATCH] Wikifetch: factorize --site checking, and don't assume
 Wikipedia

---
 Wikifetch/plugin.py | 17 +++++++----------
 1 file changed, 7 insertions(+), 10 deletions(-)

diff --git a/Wikifetch/plugin.py b/Wikifetch/plugin.py
index 607fd79..a3bcb74 100644
--- a/Wikifetch/plugin.py
+++ b/Wikifetch/plugin.py
@@ -71,16 +71,13 @@ class Wikifetch(callbacks.Plugin):
         # such as filling in http:// and appending /wiki to links...
 
         # Try using '--site lyrics.wikia.com' or '--site wiki.archlinux.org'.
-        if 'site' in optlist:
-            baseurl = optlist['site']
-            if 'wikia.com' in baseurl:
-                baseurl += '/wiki'
-            elif 'wiki.archlinux.org' in baseurl:
-                baseurl += '/index.php'
-            if not baseurl.lower().startswith(('http://', 'https://')):
-                baseurl = 'http://' + baseurl
-        else:
-            baseurl = 'https://%s/wiki' % self.registryValue('url', msg.args[0])
+        baseurl = optlist.get('site') or self.registryValue('url', msg.args[0])
+        if 'wikia.com' in baseurl or 'wikipedia.org' in baseurl:
+            baseurl += '/wiki'
+        elif 'wiki.archlinux.org' in baseurl:
+            baseurl += '/index.php'
+        if not baseurl.lower().startswith(('http://', 'https://')):
+            baseurl = 'http://' + baseurl
         # first, we get the page
         addr = '%s/Special:Search?search=%s' % \
                     (baseurl, quote_plus(search))