removes type restriction fixes #333

example/models/customer_revenue_by_day.sql

@@ -32,4 +32,4 @@ WHERE
   o.ds BETWEEN @start_ds AND @end_ds
 GROUP BY
   o.customer_id,
-  o.ds
+  o.ds

example/models/customers.sql

@@ -7,4 +7,4 @@ MODEL (
 
 SELECT DISTINCT
   customer_id::INT AS customer_id
-FROM sushi.orders AS o
+FROM sushi.orders AS o

example/models/raw_items.py → example/models/items.py

@@ -47,7 +47,7 @@
 @model(
     """
     MODEL(
-        name raw.items,
+        name sushi.items,
         kind incremental,
         time_column ds,
         start 'Jan 1 2022',

example/models/raw_order_items.py → example/models/order_items.py

@@ -12,10 +12,10 @@
 @model(
     """
     MODEL(
-        name raw.order_items,
+        name sushi.order_items,
         kind incremental,
         time_column ds,
-        depends_on [raw.orders, raw.items],
+        depends_on [sushi.orders, sushi.items],
         cron '@daily',
         batch_size 30,
         columns (
@@ -40,7 +40,7 @@ def execute(
     dfs = []
 
     raw_orders = (
-        snapshots["raw.orders"].table_name if snapshots else mapping["raw.orders"]
+        snapshots["sushi.orders"].table_name if snapshots else mapping["sushi.orders"]
     )
 
     for dt in iter_dates(start, end):

example/models/raw_orders.py → example/models/orders.py

@@ -13,8 +13,9 @@
 
 @model(
     """
+    -- Table of sushi orders.
     MODEL(
-        name raw.orders,
+        name sushi.orders,
         kind incremental,
         time_column ds,
         start '2022-01-01',

example/models/top_waiters.sql

@@ -17,4 +17,4 @@ WHERE
   )
 ORDER BY
   revenue DESC
-LIMIT 10
+LIMIT 10

example/models/waiter_revenue_by_day.sql

@@ -20,4 +20,4 @@ WHERE
   o.ds BETWEEN @start_ds AND @end_ds
 GROUP BY
   o.waiter_id,
-  o.ds
+  o.ds

example/models/waiters.sql

@@ -11,4 +11,4 @@ SELECT DISTINCT
   ds::TEXT AS ds
 FROM sushi.orders AS o
 WHERE
-  ds BETWEEN @start_ds AND @end_ds
+  ds BETWEEN @start_ds AND @end_ds

setup.py

@@ -32,7 +32,7 @@
         "requests",
         "rich",
         "ruamel.yaml",
-        "sqlglot>=10.2.0",
+        "sqlglot>=10.2.1",
     ],
     extras_require={
         "dev": [

sqlmesh/core/engine_adapter.py

@@ -92,6 +92,8 @@ def replace_query(self, table_name: str, query_or_df: QueryOrDF):
         if self.supports_partitions:
             self.insert_overwrite(table_name, query_or_df)
         else:
+            if isinstance(query_or_df, pd.DataFrame):
+                query_or_df = next(pandas_to_sql(query_or_df))
             self.execute(
                 exp.Create(
                     kind="TABLE",
@@ -104,29 +106,42 @@ def replace_query(self, table_name: str, query_or_df: QueryOrDF):
     def create_table(
         self,
         table_name: str,
-        columns: t.Dict[str, exp.DataType],
+        query_or_columns: Query | t.Dict[str, exp.DataType],
         exists: bool = True,
         **kwargs,
     ) -> None:
-        """Create a table using a DDL statement
+        """Create a table using a DDL statement or a CTAS.
+
+        If a query is passed in instead of column type map, CREATE TABLE AS will be used.
 
         Args:
-            table_name: The name of the table to create. Can be fully qualified or just table name
-            columns: A mapping between the column name and its data type
-            exists: Indicates if you to include an IF EXISTS check
+            table_name: The name of the table to create. Can be fully qualified or just table name.
+            query_columns: A query or mapping between the column name and its data type.
+            exists: Indicates if you to include an IF EXISTS check.
             kwargs: Optional create table properties.
         """
         properties = self._create_table_properties(**kwargs)
 
-        schema = exp.Schema(
-            this=exp.to_table(table_name),
-            expressions=[
-                exp.ColumnDef(this=exp.to_identifier(column), kind=kind)
-                for column, kind in columns.items()
-            ],
-        )
+        query = None
+        schema: t.Optional[exp.Schema | exp.Table] = exp.to_table(table_name)
+
+        if isinstance(query_or_columns, dict):
+            schema = exp.Schema(
+                this=schema,
+                expressions=[
+                    exp.ColumnDef(this=exp.to_identifier(column), kind=kind)
+                    for column, kind in query_or_columns.items()
+                ],
+            )
+        else:
+            query = query_or_columns
+
         create_expression = exp.Create(
-            this=schema, kind="TABLE", exists=exists, properties=properties
+            this=schema,
+            kind="TABLE",
+            exists=exists,
+            properties=properties,
+            expression=query,
         )
         self.execute(create_expression)