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
35
from ivle.worksheet.rst import rst
36
37
__all__ = ['get_store',
39
40
'ProjectSet', 'Project', 'ProjectGroup', 'ProjectGroupMembership',
40
41
'Exercise', 'Worksheet', 'WorksheetExercise',
41
42
'ExerciseSave', 'ExerciseAttempt',
42
'AlreadyEnrolledError', 'TestCase', 'TestSuite', 'TestSuiteVar'
43
'TestCase', 'TestSuite', 'TestSuiteVar'
45
46
def _kwarg_init(self, **kwargs):
99
100
studentid = Unicode()
100
101
settings = Unicode()
103
if self.rolenm is None:
105
return ivle.caps.Role(self.rolenm)
106
def _set_role(self, value):
107
if not isinstance(value, ivle.caps.Role):
108
raise TypeError("role must be an ivle.caps.Role")
109
self.rolenm = unicode(value)
110
role = property(_get_role, _set_role)
112
103
__init__ = _kwarg_init
114
105
def __repr__(self):
126
117
return self.hash_password(password) == self.passhash
128
def hasCap(self, capability):
129
"""Given a capability (which is a Role object), returns True if this
130
User has that capability, False otherwise.
132
return self.role.hasCap(capability)
135
120
def password_expired(self):
136
121
fieldval = self.pass_exp
212
197
return store.find(cls, cls.login == unicode(login)).one()
214
199
def get_permissions(self, user):
215
if user and user.rolenm == 'admin' or user is self:
200
if user and user.admin or user is self:
216
201
return set(['view', 'edit'])
283
275
return "<%s %r in %r>" % (type(self).__name__, self.subject,
286
def enrol(self, user):
278
def enrol(self, user, role=u'student'):
287
279
'''Enrol a user in this offering.'''
288
# We'll get a horrible database constraint violation error if we try
289
# to add a second enrolment.
290
if Store.of(self).find(Enrolment,
291
Enrolment.user_id == user.id,
292
Enrolment.offering_id == self.id).count() == 1:
293
raise AlreadyEnrolledError()
295
e = Enrolment(user=user, offering=self, active=True)
296
self.enrolments.add(e)
280
enrolment = Store.of(self).find(Enrolment,
281
Enrolment.user_id == user.id,
282
Enrolment.offering_id == self.id).one()
284
if enrolment is None:
285
enrolment = Enrolment(user=user, offering=self)
286
self.enrolments.add(enrolment)
288
enrolment.active = True
289
enrolment.role = role
291
def unenrol(self, user):
292
'''Unenrol a user from this offering.'''
293
enrolment = Store.of(self).find(Enrolment,
294
Enrolment.user_id == user.id,
295
Enrolment.offering_id == self.id).one()
296
Store.of(enrolment).remove(enrolment)
298
298
def get_permissions(self, user):
300
300
if user is not None:
302
if user.rolenm == 'admin':
301
enrolment = self.get_enrolment(user)
302
if enrolment or user.admin:
304
if (enrolment and enrolment.role in (u'tutor', u'lecturer')) \
303
306
perms.add('edit')
309
def get_enrolment(self, user):
311
enrolment = self.enrolments.find(user=user).one()
306
317
class Enrolment(Storm):
307
318
__storm_table__ = "enrolment"
308
319
__storm_primary__ = "user_id", "offering_id"
407
416
# WORKSHEETS AND EXERCISES #
409
418
class Exercise(Storm):
410
# Note: Table "problem" is called "Exercise" in the Object layer, since
411
# it's called that everywhere else.
412
__storm_table__ = "problem"
413
#TODO: Add in a field for the user-friendly identifier
419
__storm_table__ = "exercise"
414
420
id = Unicode(primary=True, name="identifier")
416
422
description = Unicode()
419
425
include = Unicode()
428
worksheet_exercises = ReferenceSet(id,
429
'WorksheetExercise.exercise_id')
422
431
worksheets = ReferenceSet(id,
423
432
'WorksheetExercise.exercise_id',
424
433
'WorksheetExercise.worksheet_id',
428
test_suites = ReferenceSet(id, 'TestSuite.exercise_id')
437
test_suites = ReferenceSet(id,
438
'TestSuite.exercise_id',
430
441
__init__ = _kwarg_init
432
443
def __repr__(self):
433
444
return "<%s %s>" % (type(self).__name__, self.name)
446
def get_permissions(self, user):
453
elif 'lecturer' in set((e.role for e in user.active_enrolments)):
459
def get_description(self):
460
return rst(self.description)
463
"""Deletes the exercise, providing it has no associated worksheets."""
464
if (self.worksheet_exercises.count() > 0):
465
raise IntegrityError()
466
for suite in self.test_suites:
468
Store.of(self).remove(self)
436
470
class Worksheet(Storm):
437
471
__storm_table__ = "worksheet"
439
473
id = Int(primary=True, name="worksheetid")
440
# XXX subject is not linked to a Subject object. This is a property of
441
# the database, and will be refactored.
442
474
offering_id = Int(name="offeringid")
443
name = Unicode(name="identifier")
475
identifier = Unicode()
444
477
assessable = Bool()
447
482
attempts = ReferenceSet(id, "ExerciseAttempt.worksheetid")
448
483
offering = Reference(offering_id, 'Offering.id')
450
exercises = ReferenceSet(id,
451
'WorksheetExercise.worksheet_id',
452
'WorksheetExercise.exercise_id',
454
# Use worksheet_exercises to get access to the WorksheetExercise objects
455
# binding worksheets to exercises. This is required to access the
485
all_worksheet_exercises = ReferenceSet(id,
486
'WorksheetExercise.worksheet_id')
488
# Use worksheet_exercises to get access to the *active* WorksheetExercise
489
# objects binding worksheets to exercises. This is required to access the
456
490
# "optional" field.
457
worksheet_exercises = ReferenceSet(id,
458
'WorksheetExercise.worksheet_id')
493
def worksheet_exercises(self):
494
return self.all_worksheet_exercises.find(active=True)
461
496
__init__ = _kwarg_init
475
510
return store.find(cls, cls.subject == unicode(subjectname),
476
511
cls.name == unicode(worksheetname)).one()
478
def remove_all_exercises(self, store):
513
def remove_all_exercises(self):
480
515
Remove all exercises from this worksheet.
481
516
This does not delete the exercises themselves. It just removes them
482
517
from the worksheet.
519
store = Store.of(self)
520
for ws_ex in self.all_worksheet_exercises:
521
if ws_ex.saves.count() > 0 or ws_ex.attempts.count() > 0:
522
raise IntegrityError()
484
523
store.find(WorksheetExercise,
485
524
WorksheetExercise.worksheet == self).remove()
487
526
def get_permissions(self, user):
488
527
return self.offering.get_permissions(user)
530
"""Returns the xml of this worksheet, converts from rst if required."""
531
if self.format == u'rst':
532
ws_xml = rst(self.data)
538
"""Deletes the worksheet, provided it has no attempts on any exercises.
540
Returns True if delete succeeded, or False if this worksheet has
541
attempts attached."""
542
for ws_ex in self.all_worksheet_exercises:
543
if ws_ex.saves.count() > 0 or ws_ex.attempts.count() > 0:
544
raise IntegrityError()
546
self.remove_all_exercises()
547
Store.of(self).remove(self)
490
549
class WorksheetExercise(Storm):
491
__storm_table__ = "worksheet_problem"
492
__storm_primary__ = "worksheet_id", "exercise_id"
550
__storm_table__ = "worksheet_exercise"
552
id = Int(primary=True, name="ws_ex_id")
494
554
worksheet_id = Int(name="worksheetid")
495
555
worksheet = Reference(worksheet_id, Worksheet.id)
496
exercise_id = Unicode(name="problemid")
556
exercise_id = Unicode(name="exerciseid")
497
557
exercise = Reference(exercise_id, Exercise.id)
498
558
optional = Bool()
562
saves = ReferenceSet(id, "ExerciseSave.ws_ex_id")
563
attempts = ReferenceSet(id, "ExerciseAttempt.ws_ex_id")
500
565
__init__ = _kwarg_init
502
567
def __repr__(self):
503
568
return "<%s %s in %s>" % (type(self).__name__, self.exercise.name,
569
self.worksheet.identifier)
571
def get_permissions(self, user):
572
return self.worksheet.get_permissions(user)
506
575
class ExerciseSave(Storm):
512
581
ExerciseSave may be extended with additional semantics (such as
513
582
ExerciseAttempt).
515
__storm_table__ = "problem_save"
516
__storm_primary__ = "exercise_id", "user_id", "date"
518
exercise_id = Unicode(name="problemid")
519
exercise = Reference(exercise_id, Exercise.id)
584
__storm_table__ = "exercise_save"
585
__storm_primary__ = "ws_ex_id", "user_id"
587
ws_ex_id = Int(name="ws_ex_id")
588
worksheet_exercise = Reference(ws_ex_id, "WorksheetExercise.id")
520
590
user_id = Int(name="loginid")
521
591
user = Reference(user_id, User.id)
522
592
date = DateTime()
525
worksheet = Reference(worksheetid, Worksheet.id)
527
595
__init__ = _kwarg_init
543
611
they won't count (either as a penalty or success), but will still be
546
__storm_table__ = "problem_attempt"
547
__storm_primary__ = "exercise_id", "user_id", "date"
614
__storm_table__ = "exercise_attempt"
615
__storm_primary__ = "ws_ex_id", "user_id", "date"
549
617
# The "text" field is the same but has a different name in the DB table
550
618
# for some reason.
561
629
__storm_primary__ = "exercise_id", "suiteid"
564
exercise_id = Unicode(name="problemid")
632
exercise_id = Unicode(name="exerciseid")
565
633
description = Unicode()
567
635
function = Unicode()
568
636
stdin = Unicode()
569
637
exercise = Reference(exercise_id, Exercise.id)
570
test_cases = ReferenceSet(suiteid, 'TestCase.suiteid')
571
variables = ReferenceSet(suiteid, 'TestSuiteVar.suiteid')
638
test_cases = ReferenceSet(suiteid, 'TestCase.suiteid', order_by="seq_no")
639
variables = ReferenceSet(suiteid, 'TestSuiteVar.suiteid', order_by='arg_no')
642
"""Delete this suite, without asking questions."""
643
for vaariable in self.variables:
645
for test_case in self.test_cases:
647
Store.of(self).remove(self)
573
649
class TestCase(Storm):
574
650
"""A TestCase is a member of a TestSuite.
588
664
parts = ReferenceSet(testid, "TestCasePart.testid")
590
666
__init__ = _kwarg_init
669
for part in self.parts:
671
Store.of(self).remove(self)
592
673
class TestSuiteVar(Storm):
593
674
"""A container for the arguments of a Test Suite"""
594
__storm_table__ = "suite_variables"
675
__storm_table__ = "suite_variable"
595
676
__storm_primary__ = "varid"
606
687
__init__ = _kwarg_init
690
Store.of(self).remove(self)
608
692
class TestCasePart(Storm):
609
693
"""A container for the test elements of a Test Case"""
610
__storm_table__ = "test_case_parts"
694
__storm_table__ = "test_case_part"
611
695
__storm_primary__ = "partid"
added
removed
ivle/database.py
