protected_areas.py

#! /usr/bin/env python
import mwclient
import sys
from theobot import password

# CC-BY-SA Theopolisme
# Task 8 on [[User:Theo's Little Bot]]

def sokay(donenow):
	"""Simple function to check checkpage.
	This function calls a sub-function
	of the theobot.bot module, checkpage().
	"""
	if donenow % 5 == 0:
		if bot.checkpage("User:Theo's Little Bot/disable/protected areas") == True:
			return True
		else:
			return False
	else:
		return True

def pages_in_cat(cat):
	cat = mwclient.listing.Category(site, cat)
	x = []
	for item in cat:
		x.append(item.page_title)
	return x

def generate_wikicode(year):
	var = """{{Year nav topic3|""" + year + """|protected areas established}}

This is a list of [[protected area]]s established in """ + year + """.

{| class="wikitable sortable"
|-
! Name
! Country
! data-sort-type="numeric" |Area (ha)
|-"""
	
	cats = pages_in_cat("Category:Protected areas established in " + year)

	for cat in cats:
		var = var + "\n| [[" + cat + """]] ||  ||\n|-"""

	var = var + """\n|}

==See also==
*[[""" + year + """ in the environment]]

[[Category:Protected areas established in """ + year + """| ]]"""

	return var
	
def main():
	global site
	site = mwclient.Site('en.wikipedia.org')
	site.login(password.username, password.password)

	years_to_do = []

	for item in mwclient.listing.Category(site, 'Category:Protected areas by year of establishment'):
		if "Category" in str(item):
			try:
				int(item.page_title.split()[-1])
				year = item.page_title.split()[-1]
				years_to_do.append(year)
			except ValueError:
				print "Item " + str(item.page_title.split()[-1]) + " skipped."
	
	donenow = 5

	for year in years_to_do:
		if sokay(donenow) == True:
			donenow += 1
			pagename = "List of protected areas established in " + year
			if site.Pages[pagename].edit() == '':
				page_save_text = generate_wikicode(year)
				page = site.Pages[pagename]
				print "Saving page " + pagename + "."
				page.save(text,summary="Creating per [[WP:CLT]] for [[:Category:Protected areas by year of establishment]] ([[WP:BOT|bot]] - [[User:Theo's Little Bot/disable/protected areas|disable]])")
			else:
				print "Page already created, skipping."
		else:
			print "Bot was disabled. Exiting."
			sys.exit()
	
if __name__ == '__main__':
   main()