← Back to branch summary

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

« back to all changes in this revision

Viewing changes to ivle/database.py

fixed a slight bug on line 155, which referenced a non-existant variable

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/database.py
39
39
            'ProjectSet', 'Project', 'ProjectGroup', 'ProjectGroupMembership',
40
40
            'Exercise', 'Worksheet', 'WorksheetExercise',
41
41
            'ExerciseSave', 'ExerciseAttempt',
42
 
            'AlreadyEnrolledError'
 
42
            'AlreadyEnrolledError', 'TestCase', 'TestSuite', 'TestSuiteVar'
43
43
        ]
44
44
 
45
45
def _kwarg_init(self, **kwargs):
53
53
    """
54
54
    Returns the Storm connection string, generated from the conf file.
55
55
    """
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)
 
56
 
 
57
    clusterstr = ''
 
58
    if ivle.conf.db_user:
 
59
        clusterstr += ivle.conf.db_user
 
60
        if ivle.conf.db_password:
 
61
            clusterstr += ':' + ivle.conf.db_password
 
62
        clusterstr += '@'
 
63
 
 
64
    host = ivle.conf.db_host or 'localhost'
 
65
    port = ivle.conf.db_port or 5432
 
66
 
 
67
    clusterstr += '%s:%d' % (host, port)
 
68
 
 
69
    return "postgres://%s/%s" % (clusterstr, ivle.conf.db_dbname)
59
70
 
60
71
def get_store():
61
72
    """
130
141
        fieldval = self.acct_exp
131
142
        return fieldval is not None and datetime.datetime.now() > fieldval
132
143
 
 
144
    @property
 
145
    def valid(self):
 
146
        return self.state == 'enabled' and not self.account_expired
 
147
 
133
148
    def _get_enrolments(self, justactive):
134
149
        return Store.of(self).find(Enrolment,
135
150
            Enrolment.user_id == self.id,
196
211
        """
197
212
        return store.find(cls, cls.login == unicode(login)).one()
198
213
 
 
214
    def get_permissions(self, user):
 
215
        if user and user.rolenm == 'admin' or user is self:
 
216
            return set(['view', 'edit'])
 
217
        else:
 
218
            return set()
 
219
 
199
220
# SUBJECTS AND ENROLMENTS #
200
221
 
201
222
class Subject(Storm):
214
235
    def __repr__(self):
215
236
        return "<%s '%s'>" % (type(self).__name__, self.short_name)
216
237
 
 
238
    def get_permissions(self, user):
 
239
        perms = set()
 
240
        if user is not None:
 
241
            perms.add('view')
 
242
            if user.rolenm == 'admin':
 
243
                perms.add('edit')
 
244
        return perms
 
245
 
217
246
class Semester(Storm):
218
247
    __storm_table__ = "semester"
219
248
 
246
275
                           'User.id')
247
276
    project_sets = ReferenceSet(id, 'ProjectSet.offering_id')
248
277
 
 
278
    worksheets = ReferenceSet(id, 
 
279
        'Worksheet.offering_id', 
 
280
        order_by="Worksheet.seq_no"
 
281
    )
 
282
 
249
283
    __init__ = _kwarg_init
250
284
 
251
285
    def __repr__(self):
264
298
        e = Enrolment(user=user, offering=self, active=True)
265
299
        self.enrolments.add(e)
266
300
 
 
301
    def get_permissions(self, user):
 
302
        perms = set()
 
303
        if user is not None:
 
304
            perms.add('view')
 
305
            if user.rolenm == 'admin':
 
306
                perms.add('edit')
 
307
        return perms
 
308
 
267
309
class Enrolment(Storm):
268
310
    __storm_table__ = "enrolment"
269
311
    __storm_primary__ = "user_id", "offering_id"
371
413
    # Note: Table "problem" is called "Exercise" in the Object layer, since
372
414
    # it's called that everywhere else.
373
415
    __storm_table__ = "problem"
374
 
 
375
 
    id = Int(primary=True, name="problemid")
376
 
    name = Unicode(name="identifier")
377
 
    spec = Unicode()
 
416
    id = Unicode(primary=True, name="identifier")
 
417
    name = Unicode()
 
418
    description = Unicode()
 
419
    partial = Unicode()
 
420
    solution = Unicode()
 
421
    include = Unicode()
 
422
    num_rows = Int()
378
423
 
379
424
    worksheets = ReferenceSet(id,
380
425
        'WorksheetExercise.exercise_id',
381
426
        'WorksheetExercise.worksheet_id',
382
427
        'Worksheet.id'
383
428
    )
 
429
    
 
430
    test_suites = ReferenceSet(id, 'TestSuite.exercise_id')
384
431
 
385
432
    __init__ = _kwarg_init
386
433
 
387
434
    def __repr__(self):
388
435
        return "<%s %s>" % (type(self).__name__, self.name)
389
436
 
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
404
437
 
405
438
class Worksheet(Storm):
406
439
    __storm_table__ = "worksheet"
407
440
 
408
441
    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")
 
442
    offering_id = Int(name="offeringid")
 
443
    identifier = Unicode()
 
444
    name = Unicode()
413
445
    assessable = Bool()
414
 
    mtime = DateTime()
415
 
 
416
 
    exercises = ReferenceSet(id,
417
 
        'WorksheetExercise.worksheet_id',
418
 
        'WorksheetExercise.exercise_id',
419
 
        Exercise.id)
 
446
    data = Unicode()
 
447
    seq_no = Int()
 
448
    format = Unicode()
 
449
 
 
450
    attempts = ReferenceSet(id, "ExerciseAttempt.worksheetid")
 
451
    offering = Reference(offering_id, 'Offering.id')
 
452
 
420
453
    # Use worksheet_exercises to get access to the WorksheetExercise objects
421
454
    # binding worksheets to exercises. This is required to access the
422
455
    # "optional" field.
423
456
    worksheet_exercises = ReferenceSet(id,
424
457
        'WorksheetExercise.worksheet_id')
 
458
        
425
459
 
426
460
    __init__ = _kwarg_init
427
461
 
448
482
        """
449
483
        store.find(WorksheetExercise,
450
484
            WorksheetExercise.worksheet == self).remove()
 
485
            
 
486
    def get_permissions(self, user):
 
487
        return self.offering.get_permissions(user)
451
488
 
452
489
class WorksheetExercise(Storm):
453
490
    __storm_table__ = "worksheet_problem"
454
 
    __storm_primary__ = "worksheet_id", "exercise_id"
 
491
    
 
492
    id = Int(primary=True, name="ws_prob_id")
455
493
 
456
494
    worksheet_id = Int(name="worksheetid")
457
495
    worksheet = Reference(worksheet_id, Worksheet.id)
458
 
    exercise_id = Int(name="problemid")
 
496
    exercise_id = Unicode(name="problemid")
459
497
    exercise = Reference(exercise_id, Exercise.id)
460
498
    optional = Bool()
 
499
    active = Bool()
 
500
    seq_no = Int()
 
501
    
 
502
    saves = ReferenceSet(id, "ExerciseSave.ws_ex_id")
 
503
    attempts = ReferenceSet(id, "ExerciseAttempt.ws_ex_id")
461
504
 
462
505
    __init__ = _kwarg_init
463
506
 
464
507
    def __repr__(self):
465
508
        return "<%s %s in %s>" % (type(self).__name__, self.exercise.name,
466
 
                                  self.worksheet.name)
 
509
                                  self.worksheet.identifier)
467
510
 
468
511
class ExerciseSave(Storm):
469
512
    """
475
518
    ExerciseAttempt).
476
519
    """
477
520
    __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)
 
521
    __storm_primary__ = "ws_ex_id", "user_id"
 
522
 
 
523
    ws_ex_id = Int(name="ws_prob_id")
 
524
    worksheet_exercise = Reference(ws_ex_id, "WorksheetExercise.id")
 
525
 
482
526
    user_id = Int(name="loginid")
483
527
    user = Reference(user_id, User.id)
484
528
    date = DateTime()
504
548
        stored.
505
549
    """
506
550
    __storm_table__ = "problem_attempt"
507
 
    __storm_primary__ = "exercise_id", "user_id", "date"
 
551
    __storm_primary__ = "ws_ex_id", "user_id", "date"
508
552
 
509
553
    # The "text" field is the same but has a different name in the DB table
510
554
    # for some reason.
511
555
    text = Unicode(name="attempt")
512
556
    complete = Bool()
513
557
    active = Bool()
 
558
    
 
559
    def get_permissions(self, user):
 
560
        return set(['view']) if user is self.user else set()
 
561
  
 
562
class TestSuite(Storm):
 
563
    """A Testsuite acts as a container for the test cases of an exercise."""
 
564
    __storm_table__ = "test_suite"
 
565
    __storm_primary__ = "exercise_id", "suiteid"
 
566
    
 
567
    suiteid = Int()
 
568
    exercise_id = Unicode(name="problemid")
 
569
    description = Unicode()
 
570
    seq_no = Int()
 
571
    function = Unicode()
 
572
    stdin = Unicode()
 
573
    exercise = Reference(exercise_id, Exercise.id)
 
574
    test_cases = ReferenceSet(suiteid, 'TestCase.suiteid')
 
575
    variables = ReferenceSet(suiteid, 'TestSuiteVar.suiteid')
 
576
 
 
577
class TestCase(Storm):
 
578
    """A TestCase is a member of a TestSuite.
 
579
    
 
580
    It contains the data necessary to check if an exercise is correct"""
 
581
    __storm_table__ = "test_case"
 
582
    __storm_primary__ = "testid", "suiteid"
 
583
    
 
584
    testid = Int()
 
585
    suiteid = Int()
 
586
    suite = Reference(suiteid, "TestSuite.suiteid")
 
587
    passmsg = Unicode()
 
588
    failmsg = Unicode()
 
589
    test_default = Unicode()
 
590
    seq_no = Int()
 
591
    
 
592
    parts = ReferenceSet(testid, "TestCasePart.testid")
 
593
    
 
594
    __init__ = _kwarg_init
 
595
 
 
596
class TestSuiteVar(Storm):
 
597
    """A container for the arguments of a Test Suite"""
 
598
    __storm_table__ = "suite_variables"
 
599
    __storm_primary__ = "varid"
 
600
    
 
601
    varid = Int()
 
602
    suiteid = Int()
 
603
    var_name = Unicode()
 
604
    var_value = Unicode()
 
605
    var_type = Unicode()
 
606
    arg_no = Int()
 
607
    
 
608
    suite = Reference(suiteid, "TestSuite.suiteid")
 
609
    
 
610
    __init__ = _kwarg_init
 
611
    
 
612
class TestCasePart(Storm):
 
613
    """A container for the test elements of a Test Case"""
 
614
    __storm_table__ = "test_case_parts"
 
615
    __storm_primary__ = "partid"
 
616
    
 
617
    partid = Int()
 
618
    testid = Int()
 
619
    
 
620
    part_type = Unicode()
 
621
    test_type = Unicode()
 
622
    data = Unicode()
 
623
    filename = Unicode()
 
624
    
 
625
    test = Reference(testid, "TestCase.testid")
 
626
    
 
627
    __init__ = _kwarg_init