Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/postgresql/storage.py
Show First 20 Lines • Show All 948 Lines • ▼ Show 20 Lines | ) -> None: | ||||
origin_url = self.origin_get([visit_status.origin], db=db, cur=cur)[0] | origin_url = self.origin_get([visit_status.origin], db=db, cur=cur)[0] | ||||
if not origin_url: | if not origin_url: | ||||
raise StorageArgumentException(f"Unknown origin {visit_status.origin}") | raise StorageArgumentException(f"Unknown origin {visit_status.origin}") | ||||
if visit_status.type is None: | if visit_status.type is None: | ||||
origin_visit = self.origin_visit_get_by( | origin_visit = self.origin_visit_get_by( | ||||
visit_status.origin, visit_status.visit, db=db, cur=cur | visit_status.origin, visit_status.visit, db=db, cur=cur | ||||
) | ) | ||||
assert origin_visit is not None | if origin_visit is None: | ||||
raise StorageArgumentException( | |||||
f"Unknown origin visit {visit_status.visit} " | |||||
f"of origin {visit_status.origin}" | |||||
) | |||||
origin_visit_status = attr.evolve(visit_status, type=origin_visit.type) | origin_visit_status = attr.evolve(visit_status, type=origin_visit.type) | ||||
else: | else: | ||||
origin_visit_status = visit_status | origin_visit_status = visit_status | ||||
visit_statuses_.append(origin_visit_status) | visit_statuses_.append(origin_visit_status) | ||||
for visit_status in visit_statuses_: | for visit_status in visit_statuses_: | ||||
▲ Show 20 Lines • Show All 509 Lines • Show Last 20 Lines |