Skip to content

Commit 9584a13

Browse files
treyspgeorgesittas
andcommitted
Clean up from review
Co-authored-by: Jo <[email protected]>
1 parent a9579dd commit 9584a13

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

sqlglot/dialects/duckdb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,7 @@ def _escape_regex_metachars(
304304
"]": "\\]",
305305
}
306306

307-
if isinstance(delimiters, exp.Literal) and delimiters.is_string:
307+
if delimiters.is_string:
308308
literal_value = delimiters.this
309309
escaped_literal = "".join(REGEX_LITERAL_ESCAPES.get(ch, ch) for ch in literal_value)
310310
return self.sql(exp.Literal.string(escaped_literal))

sqlglot/generator.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5411,13 +5411,12 @@ def uuid_sql(self, expression: exp.Uuid) -> str:
54115411
return uuid_func_sql
54125412

54135413
def initcap_sql(self, expression: exp.Initcap) -> str:
5414-
delimiters = expression.args.get("expression")
5414+
delimiters = expression.expression
54155415
delimiters_sql = None
54165416

54175417
# do not generate delimiters arg if we are round-tripping from default delimiters
54185418
if (
54195419
delimiters
5420-
and isinstance(delimiters, exp.Literal)
54215420
and delimiters.is_string
54225421
and delimiters.this == self.dialect.INITCAP_DEFAULT_DELIMITER_CHARS
54235422
):

tests/dialects/test_dialect.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4326,7 +4326,7 @@ def duckdb_regex_literal_sql(delimiters: str) -> str:
43264326
return exp.Literal.string(escaped_literal).sql("duckdb")
43274327

43284328
# default delimiters not present in roundtrip
4329-
for dialect in delimiter_chars.keys():
4329+
for dialect in delimiter_chars:
43304330
with self.subTest(
43314331
f"Testing INITCAP roundtrips for {dialect or 'default'} dialect without default delimiters included"
43324332
):

0 commit comments

Comments
 (0)