Added page.skip_action(), skip restbase if non-wikipedia wiki #127

Author: siznax

wptools/page.py

@@ -472,6 +472,14 @@ def _update_params(self):
         if self.data.get('wikibase'):
             self.params['wikibase'] = self.data.get('wikibase')
 
+    def skip_action(self, action):
+        """
+        append action to skip flag
+        """
+        if 'skip' not in self.flags:
+            self.flags['skip'] = []
+        self.flags['skip'].append(action)
+
     def get(self, show=True, proxy=None, timeout=0):
         """
         Make Mediawiki, RESTBase, and Wikidata requests for page data
@@ -506,14 +514,16 @@ def get(self, show=True, proxy=None, timeout=0):
             self.get_parse(False, proxy, timeout)
 
             if not self.data.get('wikibase'):
-                if 'skip' not in self.flags:
-                    self.flags['skip'] = []
-                self.flags['skip'].append('wikidata')
+                self.skip_action('wikidata')
 
             self.get_wikidata(False, proxy, timeout)
 
             self.flags['defer_imageinfo'] = False
 
+            wiki = self.params.get('wiki')
+            if wiki and 'wikipedia.org' not in wiki:
+                self.skip_action('restbase')
+
             self.get_restbase('/page/summary/', False, proxy, timeout)
 
             if show and not self.flags.get('silent'):