feat: Remove old articles from database
Some checks failed
Create and publish a Docker image / build-and-push-image (push) Has been cancelled
Some checks failed
Create and publish a Docker image / build-and-push-image (push) Has been cancelled
This commit is contained in:
parent
90a419eebe
commit
434773025b
3 changed files with 50 additions and 4 deletions
|
|
@ -1,7 +1,6 @@
|
|||
from datetime import timezone
|
||||
from datetime import datetime, timezone
|
||||
from typing import override
|
||||
|
||||
from datetime import datetime
|
||||
from supabase import Client
|
||||
|
||||
from prophet.config import SupaConfig
|
||||
|
|
@ -59,6 +58,28 @@ class ImprovementSupaRepo(IImprovementRepo):
|
|||
.data
|
||||
]
|
||||
|
||||
@override
|
||||
def remove(self, id: str) -> Improvement:
|
||||
resp = (
|
||||
self.client.table(self.config.TABLE).delete().eq("uuid", id).execute().data
|
||||
)
|
||||
if not resp:
|
||||
raise ValueError
|
||||
return self._from_tbl_row(resp[0])
|
||||
|
||||
@override
|
||||
def remove_all(self, ids: list[str]) -> list[Improvement]:
|
||||
resp = (
|
||||
self.client.table(self.config.TABLE)
|
||||
.delete()
|
||||
.in_("uuid", ids)
|
||||
.execute()
|
||||
.data
|
||||
)
|
||||
if not resp:
|
||||
raise ValueError
|
||||
return [self._from_tbl_row(item) for item in resp]
|
||||
|
||||
def _to_tbl_row(self, imp: Improvement) -> dict[str, str | int]:
|
||||
return {
|
||||
"uuid": imp.id,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue