Skip to content

Commit 38fa6d8

Browse files
committed
Fix mypy issues
Now I can reproduce them.
1 parent 35afb6e commit 38fa6d8

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

src/sql_tstring/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from sql_tstring.t import t, Template as TTemplate
2626

2727
try:
28-
from string.templatelib import Template # type: ignore[import-untyped]
28+
from string.templatelib import Template
2929
except ImportError:
3030

3131
class Template: # type: ignore[no-redef]
@@ -87,7 +87,7 @@ def sql_context(
8787
def sql(
8888
query_or_template: str | Template | TTemplate, values: dict[str, typing.Any] | None = None
8989
) -> tuple[str, list]:
90-
template: Template
90+
template: Template | TTemplate
9191
if isinstance(query_or_template, (Template, TTemplate)) and values is None:
9292
template = query_or_template
9393
elif isinstance(query_or_template, str) and values is not None:

src/sql_tstring/parser.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from sql_tstring.t import Interpolation as TInterpolation, Template as TTemplate
99

1010
try:
11-
from string.templatelib import Interpolation, Template # type: ignore[import-untyped]
11+
from string.templatelib import Interpolation, Template
1212
except ImportError:
1313

1414
class Interpolation: # type: ignore[no-redef]
@@ -355,17 +355,17 @@ class Operator:
355355
type Element = Node | Operator | Part | Placeholder
356356

357357

358-
def parse(template: Template) -> list[Statement]:
358+
def parse(template: Template | TTemplate) -> list[Statement]:
359359
statements = [Statement()]
360360
current_node: Node = statements[0]
361361
_parse_template(template, current_node, statements)
362362
return statements
363363

364364

365-
def _parse_template(template: Template, current_node: Node, statements: list[Statement]) -> None:
365+
def _parse_template(template: Template | TTemplate, current_node: Node, statements: list[Statement]) -> None:
366366
for item in template:
367367
match item:
368-
case Interpolation(value, _, _, _): # type: ignore[misc]
368+
case Interpolation(value, _, _, _):
369369
if isinstance(value, (Template, TTemplate)):
370370
_parse_template(value, current_node, statements)
371371
else:

0 commit comments

Comments
 (0)