/usr/share/pyshared/arcom/store/zodbstore.py is in nordugrid-arc-python 1.1.1-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | import arc
import os, copy, threading, time
from ZODB import FileStorage, DB
from persistent import Persistent
from arcom.store.basestore import BaseStore
from arcom.logger import Logger
log = Logger(arc.Logger(arc.Logger_getRootLogger(), 'arcom.ZODBStore'))
class Metadata(Persistent):
"""
Class for persistent mutable metadata
"""
def __init__(self):
self.meta={}
def addMeta(self, ID, object):
self.meta[ID]=object
self._p_changed = 1
def delMeta(self, ID):
del(self.meta[ID])
self._p_changed = 1
def __getitem__(self, ID):
return self.meta[ID]
def list(self):
return self.meta.keys()
class ZODBStore(BaseStore):
def __init__(self, storecfg, non_existent_object = {}):
""" Constructor of ZODBStore.
ZODBStore(storecfg)
'storecfg' is an XMLNode with a 'DataDir'
'non_existent_object' will be returned if an object not found
"""
BaseStore.__init__(self, storecfg, non_existent_object)
log.msg(arc.VERBOSE, "ZODBStore constructor called")
log.msg(arc.VERBOSE, "datadir:", self.datadir)
self.dbfile = os.path.join(self.datadir,'metadata.fs')
if os.path.isfile(self.dbfile):
self.db = DB(FileStorage.FileStorage(self.dbfile))
self.db.setCacheSize(4000)
# decrease db file size in case of previous deletions
# (should maybe do this in a separate script?)
#db.pack()
self.metadata = self.db.open().root()['meta']
else:
self.db = DB(FileStorage.FileStorage(self.dbfile))
root = self.db.open().root()
root['meta'] = Metadata()
get_transaction().commit()
self.metadata = root['meta']
self.period = 60
threading.Thread(target = self.packingThread, args = [self.period]).start()
def packingThread(self,period):
time.sleep(7)
while True:
self.db.pack()
time.sleep(period)
def list(self):
""" List the IDs of the existing entries.
list()
Returns a list of IDs.
Only relevant for shepherd
"""
#get_transaction().commit()
return self.metadata.list()
def get(self, ID):
""" Returns the object with the given ID.
get(ID)
'ID' is the ID of the requested object.
If there is no object with this ID, returns the given non_existent_object value.
"""
try:
#get_transaction().commit()
return self.metadata[ID]
except KeyError:
# don't print 'KeyError' if there is no such ID
pass
except:
# print whatever exception happened
log.msg()
log.msg(arc.ERROR, "ID", ID)
# if there was an exception, return the given non_existent_object
return copy.deepcopy(self.non_existent_object)
def set(self, ID, object):
""" Stores an object with the given ID..
set(ID, object)
'ID' is the ID of the object
'object' is the object itself
If there is already an object with this ID it will be overwritten completely.
"""
if not ID:
raise Exception, 'ID is empty'
try:
if object == None:
self.metadata.delMeta(ID)
else:
self.metadata.addMeta(ID, object)
get_transaction().commit()
except:
log.msg()
|