diff --git a/src/requela/transformer.py b/src/requela/transformer.py index e7953ea..ad612a4 100644 --- a/src/requela/transformer.py +++ b/src/requela/transformer.py @@ -115,6 +115,12 @@ def DATE(self, token): def UNQUOTED_VAL(self, token): return str(token) + def QUOTED_STRING(self, token): + return str(token)[1:-1] + + def DOUBLE_QUOTED_STRING(self, token): + return str(token)[1:-1] + def UUID(self, token): return str(token) diff --git a/tests/sqlalchemy/test_comparison.py b/tests/sqlalchemy/test_comparison.py index 4f1b98a..d64396d 100644 --- a/tests/sqlalchemy/test_comparison.py +++ b/tests/sqlalchemy/test_comparison.py @@ -13,6 +13,8 @@ ("model", "field", "value", "expected"), [ (User, "name", "Ratatouille 123", select(User).filter(User.name == "Ratatouille 123")), + (User, "name", "'Ratatouille 123'", select(User).filter(User.name == "Ratatouille 123")), + (User, "name", '"Ratatouille 123"', select(User).filter(User.name == "Ratatouille 123")), (User, "age", 25, select(User).filter(User.age == 25)), (Account, "name", 25, select(Account).filter(Account.name == "25")), (Account, "balance", 25.13, select(Account).filter(Account.balance == 25.13)),