earwigbot/bot/watcher_logic.py

# -*- coding: utf-8  -*-

"""
EarwigBot's IRC Watcher Logic

This file contains (configurable!) rules that EarwigBot's watcher uses after it
recieves an event from IRC.

This should, ideally, be in config.json somehow, but Python code makes more
sense for this sort of thing... so...
"""

import re

from wiki import task_manager as tasks

afc_prefix = "wikipedia( talk)?:(wikiproject )?articles for creation"

# compile some regexps used when finding specific events
r_page = re.compile(afc_prefix)
r_ffu = re.compile("wikipedia( talk)?:files for upload")
r_move1 = re.compile("moved \[\[{}".format(afc_prefix))
r_move2 = re.compile("moved \[\[(.*?)\]\] to \[\[{}".format(afc_prefix))
r_moved_pages = re.compile("^moved \[\[(.*?)\]\] to \[\[(.*?)\]\]")
r_delete = re.compile("deleted \"\[\[{}".format(afc_prefix))
r_deleted_page = re.compile("^deleted \"\[\[(.*?)\]\]")
r_restore = re.compile("restored \"\[\[{}".format(afc_prefix))
r_restored_page = re.compile("^restored \"\[\[(.*?)\]\]")
r_protect = re.compile("protected \"\[\[{}".format(afc_prefix))

def process(rc):
    """Given an RC() object, return a list of channels to report this event to.
    Also, start any wiki bot tasks within this function if necessary."""
    chans = set()  # channels to report this message to
    page_name = rc.page.lower()
    comment = rc.comment.lower()
    
    if "!earwigbot" in rc.msg.lower():
        chans.update(("##earwigbot", "#wikipedia-en-afc"))
        
    if r_page.search(page_name):
        tasks.start_task("afc_statistics", action="process_edit", page=rc.page)
        tasks.start_task("afc_copyvios", action="process_edit", page=rc.page)
        chans.add("#wikipedia-en-afc")
        
    elif r_ffu.match(page_name):
        chans.add("#wikipedia-en-afc")
        
    elif page_name.startswith("template:afc submission"):
        chans.add("#wikipedia-en-afc")
    
    elif rc.flags == "move" and (r_move1.match(comment) or
            r_move2.match(comment)):
        p = r_moved_pages.findall(rc.comment)[0]
        tasks.start_task("afc_statistics", action="process_move", pages=p)
        tasks.start_task("afc_copyvios", action="process_move", pages=p)
        chans.add("#wikipedia-en-afc")
    
    elif rc.flags == "delete" and r_delete.match(comment):
        p = r_deleted_page.findall(rc.comment)[0]
        tasks.start_task("afc_statistics", action="process_delete", page=p)
        tasks.start_task("afc_copyvios", action="process_delete", page=p)
        chans.add("#wikipedia-en-afc")
    
    elif rc.flags == "restore" and r_restore.match(comment):
        p = r_restored_page.findall(rc.comment)[0]
        tasks.start_task("afc_statistics", action="process_restore", page=p)
        tasks.start_task("afc_copyvios", action="process_restore", page=p)
        chans.add("#wikipedia-en-afc")
    
    elif rc.flags == "protect" and r_protect.match(comment):
        chans.add("#wikipedia-en-afc")

    return chans