Skip to content

Commit 127512c

Browse files
committed
hotfix: get_rows date checker with non-string values
1 parent e2d4032 commit 127512c

2 files changed

Lines changed: 21 additions & 1 deletion

File tree

sqlmodel_crud_utils/a_sync.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,12 @@ async def get_rows(
250250
and is_date(val, fuzzy=False)
251251
):
252252
val = date_parse(val)
253-
stmnt = stmnt.where(getattr(model, model_key) < int(val))
253+
val = (
254+
int(val)
255+
if isinstance(val, str) and val.isdigit()
256+
else val
257+
)
258+
stmnt = stmnt.where(getattr(model, model_key) < val)
254259
elif "__gte" in key:
255260
model_key = key.replace("__gte", "")
256261
val = kwargs.pop(key)
@@ -260,6 +265,11 @@ async def get_rows(
260265
and is_date(val, fuzzy=False)
261266
):
262267
val = date_parse(val)
268+
val = (
269+
int(val)
270+
if isinstance(val, str) and val.isdigit()
271+
else val
272+
)
263273
stmnt = stmnt.where(getattr(model, model_key) > int(val))
264274
sort_desc, sort_field = (
265275
kwargs.pop(x, None) for x in ("sort_desc", "sort_field")

sqlmodel_crud_utils/sync.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,11 @@ def get_rows(
247247
and is_date(val, fuzzy=False)
248248
):
249249
val = date_parse(val)
250+
val = (
251+
int(val)
252+
if isinstance(val, str) and val.isdigit()
253+
else val
254+
)
250255
stmnt = stmnt.where(getattr(model, model_key) < int(val))
251256
elif "__gte" in key:
252257
model_key = key.replace("__gte", "")
@@ -257,6 +262,11 @@ def get_rows(
257262
and is_date(val, fuzzy=False)
258263
):
259264
val = date_parse(val)
265+
val = (
266+
int(val)
267+
if isinstance(val, str) and val.isdigit()
268+
else val
269+
)
260270
stmnt = stmnt.where(getattr(model, model_key) > int(val))
261271
sort_desc, sort_field = (
262272
kwargs.pop(x, None) for x in ("sort_desc", "sort_field")

0 commit comments

Comments
 (0)