summaryrefslogtreecommitdiffstats
path: root/admin/lib/storage.py
blob: 5627ff77d994dfa08b6c82b3a00a5f2152993050 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import shelve

import csvstore
import jsonstore
import options
import storage_encapsulation

if options.storage_encrypt == True
    encapsulation_class = storage_encapsulation.DESFileEncapsulation
else:
    encapsulation_class = storage_encapsulation.RawFileEncapsulation

if options.storage == 'shelve':
    if options.storage_encrypt == True:
        raise NotImplementedError("Who the fuck uses Shelve anyway.")
    storage = shelve.open(options.shelf)
if options.storage == 'json':
    encapsulation = encapsulation_class(options.json)
    storage = jsonstore.Storage(encapsulation)
if options.storage == 'csv':
    encapsulation = encapsulation_class(options.csv)
    storage = csvstore.Storage(encapsulation)

nobody = (None, '-unknown-')

to_hex = lambda v: '%x' % v
from_hex = lambda v: int(v, 16)
dehex = lambda s: int(s.translate(None, 'xXlL'), 16)

get_card = lambda mid: storage.get(to_hex(mid), nobody)
cards_for_user = lambda name: map(lambda (k,v): from_hex(k), 
    filter(lambda (k,(u,n)): n == name, storage.iteritems()))

def add_user(username, mid, uid):
    storage[to_hex(mid)] = (uid, username)

def del_card(mid):
    return storage.pop(to_hex(mid), nobody)

def del_filter(f):
    cards = map(lambda (k,v): k, filter(f, storage.iteritems()))
    r = []
    for c in cards:
        r.append(storage.pop(c, nobody))
    return r

del_uid = lambda uid: del_filter(lambda (k, (u,n)): u == uid)
del_username = lambda name: del_filter(lambda (k, (u,n)): n == name)