Turn _ratelimit_resource into class attribute

justfortherec · justfortherec · commit 89ac06d18b87 · 2018-01-26T18:51:00.000+01:00
diff --git a/github3/models.py b/github3/models.py
@@ -31,6 +31,7 @@ class GitHubCore(object):
     basic attributes and methods to other sub-classes that are very useful to
     have.
     """
+    _ratelimit_resource = 'core'
 
     def __init__(self, json, session=None):
         if hasattr(session, 'session'):
@@ -42,7 +43,6 @@ def __init__(self, json, session=None):
 
         # set a sane default
         self._github_url = 'https://api.github.com'
-        self._ratelimit_resource = 'core'
 
         if json is not None:
             self.etag = json.pop('ETag', None)
@@ -507,6 +507,4 @@ def _repr(self):
 
 class GitHubSearch(GitHubCore):
     """The base object for all search objects."""
-    def __init__(self, json, session=None):
-        super(GitHubSearch, self).__init__(json, session)
-        self._ratelimit_resource = 'search'
+    _ratelimit_resource = 'search'
diff --git a/github3/structs.py b/github3/structs.py
@@ -129,6 +129,7 @@ class SearchIterator(GitHubIterator):
     class. For other members and methods, check its parent class.
 
     """
+    _ratelimit_resource = 'search'
 
     def __init__(self, count, url, cls, session, params=None, etag=None,
                  headers=None):
@@ -138,7 +139,6 @@ def __init__(self, count, url, cls, session, params=None, etag=None,
         self.total_count = 0
         #: Items array returned in the last request
         self.items = []
-        self._ratelimit_resource = 'search'
 
     def _repr(self):
         return '<SearchIterator [{0}, {1}?{2}]>'.format(self.count, self.path,
