ExpeditionSummaries.py

import pywikibot
from pywikibot import Category
import re, Expedition
import datetime
from UserListGenerator import *

class ExpeditionSummaries:
  '''
  This contains all important information about a given expedition
  '''
  def __init__(self, site, date, db):
    self.site = site
    self.date = date
    self.pageList = self._getAllCategoryPages()
    self.expedList = []
    self._putExpeditionSummaries(db)

  def _putExpeditionSummaries(self, db):
    allSummaries = []
    for page in self.pageList:
      if(re.match("\d{4}-\d{2}-\d{2} ([-0-9]{1,4} [-0-9]{1,4}|global)$", page.title())):
        pywikibot.output("Parsing page " + str(self.pageList.index(page)) + " of " + str(len(self.pageList)) + " : " + page.title())
        exped = Expedition.Expedition(page.site, page.title(), db)
        self.expedList.append(exped)
#        pywikibot.output(str(exped.subFormat()))
        allSummaries.append(exped.getExpeditionSummary())
        

    page = pywikibot.Page(self.site, "Template:Expedition_summaries/" + self.date)
    self._pageWrite(page, u'<noinclude>This page is automatically generated.  Any edits to this page will be overwritten by a bot.\n{| style="width: 100%; border: 1px solid grey; border-collapse:collapse;" cellpadding="5" cellspacing="0" border="1"</noinclude>\n|-\n|' + "\n|-\n|".join(allSummaries) + '\n<noinclude>\n|-\n|}\n</noinclude>')
    # if(datetime.date.today().isoformat() <= self.date):
    self._datePageWrite()

  def _pageWrite(self, page, text):
    if(self._checkBanana() == 0):
#      pywikibot.output("Would put page with " + text)
      try:
        old_text = page.get()
      except:
        page.put(text, u"Ook.")
      else:
        if (text == old_text):
          pywikibot.output("Page " + page.title() + " has not changed, skipping")
        else:
          page.put(text, u"Ook.")
    
  def _datePageWrite(self):
    pageText = u"<noinclude>{{date navigation}}</noinclude>\n"
    pageText += u"{{auto coordinates|" + self.date + "}}\n"
    pageText += u"{{auto gallery2|" + self.date + "}}\n"
    pageText += u"<noinclude>{{expedition summaries|" + self.date + "}}</noinclude>\n"

    page = pywikibot.Page(self.site, self.date)
    if(not page.exists()):
      page = pywikibot.Page(self.site, self.date)
      self._pageWrite(page, pageText)

  def _checkBanana(self):
    checkPage = pywikibot.Page(self.site, "User:AperfectBot")
    checkText = checkPage.get(True)
    checkRegex = getSectionRegex(checkText, "distraction banana", False).strip()
    if(len(checkRegex) == 0):
      return 0
    else:
      return 1

  def getSubFormats(self, format = None, user = None, oldText = None, grat = None):
    formats = {}
    if (oldText == None):
      oldText = {}
    for exped in self.expedList:
      resultText = None
      if exped.getPagename() in oldText:
        resultText = exped.subFormat(format, user, oldText[exped.getPagename()], grat = grat)
      else:
        resultText = exped.subFormat(format, user, grat = grat)
      if (resultText != None):
        formats[exped.getPagename()] = resultText

    return formats

  def _getAllCategoryPages(self):
    page = pywikibot.Page(self.site, "Category:Meetup on " + self.date)
    pageText = u"[[Category:Meetup in " + re.sub("-\d{2}$","",self.date) + u"]]"
    self._pageWrite(page, pageText)

    cat = Category(self.site, "Meetup on " + self.date)
    articleList = list(cat.articles())
    return articleList