Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/in_memory.py
| Show First 20 Lines • Show All 1,183 Lines • ▼ Show 20 Lines | def origin_visit_add(self, origin, date=None, type=None, *, ts=None): | ||||
| 'visit': visit_id | 'visit': visit_id | ||||
| } | } | ||||
| self._origin_visits[origin_id-1].append(visit) | self._origin_visits[origin_id-1].append(visit) | ||||
| visit_ret = { | visit_ret = { | ||||
| 'origin': origin_id, | 'origin': origin_id, | ||||
| 'visit': visit_id, | 'visit': visit_id, | ||||
| } | } | ||||
| self._objects[(origin_id, visit_id)].append( | |||||
| ('origin_visit', None)) | |||||
| if self.journal_writer: | if self.journal_writer: | ||||
| origin = self.origin_get([{'id': origin_id}])[0] | origin = self.origin_get([{'id': origin_id}])[0] | ||||
| del origin['id'] | del origin['id'] | ||||
| self.journal_writer.write_addition('origin_visit', { | self.journal_writer.write_addition('origin_visit', { | ||||
| **visit, 'origin': origin}) | **visit, 'origin': origin}) | ||||
| return visit_ret | return visit_ret | ||||
| ▲ Show 20 Lines • Show All 69 Lines • ▼ Show 20 Lines | def origin_visit_upsert(self, visits): | ||||
| visit = visit.copy() | visit = visit.copy() | ||||
| visit['origin'] = self.origin_get([{'id': visit['origin']}])[0] | visit['origin'] = self.origin_get([{'id': visit['origin']}])[0] | ||||
| del visit['origin']['id'] | del visit['origin']['id'] | ||||
| self.journal_writer.write_addition('origin_visit', visit) | self.journal_writer.write_addition('origin_visit', visit) | ||||
| for visit in visits: | for visit in visits: | ||||
| origin_id = visit['origin'] | origin_id = visit['origin'] | ||||
| visit_id = visit['visit'] | visit_id = visit['visit'] | ||||
| self._objects[(origin_id, visit_id)].append( | |||||
| ('origin_visit', None)) | |||||
| while len(self._origin_visits[origin_id-1]) < visit_id: | while len(self._origin_visits[origin_id-1]) < visit_id: | ||||
| self._origin_visits[origin_id-1].append(None) | self._origin_visits[origin_id-1].append(None) | ||||
| visit = self._origin_visits[origin_id-1][visit_id-1] = visit | visit = self._origin_visits[origin_id-1][visit_id-1] = visit | ||||
| def origin_visit_get(self, origin, last_visit=None, limit=None): | def origin_visit_get(self, origin, last_visit=None, limit=None): | ||||
| """Retrieve all the origin's visit's information. | """Retrieve all the origin's visit's information. | ||||
| Args: | Args: | ||||
| ▲ Show 20 Lines • Show All 319 Lines • Show Last 20 Lines | |||||