← Back to branch summary

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

« back to all changes in this revision

Viewing changes to userdb/query.py

  • Committer: Matt Giuca
  • Date: 2009-12-01 04:27:58 UTC
  • mfrom: (1164.2.46 sphinx-docs)
  • Revision ID: matt.giuca@gmail.com-20091201042758-wuxd9bdec00c283i
Merged sphinx-docs branch. This adds Sphinx documentation for the entire IVLE system (for system administrators and developers), and removes all of our random old document files (all either irrelevant, or moved into the Sphinx docs nicely). Currently incomplete, but ready to merge.

Show diffs side-by-side

added added

removed removed

Lines of Context:
userdb/query.py
1
 
import pg
2
 
 
3
 
def getCredentials(db, login):
4
 
    cred = {}
5
 
 
6
 
    res = db.query('SELECT users.nick FROM users WHERE users.login=\'' + login + '\'').dictresult()
7
 
    if len(res) <> 1:
8
 
        return None
9
 
    cred['nick'] = res[0]['nick']
10
 
 
11
 
    res = db.query('SELECT groupid FROM group_members WHERE login=\'' + login + '\'').dictresult()
12
 
    cred['groups'] = map(lambda x: x.get('groupid'), res)
13
 
 
14
 
    res = db.query('SELECT role FROM roles WHERE login=\'' + login + '\'').dictresult()
15
 
    if len(res) <> 1:
16
 
        return None
17
 
    cred['role'] = res[0]['role']
18
 
 
19
 
    return cred
20
 
 
21
 
def contains(m, l):
22
 
    for i in l:
23
 
        if m == i:
24
 
            return True
25
 
    return False
26
 
 
27
 
db = pg.connect('tom', 'localhost', 5432, None, None, 'tom', 'sh0r3ham')
28
 
c = getCredentials(db, 'conway')
29
 
 
30
 
if c['role'] != 'student' or contains('2007-INFO10001-321', c['groups']):
31
 
    print c['nick'] + ' may view files for group \'2007-INFO10001-321\'.'
32