Set a row factory instead of nesting the iterator, iterate through the cursor

This commit is contained in:
Lee Miller 2024-09-11 02:35:09 +03:00
parent addf75814e
commit f429143df8
Signed by: lee.miller
GPG Key ID: 4F97A5EA88F4AB63

View File

@ -41,14 +41,16 @@ class Inventory():
cur.execute("INSERT INTO status VALUES ('lastvacuumtime', ?)", (now,))
self._db.commit()
def __objects(self, cur):
return (
structure.Object(
@staticmethod
def __object(cursor, row):
vector, expires, obj_type, version, stream, tag, data, offset = row
return structure.Object(
expires, obj_type, version, stream, data, offset,
tag=tag, vector=vector)
for (vector, expires, obj_type, version, stream, tag, data, offset)
in cur.fetchall()
)
def __objects(self, cur):
cur.row_factory = self.__object
return cur
def cleanup(self):
with shared.objects_lock:
@ -80,7 +82,7 @@ class Inventory():
' ORDER BY random()',
(int(time.time()), stream or shared.stream)
)
return [v for v, in cur.fetchall()]
return [v for v, in cur]
def get(self, vector, default=None):
try:
@ -90,7 +92,7 @@ class Inventory():
def keys(self):
cur = self._db.execute('SELECT vector FROM objects')
return (v for v, in cur.fetchall())
return (v for v, in cur)
def values(self):
cur = self._db.execute('SELECT * FROM objects')
@ -114,14 +116,11 @@ class Inventory():
def __getitem__(self, vector):
cur = self._db.execute(
'SELECT * FROM objects WHERE vector = ?', (vector,))
cur.row_factory = self.__object
item = cur.fetchone()
if item is None:
raise KeyError(vector)
vector, expires, obj_type, version, stream, tag, data, offset = item
return structure.Object(
expires, obj_type, version, stream, data, offset,
tag=tag, vector=vector
)
return item
def __delitem__(self, vector):
with shared.objects_lock: # KeyError