daisy.py

# -*- coding: iso-8859-1 -*-

# Skapat av Christian Davén 2004

import urllib
import re
import timetable
import calendar
import error
import settings
import htmlentitydefs
from i18n import *

# -----------------------------------------------------------
class Conduit:
    "Loggar in och hämtar genererat schema från Daisy"
    
    def __init__(self, callback = None):
        self.callback = callback

    def getvCalendarData(self, courseids):
        getdata = "?generator=true"

        for id in courseids:
            getdata += "&momentid=" + str(id)

        try:
            generated = urllib.urlopen("http://daisy.it.kth.se/servlet/anstalld.schema.BoardSchema" +
                getdata).read()
            if self.callback: self.callback()
        except IOError:
            raise error.ReadError("Could not read from the timetable server")

        start = generated.find("/servlet/schema.ics")
        end = generated.find("\"", start)

        if start == -1 or end == -1 or generated[start:end].endswith("null"):
            raise error.DataError

        url = urllib.urlopen("http://daisy.it.kth.se" + generated[start:end])
        if self.callback: self.callback()

        return url.readlines()

    def getCourses(self, callback = None):
        """
            Hämtar alla valbara kurser på IT-universitetet för aktuell termin.
            Callback används för att visa förloppet för användaren
        """

        vt = "1"
        ht = "2"
        today = calendar.Date()
        year = today.getYear()
        courses = []

        # efter oktober hämtas aktuell hösttermin plus kommande vårtermin
        if today.getMonth() > 9:
            courses = self.getCoursesForPeriod(str(year) + ht, callback)
            courses += self.getCoursesForPeriod(str(year + 1) + vt, callback)
        # före mars hämtas föregående års hösttermin plus aktuell vårtermin
        elif today.getMonth() < 3:
            courses = self.getCoursesForPeriod(str(year - 1) + ht, callback)
            courses += self.getCoursesForPeriod(str(year) + vt, callback)
        # mellan mars och oktober hämtas vårtermin och hösttermin för samma år
        else:
            courses = self.getCoursesForPeriod(str(year) + vt, callback)
            courses += self.getCoursesForPeriod(str(year) + ht, callback)

        return courses

    def getCoursesForPeriod(self, period, callback):
        try:
            url = urllib.urlopen("http://www.it.kth.se/schema.html?termin=" + period +
                "&program=0&institution=0&Visa=Visa")
        except IOError:
            raise error.ReadError("Could not read from www.it.kth.se")

        if callback: callback()
        input = url.read()

        if len(input) == 0:
            raise error.DataError("Received no data from the timetable server")

        all = re.findall("value=\"(\d{3,})\"[^*\wÅÄÖ]*([^\r\n,]*),\s([^\r\n]*)", self.descape(input))
        courses = []
        for a in all:
            courses.append(timetable.Course(code=a[1], name=a[2], id=a[0], term=period))

        if callback: callback()
        return courses

    def descape(self, text):
        "Översätter från html till specialtecken"

        text = text.replace("&#38;", "&")
        return re.sub("&(\w+?);", self.descape_entity, text)

    def descape_entity(self, rx):
        "Används endast av descape() via re.sub()"

        if rx.group(1) in htmlentitydefs.entitydefs:
            return htmlentitydefs.entitydefs[rx.group(1)]
        else:
            return rx.group(0)


# -----------------------------------------------------------
class SummaryParser:
    """
        Tolkar Daisy-systemets 'summary'-fält i vCalendar-exporterade
        scheman och extraherar data.

        Denna "parsning" bygger på att summary-fältet i Daisys
        genererade scheman alltid har samma struktur -- förändras
        den måste denna kod förändras.

        ex. "Föreläsning 8 Sal E - Digital elektro"
    """
    
    def __init__(self):
        self.rxtype = re.compile("^(.*?)(  | [0-9A-ZÅÄÖ])")
        self.rxseries = re.compile("^[A-ZÅÄÖ][a-zåäö ]+ (\d{1,2}) ")
        self.rxgroup = re.compile("grp (\d+) ")
        self.rxcourse = re.compile("- (.+)$")

    # -----
    #  Observerade händelsetyper i Daisy:
    #  (anställda kan lägga in fritext som typ)

    #  Föreläsning

    #  Övning
    #  Lektion
    #  Seminarium
    #  Laboration
    #  Designseminarium
    
    #  Kontrollskrivning
    #  Dugga
    #  Salsskrivning

    #  Redovisning
    #  Information
    #  Introduktion
    #  Handledning
    #  SI-möte
    # -----

    def parse(self, summary):
        data = {}
        data["type"] = self.extractType(summary)
        data["seriesno"] = self.extractSeriesNo(summary)
        data["group"] = self.extractGroup(summary)
        data["course"] = self.extractCourse(summary)
        data["location"] = ""
        return data
    
    def extractType(self, text):
        "Hittar händelsens typ (ex. Föreläsning)"
        
        type = ""
        rx = self.rxtype.search(text)
        if rx:
            type = rx.group(1)
            
        return type

    def extractSeriesNo(self, text):
        "Hittar händelsens löpnummer (ex. (Lektion) 3)"
        
        number = 0
        rx = self.rxseries.search(text)
        if rx:
            number = int(rx.group(1))
                
        return number

    def extractGroup(self, text):
        "Hittar eventuellt gruppnummer"
        
        group = ""
        rx = self.rxgroup.search(text)
        if rx:
            group = rx.group(1)
        
        return group
        
    def extractCourse(self, text):    
        "Hittar kursbeteckningen"
        
        code = ""
        # kan innehålla alla tecken, ex. "SPÖK , kvällskurs" och "*:59/2I1042/2I4033"
        rx = self.rxcourse.search(text)
        if rx:
            code = rx.group(1)
        
        return code