User:MacMedBot/Source

1. !usr/bin/python

"""

This is a script that will add {{oldprodfull}} to the talk pages

of all pages in a certain category that were PRODded in the past,

and do not currently have one.

Written by The Earwig:

With assistance from MacMed:

"""

1. Imports

import sys, urllib, re, datetime, codecs # System imports

import wikipedia, pagegenerators, catlib # Pywikipedia imports

import simplejson as json # JSON import

global site

site = wikipedia.getSite()

def main(): # Main function, parses the category given to it through the command line.

if sys.argv[1] == '--version':

print 'MacMedBot is currently in alpha testing, version 0.5.7'

sys.exit()

else:

try:

category = sys.argv[1] # Get the category from the user's input.

except IndexError:

print "Sorry; please specify a category to use."

exit()

if "Category:" not in category:

category = "Category:%s" % category

cat = catlib.Category(site, category)

generator = pagegenerators.CategorizedPageGenerator(cat, recurse = True) # Now create a generator for it.

for page in generator: # Iterate over said generator.

check(page.title()) # Process the page.

def check(pagename):

page = wikipedia.Page(site, pagename)

if page.namespace() != 0:

wikipedia.output(u"Not an article, skipping...")

return

else:

pagename1 = 'Talk:'+pagename

page = wikipedia.Page(site, pagename1)

wikipedia.output(u"\nChecking page %s for {{oldprodfull}}." % page.aslink())

oldtext = ""

try: # In case page does not exist.

oldtext = page.get() # To check for {{oldprodfull}}

except wikipedia.NoPage: # If page does not exist

wikipedia.output(u"Talk page does not exist, processing...")

process(pagename) # Process the page (no template)

except wikipedia.IsRedirectPage: # If page is a redirect

wikipedia.output(u"Talk page is a redirect, skipping...")

return # Do not process

oldtext2 = oldtext.lower() # Make everything lowercase

if "{{oldprodfull" not in oldtext2: # Look for {{oldprodfull}}

wikipedia.output(u"{{oldprodfull}} not found on talk page, processing...")

process(pagename) # If it's there, move on

else: #If it's not...

wikipedia.output(u"{{oldprodfull}} found on talk page, skipping...")

return # Do not process

def process(page): # Process function, plug in the pagename here to process.

page = wikipedia.Page(site, page)

if ifChecked(page.title()):

wikipedia.output(u"This page has already been checked, skipping...")

return

try:

page.get()

except Exception:

wikipedia.output(u"\nPage %s does not exist." % page.title())

return

wikipedia.output(u"\nChecking page %s." % page.aslink())

checked(page.title())

content, nextid = getRevision(page.title())

if "{{dated prod" in content:

wikipedia.output(u"There is PROD on the page but it has not yet expired.")

return

for x in range(50):

content, nextid = getRevision(page.title(), nextid)

if "{{dated prod" in content:

putTemplate(page.title(), content)

if not nextid:

wikipedia.output(u"No PROD template found, moving on...")

break

def checked(page):

wikipedia.output(u"Logging page...\nProcessing...")

f = codecs.open("edited_pages.txt", "a")

f.write("\n%s" % page)

f.close()

def ifChecked(page=""):

f = codecs.open("edited_pages.txt", "r")

checks = f.read()

page = "%s" % page

if page in checks:

return True

else:

return False

def getRevision(title, nextid=None):

params = {'action':'query', 'prop':'revisions', 'rvprop':'content', 'rvlimit':1, 'format':'json'}

params['titles'] = title

if nextid:

params['rvstartid'] = nextid

data = urllib.urlencode(params)

raw = urllib.urlopen("http://en.wikipedia.org/w/api.php", data)

res = json.loads(raw.read())

pageid = res['query']['pages'].keys()[0]

nextif = None

if 'query-continue' in res:

nextid = res['query-continue']['revisions']['rvstartid']

content = res['query']['pages'][pageid]['revisions'][0]['*']

return content, nextid

def putTemplate(pagename, content): # Put the {{oldprodfull}} template on the page specified.

page = wikipedia.Page(site, pagename) # Load the page.

try: # In case the page doesn't exist.

oldtext = page.get() # Get the old text, so we don't erase it.

except wikipedia.NoPage: # If the page doesn't exist...

oldtext = "" # Set this to an empty string.

oldtext2 = oldtext.lower()

wikipedia.output(u"Saving...")

prod = content.split("}}")[0]

nomreason = prod.split("concern = ")[1].split("|")[0]

addition = "\n{{oldprodfull|nomreason="+nomreason+"}}" # Whatever you're adding.

newtext = addition + oldtext # Create the new text, the save it.

# page.put(newtext, comment="(Bot): Adding {{oldprodfull}} to PRODded page.") # Use whatever edit summary you want here.

if __name__ == '__main__':

try:

main()

finally:

wikipedia.stopme()