← Back to branch summary

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

« back to all changes in this revision

Viewing changes to lib/common/user.py

  • Committer: mattgiuca
  • Date: 2008-01-12 15:35:53 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:201
Added "test" directory.
Added make_date_test.py, a short script I wrote to test the date format
algorithm committed in the previous revision.

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/common/user.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: User
19
 
# Author: Matt Giuca
20
 
# Date:   19/2/2008
21
 
 
22
 
# Provides a User class which stores the login details of a particular IVLE
23
 
# user. Objects of this class are expected to be stored in the request
24
 
# session.
25
 
 
26
 
import common.caps
27
 
 
28
 
# Similar to db.login_fields_list but contains a different set of fields.
29
 
# The User object does not contain all the fields.
30
 
user_fields_required = frozenset((
31
 
    "login", "fullname", "role", "state", "unixid"
32
 
))
33
 
user_fields_list = (
34
 
    "login", "state", "unixid", "email", "nick", "fullname",
35
 
    "role", "studentid", "acct_exp", "pass_exp", "svn_pass",
36
 
    "local_password"
37
 
)
38
 
# Fields not included: passhash, last_login
39
 
 
40
 
class UserException(Exception):
41
 
    pass
42
 
 
43
 
class User(object):
44
 
    """
45
 
    Stores the login details of a particular IVLE user.
46
 
    Its fields correspond to (most of) the fields of the "login" table of the
47
 
    IVLE db.
48
 
    All fields are always present, but some may be None.
49
 
    """
50
 
    __slots__ = user_fields_list
51
 
    def __init__(self, **kwargs):
52
 
        # XXX Will ignore unknown fields instead of erroring
53
 
        if "rolenm" in kwargs and "role" not in kwargs:
54
 
            kwargs['role'] = common.caps.Role(kwargs['rolenm'])
55
 
        if "passhash" in kwargs and "local_password" not in kwargs:
56
 
            kwargs['local_password'] = kwargs['passhash'] is not None
57
 
        for r in user_fields_list:
58
 
            if r in kwargs:
59
 
                self.__setattr__(r, kwargs[r])
60
 
            elif r in user_fields_required:
61
 
                # Required argument, not specified
62
 
                raise TypeError("User: Required field %s missing" % repr(r))
63
 
            else:
64
 
                # Optional arguments
65
 
                if r == "nick":
66
 
                    try:
67
 
                        self.nick = kwargs['fullname']
68
 
                    except KeyError:
69
 
                        raise TypeError("User: Required field "
70
 
                            "'fullname' missing")
71
 
                else:
72
 
                    self.__setattr__(r, None)
73
 
    def __repr__(self):
74
 
        items = ["%s=%s" % (r, repr(self.__getattribute__(r)))
75
 
            for r in user_fields_list]
76
 
        return "User(" + ', '.join(items) + ")"
77
 
    def __iter__(self):
78
 
        """Iteration yielding field:value pairs.
79
 
        (Allows the "dict" function to work on Users)
80
 
        """
81
 
        for r in user_fields_list:
82
 
            yield (r, self.__getattribute__(r))
83
 
 
84
 
    def hasCap(self, capability):
85
 
        """Given a capability (which is a Role object), returns True if this
86
 
        User has that capability, False otherwise.
87
 
        """
88
 
        return self.role.hasCap(capability)
89
 
 
90
 
    def pass_expired(self):
91
 
        """Determines whether the pass_exp field indicates that
92
 
           login should be denied.
93
 
        """
94
 
        fieldval = self.pass_exp
95
 
        return fieldval is not None and time.localtime() > fieldval
96
 
    def acct_expired(self):
97
 
        """Determines whether the acct_exp field indicates that
98
 
           login should be denied.
99
 
        """
100
 
        fieldval = self.acct_exp
101
 
        return fieldval is not None and time.localtime() > fieldval