|
@@ -104,6 +104,13 @@ class Category(Page): |
|
|
|
|
|
|
|
|
def _get_size_via_api(self, member_type): |
|
|
def _get_size_via_api(self, member_type): |
|
|
"""Return the size of the category using the API.""" |
|
|
"""Return the size of the category using the API.""" |
|
|
|
|
|
result = self.site.api_query(action="query", prop="categoryinfo", |
|
|
|
|
|
cmtitle=self.title) |
|
|
|
|
|
info = result["query"]["pages"].values()[0]["categoryinfo"] |
|
|
|
|
|
return info[member_type] |
|
|
|
|
|
|
|
|
|
|
|
def _get_size_via_sql(self, member_type): |
|
|
|
|
|
"""Return the size of the category using SQL.""" |
|
|
query = "SELECT COUNT(*) FROM categorylinks WHERE cl_to = ?" |
|
|
query = "SELECT COUNT(*) FROM categorylinks WHERE cl_to = ?" |
|
|
title = self.title.replace(" ", "_").split(":", 1)[1] |
|
|
title = self.title.replace(" ", "_").split(":", 1)[1] |
|
|
if member_type == "size": |
|
|
if member_type == "size": |
|
@@ -111,14 +118,7 @@ class Category(Page): |
|
|
else: |
|
|
else: |
|
|
query += " AND cl_type = ?" |
|
|
query += " AND cl_type = ?" |
|
|
result = self.site.sql_query(query, (title, member_type[:-1])) |
|
|
result = self.site.sql_query(query, (title, member_type[:-1])) |
|
|
return list(result)[0] |
|
|
|
|
|
|
|
|
|
|
|
def _get_size_via_sql(self, member_type): |
|
|
|
|
|
"""Return the size of the category using SQL.""" |
|
|
|
|
|
result = self.site.api_query(action="query", prop="categoryinfo", |
|
|
|
|
|
cmtitle=self.title) |
|
|
|
|
|
info = result["query"]["pages"].values()[0]["categoryinfo"] |
|
|
|
|
|
return info[member_type] |
|
|
|
|
|
|
|
|
return list(result)[0][0] |
|
|
|
|
|
|
|
|
def _get_size(self, member_type): |
|
|
def _get_size(self, member_type): |
|
|
"""Return the size of the category.""" |
|
|
"""Return the size of the category.""" |
|
|