Allow columnfactory to accept item-style lookups.

Fixes #2776
coleifer · Sep 11, 2023 · fda81ca · fda81ca
1 parent 2ed7181
commit fda81ca
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/peewee.py b/peewee.py
@@ -757,6 +757,7 @@ def __init__(self, node):
 
     def __getattr__(self, attr):
         return Column(self.node, attr)
+    __getitem__ = __getattr__
 
 
 class _DynamicColumn(object):

diff --git a/tests/sql.py b/tests/sql.py
@@ -30,6 +30,14 @@ def test_select(self):
             'FROM "users" AS "t1" '
             'WHERE ("t1"."username" = ?)'), ['foo'])
 
+        query = (User
+                 .select(User.c['id'], User.c['username'])
+                 .where(User.c['username'] == 'test'))
+        self.assertSQL(query, (
+            'SELECT "t1"."id", "t1"."username" '
+            'FROM "users" AS "t1" '
+            'WHERE ("t1"."username" = ?)'), ['test'])
+
     def test_select_extend(self):
         query = User.select(User.c.id, User.c.username)
         self.assertSQL(query, (