jenkins-bot has submitted this change and it was merged.

Change subject: Simplify initialisation logic
......................................................................


Simplify initialisation logic

Change-Id: Ia8df25bc2d06c7a53c18e869aae87bac901321ac
---
M scripts/makecat.py
1 file changed, 12 insertions(+), 8 deletions(-)

Approvals:
  Nullzero: Looks good to me, approved
  jenkins-bot: Verified



diff --git a/scripts/makecat.py b/scripts/makecat.py
index cb2acf4..0b28870 100644
--- a/scripts/makecat.py
+++ b/scripts/makecat.py
@@ -235,10 +235,13 @@
     except IOError:
         # File does not exist
         excludefile = codecs.open(filename, 'w', encoding=mysite.encoding())
+
+    # Get parent categories in order to `removeparent`
     try:
         parentcats = workingcat.categories()
     except pywikibot.Error:
         parentcats = []
+
     # Do not include articles already in subcats; only checking direct subcats
     subcatlist = list(workingcat.subcategories())
     if subcatlist:
@@ -247,14 +250,12 @@
             artlist = list(cat.articles())
             for page in artlist:
                 checked[page] = page
-    list = [x for x in workingcat.articles()]
-    if list:
-        for pl in list:
-            checked[pl] = pl
-        list = pagegenerators.PreloadingGenerator(list)
-        for pl in list:
-            include(pl)
-    else:
+
+    # Fetch articles in category, and mark as already checked (seen)
+    # If category is empty, ask user if they want to look for pages
+    # in a diferent category.
+    articles = list(workingcat.articles(content=True))
+    if not articles:
         pywikibot.output(
             u"Category %s does not exist or is empty. Which page to start 
with?"
             % workingcatname)
@@ -263,6 +264,9 @@
             answer = workingcatname
         pywikibot.output(u'' + answer)
         pl = pywikibot.Page(mysite, answer)
+        articles = [pl]
+
+    for pl in articles:
         checked[pl] = pl
         include(pl)
 

-- 
To view, visit https://gerrit.wikimedia.org/r/169640
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings

Gerrit-MessageType: merged
Gerrit-Change-Id: Ia8df25bc2d06c7a53c18e869aae87bac901321ac
Gerrit-PatchSet: 1
Gerrit-Project: pywikibot/core
Gerrit-Branch: master
Gerrit-Owner: John Vandenberg <[email protected]>
Gerrit-Reviewer: Ladsgroup <[email protected]>
Gerrit-Reviewer: Merlijn van Deen <[email protected]>
Gerrit-Reviewer: Nullzero <[email protected]>
Gerrit-Reviewer: jenkins-bot <>

_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits

Reply via email to