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

« back to all changes in this revision

Viewing changes to ivle/database.py

  • Committer: Matt Giuca
  • Date: 2009-04-25 15:04:46 UTC
  • Revision ID: matt.giuca@gmail.com-20090425150446-f6z2k8ogs8kgyh1y
Tags: 0.1.9.12
ivle.chat, ivle.database, ivle.makeuser: Replaced use of md5 library with
    hashlib to avoid Python 2.6 producing DeprecationWarnings.
    Also use hexdigest() instead of digest().encode('hex').

Show diffs side-by-side

added added

removed removed

Lines of Context:
24
24
It also provides miscellaneous utility functions for database interaction.
25
25
"""
26
26
 
27
 
import md5
 
27
import hashlib
28
28
import datetime
29
29
 
30
30
from storm.locals import create_database, Store, Int, Unicode, DateTime, \
31
31
                         Reference, ReferenceSet, Bool, Storm, Desc
 
32
from storm.exceptions import NotOneError, IntegrityError
32
33
 
33
34
import ivle.conf
34
 
import ivle.caps
 
35
from ivle.worksheet.rst import rst
35
36
 
36
37
__all__ = ['get_store',
37
38
            'User',
38
39
            'Subject', 'Semester', 'Offering', 'Enrolment',
39
40
            'ProjectSet', 'Project', 'ProjectGroup', 'ProjectGroupMembership',
 
41
            'Assessed', 'ProjectSubmission', 'ProjectExtension',
40
42
            'Exercise', 'Worksheet', 'WorksheetExercise',
41
43
            'ExerciseSave', 'ExerciseAttempt',
42
 
            'AlreadyEnrolledError'
 
44
            'TestCase', 'TestSuite', 'TestSuiteVar'
43
45
        ]
44
46
 
45
47
def _kwarg_init(self, **kwargs):
53
55
    """
54
56
    Returns the Storm connection string, generated from the conf file.
55
57
    """
56
 
    return "postgres://%s:%s@%s:%d/%s" % (ivle.conf.db_user,
57
 
        ivle.conf.db_password, ivle.conf.db_host, ivle.conf.db_port,
58
 
        ivle.conf.db_dbname)
 
58
 
 
59
    clusterstr = ''
 
60
    if ivle.conf.db_user:
 
61
        clusterstr += ivle.conf.db_user
 
62
        if ivle.conf.db_password:
 
63
            clusterstr += ':' + ivle.conf.db_password
 
64
        clusterstr += '@'
 
65
 
 
66
    host = ivle.conf.db_host or 'localhost'
 
67
    port = ivle.conf.db_port or 5432
 
68
 
 
69
    clusterstr += '%s:%d' % (host, port)
 
70
 
 
71
    return "postgres://%s/%s" % (clusterstr, ivle.conf.db_dbname)
59
72
 
60
73
def get_store():
61
74
    """
76
89
    login = Unicode()
77
90
    passhash = Unicode()
78
91
    state = Unicode()
79
 
    rolenm = Unicode()
 
92
    admin = Bool()
80
93
    unixid = Int()
81
94
    nick = Unicode()
82
95
    pass_exp = DateTime()
88
101
    studentid = Unicode()
89
102
    settings = Unicode()
90
103
 
91
 
    def _get_role(self):
92
 
        if self.rolenm is None:
93
 
            return None
94
 
        return ivle.caps.Role(self.rolenm)
95
 
    def _set_role(self, value):
96
 
        if not isinstance(value, ivle.caps.Role):
97
 
            raise TypeError("role must be an ivle.caps.Role")
98
 
        self.rolenm = unicode(value)
99
 
    role = property(_get_role, _set_role)
100
 
 
101
104
    __init__ = _kwarg_init
102
105
 
103
106
    def __repr__(self):
114
117
            return None
115
118
        return self.hash_password(password) == self.passhash
116
119
 
117
 
    def hasCap(self, capability):
118
 
        """Given a capability (which is a Role object), returns True if this
119
 
        User has that capability, False otherwise.
120
 
        """
121
 
        return self.role.hasCap(capability)
 
120
    @property
 
121
    def display_name(self):
 
122
        return self.fullname
122
123
 
123
124
    @property
124
125
    def password_expired(self):
130
131
        fieldval = self.acct_exp
131
132
        return fieldval is not None and datetime.datetime.now() > fieldval
132
133
 
 
134
    @property
 
135
    def valid(self):
 
136
        return self.state == 'enabled' and not self.account_expired
 
137
 
133
138
    def _get_enrolments(self, justactive):
134
139
        return Store.of(self).find(Enrolment,
135
140
            Enrolment.user_id == self.id,
184
189
        '''A sanely ordered list of all of the user's enrolments.'''
185
190
        return self._get_enrolments(False) 
186
191
 
 
192
    def get_projects(self, offering=None, active_only=True):
 
193
        '''Return Projects that the user can submit.
 
194
 
 
195
        This will include projects for offerings in which the user is
 
196
        enrolled, as long as the project is not in a project set which has
 
197
        groups (ie. if maximum number of group members is 0).
 
198
 
 
199
        Unless active_only is False, only projects for active offerings will
 
200
        be returned.
 
201
 
 
202
        If an offering is specified, returned projects will be limited to
 
203
        those for that offering.
 
204
        '''
 
205
        return Store.of(self).find(Project,
 
206
            Project.project_set_id == ProjectSet.id,
 
207
            ProjectSet.max_students_per_group == None,
 
208
            ProjectSet.offering_id == Offering.id,
 
209
            (offering is None) or (Offering.id == offering.id),
 
210
            Semester.id == Offering.semester_id,
 
211
            (not active_only) or (Semester.state == u'current'),
 
212
            Enrolment.offering_id == Offering.id,
 
213
            Enrolment.user_id == self.id)
 
214
 
187
215
    @staticmethod
188
216
    def hash_password(password):
189
 
        return md5.md5(password).hexdigest()
 
217
        return hashlib.md5(password).hexdigest()
190
218
 
191
219
    @classmethod
192
220
    def get_by_login(cls, store, login):
196
224
        """
197
225
        return store.find(cls, cls.login == unicode(login)).one()
198
226
 
 
227
    def get_permissions(self, user):
 
228
        if user and user.admin or user is self:
 
229
            return set(['view', 'edit', 'submit_project'])
 
230
        else:
 
231
            return set()
 
232
 
199
233
# SUBJECTS AND ENROLMENTS #
200
234
 
201
235
class Subject(Storm):
214
248
    def __repr__(self):
215
249
        return "<%s '%s'>" % (type(self).__name__, self.short_name)
216
250
 
 
251
    def get_permissions(self, user):
 
252
        perms = set()
 
253
        if user is not None:
 
254
            perms.add('view')
 
255
            if user.admin:
 
256
                perms.add('edit')
 
257
        return perms
 
258
 
 
259
    def active_offerings(self):
 
260
        """Return a sequence of currently active offerings for this subject
 
261
        (offerings whose semester.state is "current"). There should be 0 or 1
 
262
        elements in this sequence, but it's possible there are more.
 
263
        """
 
264
        return self.offerings.find(Offering.semester_id == Semester.id,
 
265
                                   Semester.state == u'current')
 
266
 
 
267
    def offering_for_semester(self, year, semester):
 
268
        """Get the offering for the given year/semester, or None."""
 
269
        return self.offerings.find(Offering.semester_id == Semester.id,
 
270
                               Semester.year == unicode(year),
 
271
                               Semester.semester == unicode(semester)).one()
 
272
 
217
273
class Semester(Storm):
218
274
    __storm_table__ = "semester"
219
275
 
220
276
    id = Int(primary=True, name="semesterid")
221
277
    year = Unicode()
222
278
    semester = Unicode()
223
 
    active = Bool()
 
279
    state = Unicode()
224
280
 
225
281
    offerings = ReferenceSet(id, 'Offering.semester_id')
 
282
    enrolments = ReferenceSet(id,
 
283
                              'Offering.semester_id',
 
284
                              'Offering.id',
 
285
                              'Enrolment.offering_id')
226
286
 
227
287
    __init__ = _kwarg_init
228
288
 
246
306
                           'User.id')
247
307
    project_sets = ReferenceSet(id, 'ProjectSet.offering_id')
248
308
 
 
309
    worksheets = ReferenceSet(id, 
 
310
        'Worksheet.offering_id', 
 
311
        order_by="seq_no"
 
312
    )
 
313
 
249
314
    __init__ = _kwarg_init
250
315
 
251
316
    def __repr__(self):
252
317
        return "<%s %r in %r>" % (type(self).__name__, self.subject,
253
318
                                  self.semester)
254
319
 
255
 
    def enrol(self, user):
 
320
    def enrol(self, user, role=u'student'):
256
321
        '''Enrol a user in this offering.'''
257
 
        # We'll get a horrible database constraint violation error if we try
258
 
        # to add a second enrolment.
259
 
        if Store.of(self).find(Enrolment,
260
 
                               Enrolment.user_id == user.id,
261
 
                               Enrolment.offering_id == self.id).count() == 1:
262
 
            raise AlreadyEnrolledError()
263
 
 
264
 
        e = Enrolment(user=user, offering=self, active=True)
265
 
        self.enrolments.add(e)
 
322
        enrolment = Store.of(self).find(Enrolment,
 
323
                               Enrolment.user_id == user.id,
 
324
                               Enrolment.offering_id == self.id).one()
 
325
 
 
326
        if enrolment is None:
 
327
            enrolment = Enrolment(user=user, offering=self)
 
328
            self.enrolments.add(enrolment)
 
329
 
 
330
        enrolment.active = True
 
331
        enrolment.role = role
 
332
 
 
333
    def unenrol(self, user):
 
334
        '''Unenrol a user from this offering.'''
 
335
        enrolment = Store.of(self).find(Enrolment,
 
336
                               Enrolment.user_id == user.id,
 
337
                               Enrolment.offering_id == self.id).one()
 
338
        Store.of(enrolment).remove(enrolment)
 
339
 
 
340
    def get_permissions(self, user):
 
341
        perms = set()
 
342
        if user is not None:
 
343
            enrolment = self.get_enrolment(user)
 
344
            if enrolment or user.admin:
 
345
                perms.add('view')
 
346
            if (enrolment and enrolment.role in (u'tutor', u'lecturer')) \
 
347
               or user.admin:
 
348
                perms.add('edit')
 
349
        return perms
 
350
 
 
351
    def get_enrolment(self, user):
 
352
        try:
 
353
            enrolment = self.enrolments.find(user=user).one()
 
354
        except NotOneError:
 
355
            enrolment = None
 
356
 
 
357
        return enrolment
266
358
 
267
359
class Enrolment(Storm):
268
360
    __storm_table__ = "enrolment"
272
364
    user = Reference(user_id, User.id)
273
365
    offering_id = Int(name="offeringid")
274
366
    offering = Reference(offering_id, Offering.id)
 
367
    role = Unicode()
275
368
    notes = Unicode()
276
369
    active = Bool()
277
370
 
289
382
        return "<%s %r in %r>" % (type(self).__name__, self.user,
290
383
                                  self.offering)
291
384
 
292
 
class AlreadyEnrolledError(Exception):
293
 
    pass
294
 
 
295
385
# PROJECTS #
296
386
 
297
387
class ProjectSet(Storm):
315
405
    __storm_table__ = "project"
316
406
 
317
407
    id = Int(name="projectid", primary=True)
 
408
    name = Unicode()
 
409
    short_name = Unicode()
318
410
    synopsis = Unicode()
319
411
    url = Unicode()
320
412
    project_set_id = Int(name="projectsetid")
321
413
    project_set = Reference(project_set_id, ProjectSet.id)
322
414
    deadline = DateTime()
323
415
 
 
416
    assesseds = ReferenceSet(id, 'Assessed.project_id')
 
417
    submissions = ReferenceSet(id,
 
418
                               'Assessed.project_id',
 
419
                               'Assessed.id',
 
420
                               'ProjectSubmission.assessed_id')
 
421
 
324
422
    __init__ = _kwarg_init
325
423
 
326
424
    def __repr__(self):
327
 
        return "<%s '%s' in %r>" % (type(self).__name__, self.synopsis,
 
425
        return "<%s '%s' in %r>" % (type(self).__name__, self.short_name,
328
426
                                  self.project_set.offering)
329
427
 
 
428
    def can_submit(self, principal):
 
429
        return (self in principal.get_projects() and
 
430
                self.deadline > datetime.datetime.now())
 
431
 
 
432
    def submit(self, principal, path, revision, who):
 
433
        """Submit a Subversion path and revision to a project.
 
434
 
 
435
        'principal' is the owner of the Subversion repository, and the
 
436
        entity on behalf of whom the submission is being made. 'path' is
 
437
        a path within that repository, and 'revision' specifies which
 
438
        revision of that path. 'who' is the person making the submission.
 
439
        """
 
440
 
 
441
        if not self.can_submit(principal):
 
442
            raise Exception('cannot submit')
 
443
 
 
444
        a = Assessed.get(Store.of(self), principal, self)
 
445
        ps = ProjectSubmission()
 
446
        ps.path = path
 
447
        ps.revision = revision
 
448
        ps.date_submitted = datetime.datetime.now()
 
449
        ps.assessed = a
 
450
        ps.submitter = who
 
451
 
 
452
        return ps
 
453
 
 
454
 
330
455
class ProjectGroup(Storm):
331
456
    __storm_table__ = "project_group"
332
457
 
350
475
        return "<%s %s in %r>" % (type(self).__name__, self.name,
351
476
                                  self.project_set.offering)
352
477
 
 
478
    @property
 
479
    def display_name(self):
 
480
        return '%s (%s)' % (self.nick, self.name)
 
481
 
 
482
    def get_projects(self, offering=None, active_only=True):
 
483
        '''Return Projects that the group can submit.
 
484
 
 
485
        This will include projects in the project set which owns this group,
 
486
        unless the project set disallows groups (in which case none will be
 
487
        returned).
 
488
 
 
489
        Unless active_only is False, projects will only be returned if the
 
490
        group's offering is active.
 
491
 
 
492
        If an offering is specified, projects will only be returned if it
 
493
        matches the group's.
 
494
        '''
 
495
        return Store.of(self).find(Project,
 
496
            Project.project_set_id == ProjectSet.id,
 
497
            ProjectSet.id == self.project_set.id,
 
498
            ProjectSet.max_students_per_group != None,
 
499
            ProjectSet.offering_id == Offering.id,
 
500
            (offering is None) or (Offering.id == offering.id),
 
501
            Semester.id == Offering.semester_id,
 
502
            (not active_only) or (Semester.state == u'current'))
 
503
 
 
504
 
 
505
    def get_permissions(self, user):
 
506
        if user.admin or user in self.members:
 
507
            return set(['submit_project'])
 
508
        else:
 
509
            return set()
 
510
 
353
511
class ProjectGroupMembership(Storm):
354
512
    __storm_table__ = "group_member"
355
513
    __storm_primary__ = "user_id", "project_group_id"
365
523
        return "<%s %r in %r>" % (type(self).__name__, self.user,
366
524
                                  self.project_group)
367
525
 
 
526
class Assessed(Storm):
 
527
    __storm_table__ = "assessed"
 
528
 
 
529
    id = Int(name="assessedid", primary=True)
 
530
    user_id = Int(name="loginid")
 
531
    user = Reference(user_id, User.id)
 
532
    project_group_id = Int(name="groupid")
 
533
    project_group = Reference(project_group_id, ProjectGroup.id)
 
534
 
 
535
    project_id = Int(name="projectid")
 
536
    project = Reference(project_id, Project.id)
 
537
 
 
538
    extensions = ReferenceSet(id, 'ProjectExtension.assessed_id')
 
539
    submissions = ReferenceSet(id, 'ProjectSubmission.assessed_id')
 
540
 
 
541
    def __repr__(self):
 
542
        return "<%s %r in %r>" % (type(self).__name__,
 
543
            self.user or self.project_group, self.project)
 
544
 
 
545
    @classmethod
 
546
    def get(cls, store, principal, project):
 
547
        t = type(principal)
 
548
        if t not in (User, ProjectGroup):
 
549
            raise AssertionError('principal must be User or ProjectGroup')
 
550
 
 
551
        a = store.find(cls,
 
552
            (t is User) or (cls.project_group_id == principal.id),
 
553
            (t is ProjectGroup) or (cls.user_id == principal.id),
 
554
            Project.id == project.id).one()
 
555
 
 
556
        if a is None:
 
557
            a = cls()
 
558
            if t is User:
 
559
                a.user = principal
 
560
            else:
 
561
                a.project_group = principal
 
562
            a.project = project
 
563
            store.add(a)
 
564
 
 
565
        return a
 
566
 
 
567
 
 
568
class ProjectExtension(Storm):
 
569
    __storm_table__ = "project_extension"
 
570
 
 
571
    id = Int(name="extensionid", primary=True)
 
572
    assessed_id = Int(name="assessedid")
 
573
    assessed = Reference(assessed_id, Assessed.id)
 
574
    deadline = DateTime()
 
575
    approver_id = Int(name="approver")
 
576
    approver = Reference(approver_id, User.id)
 
577
    notes = Unicode()
 
578
 
 
579
class ProjectSubmission(Storm):
 
580
    __storm_table__ = "project_submission"
 
581
 
 
582
    id = Int(name="submissionid", primary=True)
 
583
    assessed_id = Int(name="assessedid")
 
584
    assessed = Reference(assessed_id, Assessed.id)
 
585
    path = Unicode()
 
586
    revision = Int()
 
587
    submitter_id = Int(name="submitter")
 
588
    submitter = Reference(submitter_id, User.id)
 
589
    date_submitted = DateTime()
 
590
 
 
591
 
368
592
# WORKSHEETS AND EXERCISES #
369
593
 
370
594
class Exercise(Storm):
371
 
    # Note: Table "problem" is called "Exercise" in the Object layer, since
372
 
    # it's called that everywhere else.
373
 
    __storm_table__ = "problem"
 
595
    __storm_table__ = "exercise"
 
596
    id = Unicode(primary=True, name="identifier")
 
597
    name = Unicode()
 
598
    description = Unicode()
 
599
    partial = Unicode()
 
600
    solution = Unicode()
 
601
    include = Unicode()
 
602
    num_rows = Int()
374
603
 
375
 
    id = Int(primary=True, name="problemid")
376
 
    name = Unicode(name="identifier")
377
 
    spec = Unicode()
 
604
    worksheet_exercises =  ReferenceSet(id,
 
605
        'WorksheetExercise.exercise_id')
378
606
 
379
607
    worksheets = ReferenceSet(id,
380
608
        'WorksheetExercise.exercise_id',
381
609
        'WorksheetExercise.worksheet_id',
382
610
        'Worksheet.id'
383
611
    )
 
612
    
 
613
    test_suites = ReferenceSet(id, 
 
614
        'TestSuite.exercise_id',
 
615
        order_by='seq_no')
384
616
 
385
617
    __init__ = _kwarg_init
386
618
 
387
619
    def __repr__(self):
388
620
        return "<%s %s>" % (type(self).__name__, self.name)
389
621
 
390
 
    @classmethod
391
 
    def get_by_name(cls, store, name):
392
 
        """
393
 
        Get the Exercise from the db associated with a given store and name.
394
 
        If the exercise is not in the database, creates it and inserts it
395
 
        automatically.
396
 
        """
397
 
        ex = store.find(cls, cls.name == unicode(name)).one()
398
 
        if ex is not None:
399
 
            return ex
400
 
        ex = Exercise(name=unicode(name))
401
 
        store.add(ex)
402
 
        store.commit()
403
 
        return ex
 
622
    def get_permissions(self, user):
 
623
        perms = set()
 
624
        roles = set()
 
625
        if user is not None:
 
626
            if user.admin:
 
627
                perms.add('edit')
 
628
                perms.add('view')
 
629
            elif 'lecturer' in set((e.role for e in user.active_enrolments)):
 
630
                perms.add('edit')
 
631
                perms.add('view')
 
632
            
 
633
        return perms
 
634
    
 
635
    def get_description(self):
 
636
        return rst(self.description)
 
637
 
 
638
    def delete(self):
 
639
        """Deletes the exercise, providing it has no associated worksheets."""
 
640
        if (self.worksheet_exercises.count() > 0):
 
641
            raise IntegrityError()
 
642
        for suite in self.test_suites:
 
643
            suite.delete()
 
644
        Store.of(self).remove(self)
404
645
 
405
646
class Worksheet(Storm):
406
647
    __storm_table__ = "worksheet"
407
648
 
408
649
    id = Int(primary=True, name="worksheetid")
409
 
    # XXX subject is not linked to a Subject object. This is a property of
410
 
    # the database, and will be refactored.
411
 
    subject = Unicode()
412
 
    name = Unicode(name="identifier")
 
650
    offering_id = Int(name="offeringid")
 
651
    identifier = Unicode()
 
652
    name = Unicode()
413
653
    assessable = Bool()
414
 
    mtime = DateTime()
415
 
 
416
 
    exercises = ReferenceSet(id,
417
 
        'WorksheetExercise.worksheet_id',
418
 
        'WorksheetExercise.exercise_id',
419
 
        Exercise.id)
420
 
    # Use worksheet_exercises to get access to the WorksheetExercise objects
421
 
    # binding worksheets to exercises. This is required to access the
 
654
    data = Unicode()
 
655
    seq_no = Int()
 
656
    format = Unicode()
 
657
 
 
658
    attempts = ReferenceSet(id, "ExerciseAttempt.worksheetid")
 
659
    offering = Reference(offering_id, 'Offering.id')
 
660
 
 
661
    all_worksheet_exercises = ReferenceSet(id,
 
662
        'WorksheetExercise.worksheet_id')
 
663
 
 
664
    # Use worksheet_exercises to get access to the *active* WorksheetExercise
 
665
    # objects binding worksheets to exercises. This is required to access the
422
666
    # "optional" field.
423
 
    worksheet_exercises = ReferenceSet(id,
424
 
        'WorksheetExercise.worksheet_id')
 
667
 
 
668
    @property
 
669
    def worksheet_exercises(self):
 
670
        return self.all_worksheet_exercises.find(active=True)
425
671
 
426
672
    __init__ = _kwarg_init
427
673
 
428
674
    def __repr__(self):
429
675
        return "<%s %s>" % (type(self).__name__, self.name)
430
676
 
431
 
    # XXX Refactor this - make it an instance method of Subject rather than a
432
 
    # class method of Worksheet. Can't do that now because Subject isn't
433
 
    # linked referentially to the Worksheet.
434
 
    @classmethod
435
 
    def get_by_name(cls, store, subjectname, worksheetname):
436
 
        """
437
 
        Get the Worksheet from the db associated with a given store, subject
438
 
        name and worksheet name.
439
 
        """
440
 
        return store.find(cls, cls.subject == unicode(subjectname),
441
 
            cls.name == unicode(worksheetname)).one()
442
 
 
443
 
    def remove_all_exercises(self, store):
 
677
    def remove_all_exercises(self):
444
678
        """
445
679
        Remove all exercises from this worksheet.
446
680
        This does not delete the exercises themselves. It just removes them
447
681
        from the worksheet.
448
682
        """
 
683
        store = Store.of(self)
 
684
        for ws_ex in self.all_worksheet_exercises:
 
685
            if ws_ex.saves.count() > 0 or ws_ex.attempts.count() > 0:
 
686
                raise IntegrityError()
449
687
        store.find(WorksheetExercise,
450
688
            WorksheetExercise.worksheet == self).remove()
451
 
 
 
689
            
 
690
    def get_permissions(self, user):
 
691
        return self.offering.get_permissions(user)
 
692
    
 
693
    def get_xml(self):
 
694
        """Returns the xml of this worksheet, converts from rst if required."""
 
695
        if self.format == u'rst':
 
696
            ws_xml = rst(self.data)
 
697
            return ws_xml
 
698
        else:
 
699
            return self.data
 
700
    
 
701
    def delete(self):
 
702
        """Deletes the worksheet, provided it has no attempts on any exercises.
 
703
        
 
704
        Returns True if delete succeeded, or False if this worksheet has
 
705
        attempts attached."""
 
706
        for ws_ex in self.all_worksheet_exercises:
 
707
            if ws_ex.saves.count() > 0 or ws_ex.attempts.count() > 0:
 
708
                raise IntegrityError()
 
709
        
 
710
        self.remove_all_exercises()
 
711
        Store.of(self).remove(self)
 
712
        
452
713
class WorksheetExercise(Storm):
453
 
    __storm_table__ = "worksheet_problem"
454
 
    __storm_primary__ = "worksheet_id", "exercise_id"
 
714
    __storm_table__ = "worksheet_exercise"
 
715
    
 
716
    id = Int(primary=True, name="ws_ex_id")
455
717
 
456
718
    worksheet_id = Int(name="worksheetid")
457
719
    worksheet = Reference(worksheet_id, Worksheet.id)
458
 
    exercise_id = Int(name="problemid")
 
720
    exercise_id = Unicode(name="exerciseid")
459
721
    exercise = Reference(exercise_id, Exercise.id)
460
722
    optional = Bool()
 
723
    active = Bool()
 
724
    seq_no = Int()
 
725
    
 
726
    saves = ReferenceSet(id, "ExerciseSave.ws_ex_id")
 
727
    attempts = ReferenceSet(id, "ExerciseAttempt.ws_ex_id")
461
728
 
462
729
    __init__ = _kwarg_init
463
730
 
464
731
    def __repr__(self):
465
732
        return "<%s %s in %s>" % (type(self).__name__, self.exercise.name,
466
 
                                  self.worksheet.name)
 
733
                                  self.worksheet.identifier)
 
734
 
 
735
    def get_permissions(self, user):
 
736
        return self.worksheet.get_permissions(user)
 
737
    
467
738
 
468
739
class ExerciseSave(Storm):
469
740
    """
474
745
    ExerciseSave may be extended with additional semantics (such as
475
746
    ExerciseAttempt).
476
747
    """
477
 
    __storm_table__ = "problem_save"
478
 
    __storm_primary__ = "exercise_id", "user_id", "date"
479
 
 
480
 
    exercise_id = Int(name="problemid")
481
 
    exercise = Reference(exercise_id, Exercise.id)
 
748
    __storm_table__ = "exercise_save"
 
749
    __storm_primary__ = "ws_ex_id", "user_id"
 
750
 
 
751
    ws_ex_id = Int(name="ws_ex_id")
 
752
    worksheet_exercise = Reference(ws_ex_id, "WorksheetExercise.id")
 
753
 
482
754
    user_id = Int(name="loginid")
483
755
    user = Reference(user_id, User.id)
484
756
    date = DateTime()
503
775
        they won't count (either as a penalty or success), but will still be
504
776
        stored.
505
777
    """
506
 
    __storm_table__ = "problem_attempt"
507
 
    __storm_primary__ = "exercise_id", "user_id", "date"
 
778
    __storm_table__ = "exercise_attempt"
 
779
    __storm_primary__ = "ws_ex_id", "user_id", "date"
508
780
 
509
781
    # The "text" field is the same but has a different name in the DB table
510
782
    # for some reason.
511
783
    text = Unicode(name="attempt")
512
784
    complete = Bool()
513
785
    active = Bool()
 
786
    
 
787
    def get_permissions(self, user):
 
788
        return set(['view']) if user is self.user else set()
 
789
  
 
790
class TestSuite(Storm):
 
791
    """A Testsuite acts as a container for the test cases of an exercise."""
 
792
    __storm_table__ = "test_suite"
 
793
    __storm_primary__ = "exercise_id", "suiteid"
 
794
    
 
795
    suiteid = Int()
 
796
    exercise_id = Unicode(name="exerciseid")
 
797
    description = Unicode()
 
798
    seq_no = Int()
 
799
    function = Unicode()
 
800
    stdin = Unicode()
 
801
    exercise = Reference(exercise_id, Exercise.id)
 
802
    test_cases = ReferenceSet(suiteid, 'TestCase.suiteid', order_by="seq_no")
 
803
    variables = ReferenceSet(suiteid, 'TestSuiteVar.suiteid', order_by='arg_no')
 
804
    
 
805
    def delete(self):
 
806
        """Delete this suite, without asking questions."""
 
807
        for vaariable in self.variables:
 
808
            variable.delete()
 
809
        for test_case in self.test_cases:
 
810
            test_case.delete()
 
811
        Store.of(self).remove(self)
 
812
 
 
813
class TestCase(Storm):
 
814
    """A TestCase is a member of a TestSuite.
 
815
    
 
816
    It contains the data necessary to check if an exercise is correct"""
 
817
    __storm_table__ = "test_case"
 
818
    __storm_primary__ = "testid", "suiteid"
 
819
    
 
820
    testid = Int()
 
821
    suiteid = Int()
 
822
    suite = Reference(suiteid, "TestSuite.suiteid")
 
823
    passmsg = Unicode()
 
824
    failmsg = Unicode()
 
825
    test_default = Unicode()
 
826
    seq_no = Int()
 
827
    
 
828
    parts = ReferenceSet(testid, "TestCasePart.testid")
 
829
    
 
830
    __init__ = _kwarg_init
 
831
    
 
832
    def delete(self):
 
833
        for part in self.parts:
 
834
            part.delete()
 
835
        Store.of(self).remove(self)
 
836
 
 
837
class TestSuiteVar(Storm):
 
838
    """A container for the arguments of a Test Suite"""
 
839
    __storm_table__ = "suite_variable"
 
840
    __storm_primary__ = "varid"
 
841
    
 
842
    varid = Int()
 
843
    suiteid = Int()
 
844
    var_name = Unicode()
 
845
    var_value = Unicode()
 
846
    var_type = Unicode()
 
847
    arg_no = Int()
 
848
    
 
849
    suite = Reference(suiteid, "TestSuite.suiteid")
 
850
    
 
851
    __init__ = _kwarg_init
 
852
    
 
853
    def delete(self):
 
854
        Store.of(self).remove(self)
 
855
    
 
856
class TestCasePart(Storm):
 
857
    """A container for the test elements of a Test Case"""
 
858
    __storm_table__ = "test_case_part"
 
859
    __storm_primary__ = "partid"
 
860
    
 
861
    partid = Int()
 
862
    testid = Int()
 
863
    
 
864
    part_type = Unicode()
 
865
    test_type = Unicode()
 
866
    data = Unicode()
 
867
    filename = Unicode()
 
868
    
 
869
    test = Reference(testid, "TestCase.testid")
 
870
    
 
871
    __init__ = _kwarg_init
 
872
    
 
873
    def delete(self):
 
874
        Store.of(self).remove(self)