Global message delivery/Configuration

deliverybot-3.py

#! /usr/bin/env python
# Public domain; MZMcBride; 2011

import codecs
import os
import re
import time
import wikitools
import config

# Define variables
username = config.username
base_page = config.base_page
access_list = base_page + '/' + config.access_list
log = base_page + '/' + config.log
status_page = base_page + '/' + config.status
spam = base_page + '/' + config.spam

# Create a home wiki (Meta-Wiki) object
home_wiki = wikitools.Wiki('http://meta.wikimedia.org/w/api.php'); home_wiki.setMaxlag(-1)

# Functions to do various tasks
def kill_self():
    os._exit(0)

def check_status(home_wiki, page):
    status_page_obj = wikitools.Page(home_wiki, status_page, followRedir=False)
    return status_page_obj.getWikiText().lower()

def change_status(status_message):
    status_page_obj = wikitools.Page(home_wiki, status_page, followRedir=False)
    status_page_obj.edit(status_message, summary='Bot: Updated status.', bot=1)
    log.write('will edit with content: %s\n' % status_message)
    return

def top_user(home_wiki, page):
    params = {'action'  : 'query',
              'prop'    : 'revisions',
              'rvprop'  : 'user',
              'titles'  : page}
    req = wikitools.api.APIRequest(home_wiki, params)
    response = req.query()
    latest_editor = response[u'query'][u'pages'].popitem()[1][u'revisions'][0][u'user']
    return latest_editor

def trusted_users(home_wiki, page):
    trusted_users = []

    params = {'action'      : 'query',
              'prop'        : 'links',
              'titles'      : page,
              'pllimit'     : 500,
              'plnamespace' : 2}
    req = wikitools.api.APIRequest(home_wiki, params)
    response = req.query()
    users = response[u'query'][u'pages'].popitem()[1][u'links']
    for entry in users:
        trusted_users.append(entry[u'title'].split(':', 1)[1])
    return trusted_users

def retrieve_config(page):
    spam_page = wikitools.Page(home_wiki, spam, followRedir=False)
    spam_page_text = spam_page.getWikiText()
    spam_page_text_parts = re.search(r'''\<syntaxhighlight lang="text" >.*?
# KEY(.+)
# RECIPIENTS \(PAGE LIST\)(.+)
# SUBJECT(.+)
# BODY(.+)

, spam_page_text, re.I|re.U|re.M|re.DOTALL)

   return { 'key': spam_page_text_parts.group(1),
            'recip_page': spam_page_text_parts.group(2),
            'subject': spam_page_text_parts.group(3),
            'body': spam_page_text_parts.group(4),
            }

def read_keys():

   f = open('/home/mzmcbride/scripts/deliverybot/deliverybot-3-keys.txt', 'r')
   keys = f.read().strip('\n').split('\n')
   f.close()
   return list(keys)

def add_key(key):

   f = open('/home/mzmcbride/scripts/deliverybot/deliverybot-3-keys.txt', 'a')
   f.write('%s\n' % key)
   f.close()
   return

def edit_page(target_site, target_input_page):

   try:
       user_talk = wikitools.Page(target_site, target_input_page, followRedir=True)
   except KeyError:
       log.write('some issue: %s %s' % (target_input_page.decode('utf-8'), target_input_site))
       return
   try:
       page_text = user_talk.getWikiText()
   except:
       page_text = 
   try:
       if not re.search(r'()' % (username, current_key), page_text, re.I|re.U):
           user_talk.edit(text=body_text_final, summary=subject_line, section='new', bot=1, skipmd5=True)
           log.write('Edited: %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
       else:
           log.write('Skipped: %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
   except:
       time.sleep(2)
       try:
           try:
               page_text = user_talk.getWikiText()
           except:
               page_text = 
           if not re.search(r'()' % (username, current_key), page_text, re.I|re.U):
               user_talk.edit(text=body_text_final, summary=subject_line, section='new', bot=1, skipmd5=True)
               log.write('Edited: %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
           else:
               log.write('Skipped: %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
       except:
           log.write('WTF1 %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
           pass
       log.write('WTF2 %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
       pass
   return

def strip_cruft(str):

   str = re.sub(r'^(\s|\n)*', , str)
   str = re.sub(r'(\s|\n)*$', , str)
   return str

def parse_input_page(home_wiki, input_page):

   # Define two nasty regexen
   target_template_user_re = re.compile(r'\{\{\s*target\s*\|\s*user\s*=\s*(.+)\s*\|\s*site\s*=\s*(.+)\s*\}\}')
   target_template_page_re = re.compile(r'\{\{\s*target\s*\|\s*page\s*=\s*(.+)\s*\|\s*site\s*=\s*(.+)\s*\}\}')
   
   targets_list = []
   targets_obj = wikitools.Page(home_wiki, input_page, followRedir=False)
   targets_page_text = targets_obj.getWikiText()
   for line in targets_page_text.split('\n'):
       if target_template_user_re.search(line):
           input_target_user = target_template_user_re.search(line).group(1).strip()
           input_target_site = target_template_user_re.search(line).group(2).strip()
           if re.search(r'[A-Za-z.\-]', input_target_site):
               targets_list.append([input_target_site, 'User talk:' + input_target_user])
       elif target_template_page_re.search(line):
           input_target_user = target_template_page_re.search(line).group(1).strip()
           input_target_site = target_template_page_re.search(line).group(2).strip()
           if re.search(r'[A-Za-z.]', input_target_site):
               targets_list.append([input_target_site, input_target_user])
   targets_list = sorted(targets_list)
   return targets_list

def expand_wikitext(wiki, wikitext):

   params = {'action' : 'expandtemplates',
             'text'   : wikitext}
   req = wikitools.api.APIRequest(wiki, params)
   response = req.query()
   expanded_wikitext = response[u'expandtemplates']['*']
   return expanded_wikitext.encode('utf-8')

status = check_status(home_wiki, status_page)

log = codecs.open('/home/mzmcbride/scripts/deliverybot/deliverybot-3-log.txt', 'a', 'utf-8')

if status == 'start':

   home_wiki.login(config.username, config.password)
   top_user = top_user(home_wiki, spam)
   trusted_users = trusted_users(home_wiki, access_list)
   old_keys = read_keys()
   configuration = retrieve_config(spam)
   current_key = strip_cruft(configuration['key'])
   add_key(current_key)
   input_page = strip_cruft(configuration['recip_page'])
   targets_list = parse_input_page(home_wiki, input_page)
   subject_line = strip_cruft(configuration['subject'])
   body_text = strip_cruft(configuration['body'])
   body_text_final = body_text + '\n' % (username, current_key)

   log.write('status is fine, let\'s edit\n')
   if top_user in trusted_users:
       log.write('auth is fine, let\'s edit\n')
       if current_key not in old_keys or status == 'really start':
           log.write('key is fine, let\'s edit\n')
           if len(subject_line) > 245:
               log.write('subject line is too large, edit status page indicating so and die\n')
               change_status('Error: Subject line is too long')
               kill_self()
           else:
               log.write('use the page list\n')
               change_status('Running')
               target_sites = []
               for target in targets_list:
                   # Edit here!
                   target_input_site = target[0]
                   target_input_page = target[1]
                   log.write('processing %s\n' % target_input_page.decode('utf-8'))
                   if target_input_site not in target_sites:
                       target_sites.append(target_input_site)
                       log.write('target input site: ' + target_input_site + '\n')
                       target_site = wikitools.Wiki('http://%s/w/api.php' % target_input_site); target_site.setMaxlag(-1)
                       target_site.login(config.username, config.password)
                       log.write('logged in to: %s\n' % target_input_site)
                   if target_input_page.find('{') != -1:
                       log.write('need to expand the wikitext...\n')
                       target_input_page = expand_wikitext(target_site, target_input_page)
                       log.write('expanded wikitext: %s\n' % target_input_page.decode('utf-8'))
                   log.write('will edit %s at %s\n' % (target_input_page.decode('utf-8'), target_input_site))
                   edit_page(target_site, target_input_page)
               change_status('Completed run successfully')

       else:
           log.write('key is old, edit status page to indicate such and die\n')
           change_status('Error: Key is invalid')
           kill_self()
   else:
       log.write('user not authorized to use bot; edit status page to indicate such and die\n')
       change_status('Error: User not authorized to use bot' % access_list)
       kill_self()

1. Close the log!

log.close() </syntaxhighlight>

crontab

0,5,10,15,20,25,30,35,40,45,50,55 * * * * PYTHONPATH=$HOME/scripts python $HOME/scripts/deliverybot/deliverybot-3.py