experiments/bees-knees
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: Limit repository get_all to maximum entries

Browse source 
Can simply pass an int and will limit the sql to that amount.
This commit is contained in:
Marty Oehme 2025-07-22 05:30:26 +02:00
parent 434773025b
commit a8958c76fb
Signed by: Marty
GPG key ID: 4E535BC19C61886E
2 changed files with 18 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
prophet/domain/improvement_repo.py
View file

@ -17,7 +17,7 @@ class IImprovementRepo(Protocol):
def get(self, id: str) -> Improvement:
raise NotImplementedError
def get_all(self) -> list[Improvement]:
def get_all(self, last_n: int | None = None) -> list[Improvement]:
raise NotImplementedError
def remove(self, id: str) -> Improvement:

26
prophet/infra/improvement_supa_repo.py
View file

@ -48,15 +48,22 @@ class ImprovementSupaRepo(IImprovementRepo):
)
@override
def get_all(self) -> list[Improvement]:
return [
self._from_tbl_row(row)
for row in self.client.table(self.config.TABLE)
.select("*")
.order("date_orig_ts", desc=True)
.execute()
.data
]
def get_all(self, last_n: int | None = None) -> list[Improvement]:
if not last_n:
sql = (
self.client.table(self.config.TABLE)
.select("*")
.order("date_orig_ts", desc=True)
)
else:
sql = (
self.client.table(self.config.TABLE)
.select("*")
.order("date_orig_ts", desc=True)
.limit(last_n)
)
return [self._from_tbl_row(row) for row in sql.execute().data]
@override
def remove(self, id: str) -> Improvement:
@ -110,6 +117,7 @@ class ImprovementSupaRepo(IImprovementRepo):
if __name__ == "__main__":
# response = supabase.table("improvements").select("*").execute()
repo = ImprovementSupaRepo()
print("latest entries:\n- ", "\n- ".join([imp.title for imp in repo.get_all(3)]))
# from prophet.app import grab_latest_originals
# latest = grab_latest_originals()