40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import bson
|
|
import pydantic
|
|
|
|
from app.storage.mongo import database
|
|
from bson import codec_options
|
|
|
|
|
|
collection = database.get_collection("experiments", codec_options=codec_options.CodecOptions(tz_aware=True))
|
|
|
|
|
|
class Experiment(pydantic.BaseModel):
|
|
name: str
|
|
enabled: bool
|
|
condition: str
|
|
project: str
|
|
stage: str
|
|
_id: bson.ObjectId|None = None
|
|
|
|
|
|
async def create(experiment: Experiment) -> str:
|
|
result = await collection.insert_one(experiment.model_dump())
|
|
return result.inserted_id
|
|
|
|
|
|
async def update(project: str, stage: str, name: str, enabled: bool, condition: str) -> bool:
|
|
result = await collection.update_one({'project': project, 'stage': stage, 'name': name}, {'$set': {'enabled': enabled, 'condition': condition}})
|
|
return result.modified_count != 0
|
|
|
|
|
|
async def delete(project: str, stage: str, name: str) -> bool:
|
|
result = await collection.delete_one({'project': project, 'stage': stage, 'name': name})
|
|
return result.deleted_count != 0
|
|
|
|
|
|
async def get(project: str, stage: str) -> list[Experiment]:
|
|
result = []
|
|
async for item in collection.find({'stage': stage, 'project': project}):
|
|
result.append(Experiment.model_validate(item))
|
|
return result
|