sql-blocks 1.25.19011745__py3-none-any.whl → 1.25.26011923__py3-none-any.whl

sql_blocks/sql_blocks.py

@@ -470,7 +470,17 @@ class Where:
             values = ','.join(quoted(v) for v in values)
         return cls(f'IN ({values})')
 
+    @classmethod
+    def formula(cls, formula: str):
+        return cls( ExpressionField(formula) )
+
     def add(self, name: str, main: SQLObject):
+        if isinstance(self.expr, ExpressionField):
+            self.expr = self.expr.format(name, main)
+            main.values.setdefault(WHERE, []).append('{} {}'.format(
+                self.prefix, self.expr
+            ))
+            return
         func_type = FUNCTION_CLASS.get(name.lower())
         exists = any(
             main.is_named_field(fld, SELECT)
@@ -566,7 +576,7 @@ class Clause:
         found = re.findall(r'^_\d', name)
         if found:
             name = found[0].replace('_', '')
-        elif main.alias and not is_function():
+        elif '.' not in name and main.alias and not is_function():
             name = f'{main.alias}.{name}'
         return name
 
@@ -1008,8 +1018,11 @@ class SQLParser(Parser):
                     if not key in values:
                         continue
                     separator = self.class_type.get_separator(key)
+                    cls = {
+                        ORDER_BY: OrderBy, GROUP_BY: GroupBy
+                    }.get(key, Field)
                     obj.values[key] = [
-                        Field.format(fld, obj)
+                        cls.format(fld, obj)
                         for fld in re.split(separator, values[key])
                         if (fld != '*' and len(tables) == 1) or obj.match(fld, key)
                     ]
@@ -1416,6 +1429,47 @@ class NotSelectIN(SelectIN):
     condition_class = Not
 
 
+class CTE:
+    prefix = ''
+
+    def __init__(self, name: str, query_list: list[Select]):
+        self.name = name
+        for query in query_list:
+            query.break_lines = False
+        self.query_list = query_list
+
+    def format(self, query: Select) -> str:
+        LINE_MAX_SIZE = 50
+        result, line = [], ''
+        for word in str(query).split(' '):
+            if len(line) >= LINE_MAX_SIZE and word in KEYWORD:
+                result.append(line)
+                line = ''
+            line += word + ' '
+        if line:
+            result.append(line)
+        return '\n\t'.join(result)
+
+    def __str__(self) -> str:
+        return 'WITH {}{} AS (\n\t{}\n)SELECT * FROM {}'.format(
+            self.prefix, self.name, 
+            '\nUNION ALL\n\t'.join(
+                self.format(q) for q in self.query_list
+            ), self.name
+        )
+
+class Recursive(CTE):
+    prefix = 'RECURSIVE '
+
+    def __str__(self) -> str:
+        if len(self.query_list) > 1:
+            alias = self.name[0].lower()
+            self.query_list[-1].values[FROM].append(f', {self.name} {alias}')
+        return super().__str__()
+
+
+# ----- Rules -----
+
 class RulePutLimit(Rule):
     @classmethod
     def apply(cls, target: Select):
@@ -1554,13 +1608,14 @@ def detect(text: str, join_queries: bool = True) -> Select:
     return result
 
 
-if __name__ == '__main__':
-    query = Select(
-        'Tips t',
-        tip=[Field, Lag().over(day=OrderBy).As('last')],
-        diff=[
-            ExpressionField('Round(tip-last, 2) as {f}'),
-            Not.is_null()
-        ]
+if __name__ == "__main__":
+    MY_NAME = 'Júlio Cascalles'
+    # query = Select('SocialMedia s', post=Count, reaction=Sum, user=GroupBy)
+    # print( CTE('Metrics', [query]) )
+    q1 = Select(
+        'SocialMedia me', name=[ eq(MY_NAME), Field ]
+    )
+    q2 = Select(
+        'SocialMedia you', name=Field, id=Where.formula('{af} = n.friend')
     )
-    print(query)
+    print( Recursive('Network', [q1, q2]) )

{sql_blocks-1.25.19011745.dist-info → sql_blocks-1.25.26011923.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sql_blocks
-Version: 1.25.19011745
+Version: 1.25.26011923
 Summary: Allows you to create objects for parts of SQL query commands. Also to combine these objects by joining them, adding or removing parts...
 Home-page: https://github.com/julio-cascalles/sql_blocks
 Author: Júlio Cascalles
@@ -115,6 +115,16 @@ based_on_book=Not.is_null()
 hash_tag=inside(['space', 'monster', 'gore'])
 ```
 
+3.6 -- Combining ExpressionField with Where condition:
+* The **formula** method allows you to write an expression as a condition:
+```
+query=Select(
+    'Folks f2',
+    id=Where.formula('({af} = a.father OR {af} = a.mother)')
+)
+```
+> Results: `WHERE...f2.id = a.father OR f2.id = a.mother`
+
 ---
 ### 4 - A field can be two things at the same time:
 
@@ -631,3 +641,39 @@ For example, if your query is going to run on Oracle, do the following:
 
 `Function.dialect = Dialect.ORACLE`
 
+---
+
+### 17 - CTE and Recursive classes
+
+* **_CTE class_**
+```
+    query = Select(
+        'SocialMedia s', post=Count, reaction=Sum, user=GroupBy
+    )
+    print( CTE('Metrics', [query]) )
+```
+The result is...
+```
+    WITH Metrics AS (
+            SELECT Count(s.post), Sum(s.reaction) FROM SocialMedia s GROUP BY user
+    )SELECT * FROM Metrics
+```
+
+* **_Recursive class_**
+```
+q1 = Select(
+    'SocialMedia me', name=[ eq(MY_NAME), Field ]
+)
+q2 = Select(
+    'SocialMedia you' name=Field, id=Where.formula('{af} = n.friend')
+)
+print( Recursive('Network', [q1, q2]) )
+```
+The result is...
+```
+WITH RECURSIVE Network AS (
+        SELECT me.name FROM SocialMedia me WHERE me.name = 'Júlio Cascalles'
+UNION ALL
+        SELECT you.name FROM SocialMedia you , Network n WHERE  you.id = n.friend
+)SELECT * FROM Network
+```

sql_blocks-1.25.26011923.dist-info/RECORD

@@ -0,0 +1,7 @@
+sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
+sql_blocks/sql_blocks.py,sha256=lVV1Z09JVmwvV0vzKqZVBHpGWpZefD1RsMZy-aIJZSk,53868
+sql_blocks-1.25.26011923.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+sql_blocks-1.25.26011923.dist-info/METADATA,sha256=qvFr99uB0b95uv7MqpvpZaMrH7VrE557bmW4YWFQgnA,16187
+sql_blocks-1.25.26011923.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+sql_blocks-1.25.26011923.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
+sql_blocks-1.25.26011923.dist-info/RECORD,,

sql_blocks-1.25.19011745.dist-info/RECORD

@@ -1,7 +0,0 @@
-sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
-sql_blocks/sql_blocks.py,sha256=lPUpYOLHjvx84EVjlydJK03rm7qG1DWotRtqyldKqvo,52020
-sql_blocks-1.25.19011745.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-sql_blocks-1.25.19011745.dist-info/METADATA,sha256=KPw30aPiITpjBHpfWU9yAuTtr8aQ5cW5cdIdzM4iD0k,15031
-sql_blocks-1.25.19011745.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sql_blocks-1.25.19011745.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
-sql_blocks-1.25.19011745.dist-info/RECORD,,