Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/storage.py
Show First 20 Lines • Show All 1,101 Lines • ▼ Show 20 Lines | ) -> Iterable[Optional[Origin]]: | ||||
result: List[Optional[Origin]] = [] | result: List[Optional[Origin]] = [] | ||||
for row in rows: | for row in rows: | ||||
origin_d = dict(zip(db.origin_cols, row)) | origin_d = dict(zip(db.origin_cols, row)) | ||||
url = origin_d["url"] | url = origin_d["url"] | ||||
result.append(None if url is None else Origin(url=url)) | result.append(None if url is None else Origin(url=url)) | ||||
return result | return result | ||||
@timed | @timed | ||||
@db_transaction_generator(statement_timeout=500) | @db_transaction(statement_timeout=500) | ||||
def origin_get_by_sha1( | def origin_get_by_sha1( | ||||
self, sha1s: List[bytes], db=None, cur=None | self, sha1s: List[bytes], db=None, cur=None | ||||
) -> Iterable[Optional[Dict[str, Any]]]: | ) -> List[Optional[Dict[str, Any]]]: | ||||
for line in db.origin_get_by_sha1(sha1s, cur): | return [ | ||||
if line[0] is not None: | dict(zip(db.origin_cols, row)) if row[0] else None | ||||
yield dict(zip(db.origin_cols, line)) | for row in db.origin_get_by_sha1(sha1s, cur) | ||||
else: | ] | ||||
yield None | |||||
@timed | @timed | ||||
@db_transaction_generator() | @db_transaction_generator() | ||||
def origin_get_range(self, origin_from=1, origin_count=100, db=None, cur=None): | def origin_get_range(self, origin_from=1, origin_count=100, db=None, cur=None): | ||||
for origin in db.origin_get_range(origin_from, origin_count, cur): | for origin in db.origin_get_range(origin_from, origin_count, cur): | ||||
yield dict(zip(db.origin_get_range_cols, origin)) | yield dict(zip(db.origin_get_range_cols, origin)) | ||||
@timed | @timed | ||||
▲ Show 20 Lines • Show All 309 Lines • Show Last 20 Lines |