sqlscope 1.0.4__tar.gz → 1.0.6__tar.gz

{sqlscope-1.0.4 → sqlscope-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlscope
-Version: 1.0.4
+Version: 1.0.6
 Summary: This project extracts catalog and query metadata from a SQL query.
 Project-URL: Repository, https://github.com/DavidePonzini/sqlscope
 Project-URL: Documentation, https://sqlscope.readthedocs.io/en/latest/index.html

{sqlscope-1.0.4 → sqlscope-1.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sqlscope"
-version = "1.0.4"
+version = "1.0.6"
 authors = [
   { name="Davide Ponzini", email="davide.ponzini95@gmail.com" },
 ]

{sqlscope-1.0.4 → sqlscope-1.0.6}/src/sqlscope/query/set_operations/select.py

@@ -78,8 +78,10 @@ class Select(SetOperation, TokenizedSQL):
 
                 subquery_sql = expr.this.sql()
                 subquery = Select(subquery_sql, catalog=self.catalog, search_path=self.search_path)
+                output = subquery.output
+                output.name = table_name_out
 
-                result.append(subquery.output)
+                result.append(output)
             
             # Table: look it up in the IN catalog
             elif isinstance(expr, exp.Table):
@@ -104,7 +106,6 @@ class Select(SetOperation, TokenizedSQL):
                     result.append(table)
         
         from_expr = self.ast.args.get('from_')
-        
 
         if from_expr:
             add_tables_from_expression(from_expr.this)
@@ -480,7 +481,7 @@ class Select(SetOperation, TokenizedSQL):
                         col_name = util.ast.column.get_real_name(col)
 
                         # Resolve which table this column belongs to
-                        table_idx = self._get_table_idx_for_column(col)
+                        table_idx = self._get_table_idx_for_column(col) or 0    # default to first table if not found, for incorrect queries that group by a column not in any table
                         
                         group_by_cols.add(ConstraintColumn(col_name, table_idx))
                 # Add GROUP BY constraint