forked from baumartig/paperboy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
shelve_job_handler.py
69 lines (50 loc) · 1.16 KB
/
shelve_job_handler.py
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import shelve
from job import Job
dbName = "data/jobs.db"
def loadJobs():
db = shelve.open(dbName)
jobs = list(db.values())
db.close()
return jobs
def saveJobs(jobs):
db = shelve.open(dbName)
for job in jobs:
db[str(job.id)] = job
db.close()
def addNewJob(job):
db = shelve.open(dbName)
id = len(db)
job.id = id
db[str(id)] = job
db.close()
def deleteJob(job):
db = shelve.open(dbName)
db.pop(str(job.id))
db.close()
def clear():
db = shelve.open(dbName)
db.clear()
db.close()
def test():
print "Start test"
global dbName
dbName = "test.db"
clear()
jobs = loadJobs()
assert len(jobs) == 0
newJob = Job("test")
addNewJob(newJob)
jobs = loadJobs()
assert len(jobs) == 1
assert jobs.index(newJob) >= 0
newJob = jobs[newJob.id]
newJob.recipeRef = "new test"
saveJobs(jobs)
jobs = loadJobs()
assert newJob.recipeRef == jobs[newJob.id].recipeRef
clear()
jobs = loadJobs()
assert len(jobs) == 0
print "End test"
if __name__ == "__main__":
test()