← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to lib/pulldown_subj/__init__.py

  • Committer: William Grant
  • Date: 2009-01-13 01:36:15 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:1123
Merge setup-refactor branch. This completely breaks existing installations;
every path (both filesystem and Python) has changed. Do not upgrade without
knowing what you are doing.

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/pulldown_subj/__init__.py
1
 
# IVLE - Informatics Virtual Learning Environment
2
 
# Copyright (C) 2007-2008 The University of Melbourne
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
 
 
18
 
# Module: Subject Pulldown
19
 
# Author: Matt Giuca
20
 
# Date:   7/7/2008
21
 
 
22
 
# Pluggable subject pulldown module.
23
 
# Grabs a student's subject list from a source specified in conf.py.
24
 
 
25
 
# Each module should have a get_subjects(login) function, which takes login
26
 
# and returns a list of (subject, semester) pairs (both strings), or None
27
 
# if the user can't be found. May also raise a SubjectError which is fatal.
28
 
 
29
 
import os
30
 
import sys
31
 
 
32
 
from subjecterror import SubjectError
33
 
import conf
34
 
import common.db
35
 
 
36
 
def get_subjects(login):
37
 
    """
38
 
    Looks up the student in whatever modules are available, using login.
39
 
    If successful, returns a list of (subject, semester) pairs (both strings).
40
 
    Raises a SubjectError if unsuccessful.
41
 
    """
42
 
    for modname, m in subj_modules:
43
 
        result = m(login)
44
 
        if result is not None:
45
 
            return result
46
 
    return []
47
 
 
48
 
def enrol_user(login, year=None):
49
 
    """
50
 
    Looks up the student in whatever modules are available.
51
 
    The pulldown does not tell us what year to enrol the student for, so the
52
 
    year may be specified (as a string). If unspecified, will enrol in the
53
 
    current year according to the system clock.
54
 
    If successful, enrols the student (in the database) in all subjects in the
55
 
    system. Subjects which the pulldown tells us the student is in, but which
56
 
    are not in the system are ignored.
57
 
    Does not unenrol the student from any subjects.
58
 
    Does not complain if the student is already enrolled in any subjects.
59
 
    Raises a SubjectError if the pulldown module says so.
60
 
 
61
 
    Returns the number of subjects the user was enrolled in (not including
62
 
    subjects outside the system, or subjects already enrolled).
63
 
    """
64
 
    count = 0
65
 
    db = common.db.DB()
66
 
    for subject, semester in get_subjects(login):
67
 
        count += db.add_enrolment(login, subject, semester, year)
68
 
    return count
69
 
 
70
 
# Allow imports to get files from this directory.
71
 
# Get the directory that this module (authenticate) is in
72
 
plugpath = os.path.split(sys.modules[__name__].__file__)[0]
73
 
# Add it to sys.path
74
 
sys.path.append(plugpath)
75
 
 
76
 
# Create a global variable "subj_modules", a list of (name, function object)s.
77
 
# This list consists of null_subj, plus the "get_subject" functions of all the
78
 
# plugin subject pulldown modules.
79
 
 
80
 
subj_modules = []
81
 
for modname in conf.subject_pulldown_modules.split(','):
82
 
    try:
83
 
        mod = __import__(modname)
84
 
    except ImportError:
85
 
        raise SubjectError("Internal error: "
86
 
            "Can't import subject pulldown module %s"
87
 
            % repr(modname))
88
 
    except ValueError:
89
 
        # If auth_modules is "", we may get an empty string - ignore
90
 
        continue
91
 
    try:
92
 
        subjfunc = mod.get_subjects
93
 
    except AttributeError:
94
 
        raise SubjectError("Internal error: Subject pulldown module %r has no "
95
 
            "'get_subjects' function" % modname)
96
 
    subj_modules.append((modname, subjfunc))