# HG changeset patch
# User Marcin Cieślak <saper@saper.info>
# Date 1438806221 0
#      Wed Aug 05 20:23:41 2015 +0000
# Node ID d9c8a55117a78315fc4c66cd274f972a936ee07f
# Parent  7c8e88ed5b67b3c4d5fbb023c597b4c141f77581
Add categories for bot-generated pages

Otherwise our pages end up in
[[Special:Uncategorized pages]]

Pointed-out-by: Openbk

diff --git a/plwiki/chemia/comp.py b/plwiki/chemia/comp.py
--- a/plwiki/chemia/comp.py
+++ b/plwiki/chemia/comp.py
@@ -14,6 +14,9 @@
 mywiki = logging.getLogger('chemia')
 
 MAINPAGENAME = u"Związki chemiczne według pierwiastków/Wykaz"
+MAINPAGECAT = u"Kategoria:Wikiprojekt Chemia"
+ERRORPAGENAME = MAINPAGENAME + u"/Błędy"
+ERRORPAGECAT = u"Kategoria:Problemy z hasłami chemicznymi"
 MAINTEMPLATENAME =u"Szablon:Związek chemiczny infobox" 
 
 def CompoundGenerator(site, template):
@@ -163,7 +166,7 @@
 
 
 def processwikilive(elements):
-	errpage = wikipedia.Page(site, MAINPAGENAME + u"/Błędy")
+	errpage = wikipedia.Page(site, ERRORPAGENAME)
 
 	reallybadfile = open("bad", "w")
 	(compounds, badlist) = extractwiki(site, reallybadfile, CompoundGenerator(site, MAINTEMPLATENAME), elements)
@@ -175,6 +178,8 @@
 		elements.feed(comp, comp.elements)
 	compfile.close()
 		
+	if badlist:
+		badlist.extend(["", "[[%s]]" % ERRORPAGECAT])
 	errpage.put(u"".join(badlist), 
 		comment=u"Robot zapisuje błędy wynikłe podczas tworzenia [[%s]]" % MAINPAGENAME, minorEdit=False, botflag=False)
 	mywiki.info(u"Pierwiastki zostały przetworzone")
@@ -200,6 +205,8 @@
 	else:
 		processwikilive(elements)
 	out = formatter(elements)
+	if out:
+		out.extend(["", "[[%s]]" % MAINPAGECAT])
 	outpage.put(u"\n".join(out), comment=u"Robot tworzy stronę na podstawie [[Specjalna:Linkujące/%s|zawartości infoboksów]]" % MAINTEMPLATENAME, 
 		minorEdit=False, botflag=False)