def store(self, obj):
if self.on_store: self.on_store(obj)
bits, hash = self.lookup(obj)
filename = utils.substitute(self.config['src'], dict(bits=bits, hash=hash))
exists = os.path.exists(filename)
# Make directories.
try: os.makedirs(os.path.dirname(filename))
except os.error: pass
# Connect.
try:
conn = sqlite3.connect(filename)
conn.row_factory = sqlite3.Row
c = conn.cursor()
except sqlite3.Error as e:
raise RuntimeError('%s: %s' % (filename, e))
# Initialize database.
if not exists:
for query in self.config['init']:
q, params = utils.substitute_sql(query, obj)
try: c.execute(q, params)
except sqlite3.Error as e:
raise RuntimeError('%s: %s: %s' % (filename, q, e))
# Insert query.
hash1 = self.hashof(obj)
q, params = utils.substitute_sql(self.config['insert'], dict(obj,
_id=int(hash1,16),
_hash=hash1,
sha1=lambda arg: hashlib.sha1(arg).hexdigest(),
))
try: c.execute(q, params)
except sqlite3.Error as e:
raise RuntimeError('%s: %s: %s' % (filename, q, e))
# Commit & close.
try:
conn.commit()
conn.close()
except sqlite3.Error as e:
raise RuntimeError('%s: %s' % (filename, e))
try:
stat = os.stat(filename)
size = stat.st_size
except OSError as e:
raise RuntimeError('%s: %s' % (filename, e))
self.index_update_size(bits, hash, size)
def retrieve(self, obj, exclude=[]):
bits, hash = self.lookup(obj)
filename = utils.substitute(self.config['src'], dict(bits=bits, hash=hash))
if not os.path.exists(filename): return None
# Connect.
try:
conn = sqlite3.connect(filename)
conn.row_factory = sqlite3.Row
c = conn.cursor()
except sqlite3.Error as e:
raise RuntimeError('%s: %s' % (filename, e))
# Select query.
q, params = utils.substitute_sql(self.config['select'], dict(obj,
sha1=lambda arg: hashlib.sha1(arg).hexdigest(),
))
try:
rows = c.execute(q, params)
except sqlite3.Error as e:
raise RuntimeError('%s: %s: %s' % (filename, q, e))
row = rows.fetchone()
if row is None: return None
o = obj.copy()
o.update(dict(zip(row.keys(), row)))
# Close.
try: conn.close()
except sqlite3.Error as e: raise RuntimeError('%s: %s' % (filename, e))
return Driver.retrieve(self, o, exclude)
def store(self, obj):
if self.on_store: self.on_store(obj)
filename = utils.substitute(self.config['src'], obj)
exists = os.path.exists(filename)
# Make directories.
try: os.makedirs(os.path.dirname(filename))
except os.error: pass
# Connect.
try:
conn = sqlite3.connect(filename)
c = conn.cursor()
except sqlite3.Error as e:
raise RuntimeError('%s: %s' % (filename, e))
q = 'PRAGMA synchronous = OFF'
try: c.execute(q)
except sqlite3.Error as e:
raise RuntimeError('%s: %s: %s' % (filename, q, e))
# Initialize database.
if not exists:
for query in self.config['init']:
q, params = utils.substitute_sql(query, obj)
c.execute(q, params)
# Insert query.
q, params = utils.substitute_sql(self.config['insert'], obj)
try: c.execute(q, params)
except sqlite3.Error as e:
raise RuntimeError('%s: %s: %s' % (filename, q, e))
# Commit & close.
try:
conn.commit()
conn.close()
except sqlite3.Error as e:
raise RuntimeError('%s: %s' % (filename, e))
def retrieve(self, obj, exclude):
filename = utils.substitute(self.config['src'], obj)
if not os.path.exists(filename): return None
conn = sqlite3.connect(filename)
conn.row_factory = sqlite3.Row
c = conn.cursor()
q, params = utils.substitute_sql(self.config['select'], obj)
try: rows = c.execute(q, params)
except sqlite3.Error as e: raise RuntimeError("%s: %s" % (q, e))
row = rows.fetchone()
if row is None: return None
o = obj.copy()
o.update(dict(zip(row.keys(), row)))
conn.close()
return Driver.retrieve(self, o, exclude)
