Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/mongo/backend.py
| Show All 20 Lines | |||||
| ) | ) | ||||
| class ProvenanceStorageMongoDb: | class ProvenanceStorageMongoDb: | ||||
| def __init__(self, db: pymongo.database.Database): | def __init__(self, db: pymongo.database.Database): | ||||
| self.db = db | self.db = db | ||||
| def content_add( | def content_add( | ||||
| self, cnts: Union[Iterable[Sha1Git], Dict[Sha1Git, datetime]] | self, cnts: Union[Iterable[Sha1Git], Dict[Sha1Git, Optional[datetime]]] | ||||
| ) -> bool: | ) -> bool: | ||||
| data = cnts if isinstance(cnts, dict) else dict.fromkeys(cnts) | data = cnts if isinstance(cnts, dict) else dict.fromkeys(cnts) | ||||
| existing = { | existing = { | ||||
| x["sha1"]: x | x["sha1"]: x | ||||
| for x in self.db.content.find( | for x in self.db.content.find( | ||||
| {"sha1": {"$in": list(data)}}, {"sha1": 1, "ts": 1, "_id": 1} | {"sha1": {"$in": list(data)}}, {"sha1": 1, "ts": 1, "_id": 1} | ||||
| ) | ) | ||||
| } | } | ||||
| ▲ Show 20 Lines • Show All 106 Lines • ▼ Show 20 Lines | def content_get(self, ids: Iterable[Sha1Git]) -> Dict[Sha1Git, datetime]: | ||||
| x["sha1"]: datetime.fromtimestamp(x["ts"], timezone.utc) | x["sha1"]: datetime.fromtimestamp(x["ts"], timezone.utc) | ||||
| for x in self.db.content.find( | for x in self.db.content.find( | ||||
| {"sha1": {"$in": list(ids)}, "ts": {"$ne": None}}, | {"sha1": {"$in": list(ids)}, "ts": {"$ne": None}}, | ||||
| {"sha1": 1, "ts": 1, "_id": 0}, | {"sha1": 1, "ts": 1, "_id": 0}, | ||||
| ) | ) | ||||
| } | } | ||||
| def directory_add( | def directory_add( | ||||
| self, dirs: Union[Iterable[Sha1Git], Dict[Sha1Git, datetime]] | self, dirs: Union[Iterable[Sha1Git], Dict[Sha1Git, Optional[datetime]]] | ||||
| ) -> bool: | ) -> bool: | ||||
| data = dirs if isinstance(dirs, dict) else dict.fromkeys(dirs) | data = dirs if isinstance(dirs, dict) else dict.fromkeys(dirs) | ||||
| existing = { | existing = { | ||||
| x["sha1"]: x | x["sha1"]: x | ||||
| for x in self.db.directory.find( | for x in self.db.directory.find( | ||||
| {"sha1": {"$in": list(data)}}, {"sha1": 1, "ts": 1, "_id": 1} | {"sha1": {"$in": list(data)}}, {"sha1": 1, "ts": 1, "_id": 1} | ||||
| ) | ) | ||||
| } | } | ||||
| ▲ Show 20 Lines • Show All 300 Lines • Show Last 20 Lines | |||||