Factor star into its own node class

coleifer · Oct 20, 2023 · 8e539de · 8e539de
1 parent f400b68
commit 8e539de
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/peewee.py b/peewee.py
@@ -780,6 +780,13 @@ def __get__(self, instance, instance_type=None):
         return self
 
 
+class Star(Node):
+    def __init__(self, source):
+        self.source = source
+    def __sql__(self, ctx):
+        return ctx.sql(QualifiedNames(self.source)).literal('.*')
+
+
 class Source(Node):
     c = _DynamicColumn()
 
@@ -798,7 +805,7 @@ def select(self, *columns):
 
     @property
     def __star__(self):
-        return NodeList((QualifiedNames(self), SQL('.*')), glue='')
+        return Star(self)
 
     def join(self, dest, join_type=JOIN.INNER, on=None):
         return Join(self, dest, join_type, on)