Skip to content

Commit 71d2acd

Browse files
committed
style: Apply black and isort formatting
1 parent ebb888d commit 71d2acd

5 files changed

Lines changed: 23 additions & 10 deletions

File tree

sql_redis/parser.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -665,7 +665,9 @@ def _parse_date_to_timestamp(self, value: str) -> int | None:
665665
# No timezone - treat as UTC
666666
# Handle optional milliseconds
667667
if "." in normalized:
668-
dt = datetime.strptime(normalized.split(".")[0], "%Y-%m-%dT%H:%M:%S")
668+
dt = datetime.strptime(
669+
normalized.split(".")[0], "%Y-%m-%dT%H:%M:%S"
670+
)
669671
else:
670672
dt = datetime.strptime(normalized, "%Y-%m-%dT%H:%M:%S")
671673
dt = dt.replace(tzinfo=timezone.utc)

sql_redis/translator.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,10 @@
66

77
from sql_redis.analyzer import AnalyzedQuery, Analyzer
88
from sql_redis.parser import (
9+
SQL_TO_REDIS_DATE_FUNCTIONS,
910
Condition,
1011
ParsedQuery,
1112
SQLParser,
12-
SQL_TO_REDIS_DATE_FUNCTIONS,
1313
)
1414
from sql_redis.query_builder import QueryBuilder
1515
from sql_redis.schema import AsyncSchemaRegistry, SchemaRegistry
@@ -276,7 +276,9 @@ def _build_aggregate(
276276
if redis_func:
277277
if date_func.function == "DATE_FORMAT" and date_func.format_string:
278278
# DATE_FORMAT(field, format) -> timefmt(@field, format)
279-
expression = f'{redis_func}(@{date_func.field}, "{date_func.format_string}")'
279+
expression = (
280+
f'{redis_func}(@{date_func.field}, "{date_func.format_string}")'
281+
)
280282
else:
281283
# Simple date extraction: YEAR(field) -> year(@field)
282284
expression = f"{redis_func}(@{date_func.field})"

tests/test_date_fields.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,4 +159,3 @@ def test_date_between_generates_numeric_range(self, date_translator, date_index)
159159
assert result.command == "FT.SEARCH"
160160
# Should contain numeric range
161161
assert "@created_at:[1704067200 1706659200]" in result.query_string
162-

tests/test_date_functions.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import pytest
44

5-
from sql_redis.parser import SQLParser, DateFunctionSpec
5+
from sql_redis.parser import DateFunctionSpec, SQLParser
66

77

88
class TestDateFunctionParsing:
@@ -116,8 +116,19 @@ def date_index(self, redis_client):
116116
except redis_lib.ResponseError:
117117
pass
118118
redis_client.execute_command(
119-
"FT.CREATE", index_name, "ON", "HASH", "PREFIX", "1", "event:",
120-
"SCHEMA", "created_at", "NUMERIC", "SORTABLE", "name", "TEXT",
119+
"FT.CREATE",
120+
index_name,
121+
"ON",
122+
"HASH",
123+
"PREFIX",
124+
"1",
125+
"event:",
126+
"SCHEMA",
127+
"created_at",
128+
"NUMERIC",
129+
"SORTABLE",
130+
"name",
131+
"TEXT",
121132
)
122133
return index_name
123134

@@ -193,4 +204,3 @@ def test_group_by_year_month(self, date_translator, date_index):
193204
assert "GROUPBY" in result.args
194205
assert "REDUCE" in result.args
195206
assert "COUNT" in result.args
196-

uv.lock

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)