Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/cassandra/storage.py
| Show First 20 Lines • Show All 932 Lines • ▼ Show 20 Lines | def origin_visit_add(self, visits: List[OriginVisit]) -> Iterable[OriginVisit]: | ||||
| ) | ) | ||||
| ) | ) | ||||
| return all_visits | return all_visits | ||||
| def _origin_visit_status_add(self, visit_status: OriginVisitStatus) -> None: | def _origin_visit_status_add(self, visit_status: OriginVisitStatus) -> None: | ||||
| """Add an origin visit status""" | """Add an origin visit status""" | ||||
| if visit_status.type is None: | if visit_status.type is None: | ||||
| origin_row = self._cql_runner.origin_visit_get_one( | visit_row = self._cql_runner.origin_visit_get_one( | ||||
| visit_status.origin, visit_status.visit | visit_status.origin, visit_status.visit | ||||
| ) | ) | ||||
| if origin_row is None: | if visit_row is None: | ||||
| raise StorageArgumentException(f"Unknown origin {visit_status.origin}") | raise StorageArgumentException( | ||||
| visit_status = attr.evolve(visit_status, type=origin_row.type) | f"Unknown origin visit {visit_status.visit} " | ||||
| f"of origin {visit_status.origin}" | |||||
| ) | |||||
| visit_status = attr.evolve(visit_status, type=visit_row.type) | |||||
| self.journal_writer.origin_visit_status_add([visit_status]) | self.journal_writer.origin_visit_status_add([visit_status]) | ||||
| self._cql_runner.origin_visit_status_add_one( | self._cql_runner.origin_visit_status_add_one( | ||||
| converters.visit_status_to_row(visit_status) | converters.visit_status_to_row(visit_status) | ||||
| ) | ) | ||||
| def origin_visit_status_add(self, visit_statuses: List[OriginVisitStatus]) -> None: | def origin_visit_status_add(self, visit_statuses: List[OriginVisitStatus]) -> None: | ||||
| # First round to check existence (fail early if any is ko) | # First round to check existence (fail early if any is ko) | ||||
| ▲ Show 20 Lines • Show All 473 Lines • Show Last 20 Lines | |||||