1.1 --- a/imip_store.py Sat Oct 25 19:04:47 2014 +0200
1.2 +++ b/imip_store.py Sat Oct 25 21:43:45 2014 +0200
1.3 @@ -124,9 +124,40 @@
1.4
1.5 return True
1.6
1.7 - def queue_request(self, user, uid):
1.8 + def get_requests(self, user):
1.9 +
1.10 + "Get requests for the given 'user'."
1.11 +
1.12 + filename = self.get_object_in_store(user, "requests")
1.13 + if not filename:
1.14 + return None
1.15 +
1.16 + f = open(filename)
1.17 + try:
1.18 + return [line.strip() for line in f.readlines()]
1.19 + finally:
1.20 + f.close()
1.21 +
1.22 + def set_requests(self, user, requests):
1.23
1.24 - "Queue a request for 'user' having the given 'uid'."
1.25 + "For the given 'user', set the list of queued 'requests'."
1.26 +
1.27 + filename = self.get_object_in_store(user, "requests")
1.28 + if not filename:
1.29 + return False
1.30 +
1.31 + f = open(filename, "w")
1.32 + try:
1.33 + for request in requests:
1.34 + print >>f, request
1.35 + finally:
1.36 + f.close()
1.37 +
1.38 + return True
1.39 +
1.40 + def set_request(self, user, request):
1.41 +
1.42 + "For the given 'user', set the queued 'request'."
1.43
1.44 filename = self.get_object_in_store(user, "requests")
1.45 if not filename:
1.46 @@ -134,12 +165,23 @@
1.47
1.48 f = open(filename, "a")
1.49 try:
1.50 - print >>f, uid
1.51 + print >>f, request
1.52 finally:
1.53 f.close()
1.54
1.55 return True
1.56
1.57 + def queue_request(self, user, uid):
1.58 +
1.59 + "Queue a request for 'user' having the given 'uid'."
1.60 +
1.61 + requests = self.get_requests(user)
1.62 +
1.63 + if uid not in requests:
1.64 + return self.set_request(user, uid)
1.65 +
1.66 + return False
1.67 +
1.68 class FilePublisher(FileBase):
1.69
1.70 "A publisher of objects."