Jump to content

User:LemmeyBOT/RefHistoryFix2

From Wikipedia, the free encyclopedia
#!/usr/bin/python
# -*- coding: utf-8  -*-
""""
Bot:LemmeyBOT
FileName:RefHistoryFix2.py
Author: Lemmey 3-1-2008
Tagline:Fix articles, not conversations about conversations about records about talk pages about ...
Purpose:Restores references lost due to vandalism, bad editors, massive changes.
Method:Looks to all the linked articles for the lost reference.
""""
__version__ = '$Id: basic.py 3998 2007-08-07 20:28:27Z wikipedian $'
import wikipedia, catlib
import pagegenerators
import sys
import BeautifulSoup
import urllib
import re
from datetime import date
message = ""
autoconfirm = False
docuReplacements = {
    '&params;': pagegenerators.parameterHelp
}

rtnln = u'''

'''

class BasicBot:

    msg = {
        'de': u'Bot: Ändere ...',
        'en': u'Restored missing content of named reference using article history',
    }

    def __init__(self, generator, debug):
        """
        Constructor. Parameters:
            * generator - The page generator that determines on which pages
                          to work on.
            * debug     - If True, doesn't do any real changes, but only shows
                          what would have been changed.
        """
        self.generator = generator
        self.debug = debug

    def run(self):
        # Set the edit summary message
        wikipedia.setAction(wikipedia.translate(wikipedia.getSite(), self.msg))
        for page in self.generator:
            self.treat(page)

    def treat(self, page):
        global message
        """
        Loads the given page, does some changes, and saves it.
        """
        try:
            text = page.get(throttle = False,get_redirect=True)
            origional = text
        except wikipedia.NoPage:
            wikipedia.output(u"Page %s does not exist; skipping." % page.aslink())
            return
        except wikipedia.IsRedirectPage:
            wikipedia.output(u"Page %s is a redirect; skipping." % page.aslink())
            return
        except wikipedia.LockedPage:
            wikipedia.output(u"Page %s is locked; skipping." % page.aslink())
            return
        showtext = text
        try:
            text=fixBroken(text,page)
            showtext = text
        except wikipedia.IsRedirectPage:
            Ignore(page.title())

def save_page(page,oldtext,newtext,ref,message,confirm=True):
    print "Message: ",message
    wikipedia.showDiff(oldtext, newtext)
    if confirm:
        choice = wikipedia.inputChoice(u'Do you want to accept these changes?', ['Yes', 'No'], ['y', 'N'], 'N')
    else:
        choice = 'y' #HARD CODED
    text = newtext
    if choice == 'y':
        try:
            page.put(newtext,minorEdit=True,comment=message)
        except wikipedia.EditConflict:
            wikipedia.output(u'Skipping %s because of edit conflict')
        except wikipedia.SpamfilterError, error:
            wikipedia.output(u'Cannot change %s because of spam blacklist entry %s')
            newtext=oldtext.replace(ref,u'{{subst:fact-now}}')
            message = "original reference: "+ref+" is a blacklisted source - replaced with a Fact tag"
            text = newtext
            try:
                text = save_page(page,oldtext,newtext,ref,message,confirm=confirm)
            except:
                pass
    return text

def broken_refs(text):
    flag = False
    a=Get_Named_Ref_Whole(text)
    b=Get_Named_Ref_Stubs(text)
    
    stub_array=[]
    for match in b:
        name = match.group()
        s=re.compile('"[\w- ]+"',re.I | re.S)
        name = s.finditer(name)
        for mname in name:
            if mname.group() not in stub_array: stub_array.append( mname.group())

    whole_array=[]
    for match in a:
        name = match.group()
        s=re.compile('"[\w- ]+"',re.I | re.S)
        name = s.finditer(name)
        for mname in name:
            if mname.group() not in whole_array: whole_array.append( mname.group())
            
    for stub in stub_array:
        if stub not in whole_array:
            flag = True
            
    return flag

def get_lost_stubs(text):
    flag = False
    a=Get_Named_Ref_Whole(text)
    b=Get_Named_Ref_Stubs(text)
    
    stub_array=[]
    stub_list=[]
    for match in b:
        name = match.group()
        s=re.compile('= *[^/]+? */',re.I | re.S)
        search = s.search(name)
        if search != None:
            tag = search.group()
            tag = tag[1:-1]
            tag = tag.strip()
            if name not in stub_list: 
                stub_array.append(tag)
                stub_list.append(name)

    whole_array=[]
    for match in a:
        name = match.group()
        s=re.compile('= *[^/]+? ?>',re.I | re.S)
        search = s.search(name)
        if search != None:
            tag = search.group()
            tag = tag[1:-1]
            tag = tag.strip()
            whole_array.append(tag)

    lost_stubs=[]
    for x in range(0,len(stub_array)):
        stub=stub_array[x]
        if stub not in whole_array and (stub,stub_list[x]) not in lost_stubs:
            lost_stubs.append((stub,stub_list[x]))

    return lost_stubs

def fixBroken(text,page):
    b=get_lost_stubs(text)
    global message, autoconfirm

    wikipedia.output(u"\n\n>>> \03{lightpurple}%s\03{default} <<<" % page.title())

    pl=Get_Page_List(page)

    for item in b:
        stub = item[0]
        ref = item[1]
        for pageTitle in pl:
            try:
                page2 = wikipedia.Page(wikipedia.getSite(), pageTitle)
                text2 = page2.get(throttle = False,get_redirect=True)
                a=None
                FOUND = False
                a=Get_Specific_Named_Whole(text2,stub) 
                if a!= None and Check_for_Blank(stub,a.group()) == False:
                    newtext=text.replace(ref,a.group(),1)
                    if newtext!=text:
                        FOUND=True
                        wikipedia.output(u"\n\n>>> \03{lightpurple}%s\03{default} <<<" % page.title())
                        wikipedia.output(u"\n\n>>> \03{lightpurple}%s\03{default} <<<" % pageTitle)
                        text = save_page(page,text,newtext,ref,u'Restored missing content of named reference '+ref+' using identical reference in [['+pageTitle+']]',confirm=autoconfirm)  
                if FOUND: break
            except:
                pass
    return text

def Get_Page_List(page):
    linked = page.linkedPages()
    getlist=[]
    if page.isTalkPage():
        art=str(page.title()[5:])
        ##print art
        getlist.append(art)
        
    for article in linked:
        name = str(article).strip("[]")
        if name not in getlist:getlist.append(name)
    return getlist


def Check_for_Blank(name,tag):
    pattern='< ?ref ?name *= *'+name+' ?> *< ?/ ?ref ?>'
    a=re.compile(pattern,re.I | re.S)##<ref name = "larry">
    search = a.search(tag)
    if search!= None:
        return True
    else:
        return False

def Get_Specific_Named_Whole(text2,name):
    array=[]
    ##Named refs without closings
    pattern='< ?ref ?name *= *'+name+' ?>.+?< ?/ ?ref ?>'
    a=re.compile(pattern,re.I | re.S)##<ref name = "larry">
    search = a.search(text2)
    return search

def Get_Named_Ref_Stubs(text):
    array=[]
    ##Named refs with closings
    ##pattern = '< *ref *name *= *[\w "-]+? */ *>'
    pattern = '< *ref *name *= *[^>]+? */ *>'
    b=re.compile(pattern,re.I | re.S)##<ref name = "larry"/>
    iterator = b.finditer(text)
    return iterator

def Get_Named_Ref_Whole(text):
    array=[]
    ##Named refs without closings
    pattern = '< ?ref ?name *= *[^/]+? ?>.+?< ?/ ?ref ?>'
    a=re.compile(pattern,re.I | re.S)##<ref name = "larry">
    iterator = a.finditer(text)
    return iterator

def Ignore(article):
    f=open('list.txt', 'a')
    f.write(article + '\n')
    f.close()

def main():
    genFactory = pagegenerators.GeneratorFactory()
    gen = None
    pageTitleParts = []
    debug = False

    for arg in wikipedia.handleArgs():
        if arg.startswith("-debug"):
            debug = True
        else:
            generator = genFactory.handleArg(arg)
            if generator:
                gen = generator
            else:
                pageTitleParts.append(arg)

    if pageTitleParts != []:
        pageTitle = ' '.join(pageTitleParts)
        page = wikipedia.Page(wikipedia.getSite(), pageTitle)
        gen = iter([page])

    if gen:
        gen = pagegenerators.PreloadingGenerator(gen)
        bot = BasicBot(gen, debug)
        bot.run()
    else:
        wikipedia.showHelp()

if __name__ == "__main__":
    try:
        main()
    finally:
        wikipedia.stopme()