| import datetime | |
| from bson import ObjectId, DatetimeMS | |
| class TestCreator: | |
| def __init__(self, db): | |
| self.db = db | |
| self.test_obj = None | |
| def create(self, record_ids: list[str]) -> ObjectId: | |
| record_results = {} | |
| for batch_id in record_ids: | |
| record_results[batch_id] = {"status": "pending"} | |
| test_obj = { | |
| "status": "pending", | |
| "created_at": datetime.datetime.now(datetime.UTC), | |
| "pipeline_version": "1.0.0", | |
| "record_results": record_results | |
| } | |
| insert_result = self.db.test_evaluation.insert_one(test_obj) | |
| return insert_result.inserted_id | |