sql-blocks 1.2025.625__py3-none-any.whl → 1.2025.628__py3-none-any.whl

sql_blocks/sql_blocks.py

@@ -330,16 +330,21 @@ class DateDiff(Function):
     append_param = True
 
     def __str__(self) -> str:
-        def is_field_or_func(name: str) -> bool:
+        def is_field_or_func(obj) -> bool:
+            if not isinstance(obj, str):
+                return True
             candidate = re.sub(
-                '[()]', '', name.split('.')[-1]
+                '[()]', '', obj.split('.')[-1]
             )
+            print(f'---------------> #{candidate=}#')
             return candidate.isidentifier()
+        self.params = [
+            p if is_field_or_func(p) else f"'{p}'"
+            for p in self.params
+        ]
         if self.dialect != Dialect.SQL_SERVER:
-            params = [str(p) for p in self.params]
             return ' - '.join(
-                p if is_field_or_func(p) else f"'{p}'"
-                for p in params
+                str(p) for p in self.params
             )  # <====  Date subtract
         return super().__str__()
 
@@ -654,6 +659,8 @@ class Not(Where):
 
 
 class Case:
+    break_lines = True
+
     def __init__(self, field: str):
         self.__conditions = {}
         self.default = None
@@ -680,16 +687,24 @@ class Case:
         return self
     
     def format(self, name: str, field: str='') -> str:
+        TABULATION = '\t\t' if self.break_lines else ' '
+        LINE_BREAK = '\n' if self.break_lines else ' '
         default = self.default
         if not field:
             field = self.field
-        return 'CASE \n{}\n\tEND AS {}'.format(
-            '\n'.join(
-                f'\t\tWHEN {field} {cond.content} THEN {res}'
+        return 'CASE{brk}{cond}{df}{tab}END{alias}'.format(
+            brk=LINE_BREAK,
+            cond=LINE_BREAK.join(
+                f'{TABULATION}WHEN {field} {cond.content} THEN {res}'
                 for res, cond in self.__conditions.items()
-            ) + (f'\n\t\tELSE {default}' if default else ''),
-            name
+            ),
+            df=f'{LINE_BREAK}{TABULATION}ELSE {default}' if not default is None else '',
+            tab='\n\t' if self.break_lines else ' ',
+            alias=f' AS {name}' if name else ''
         )
+    
+    def __str__(self):
+        return self.format('', self.field)
 
     def add(self, name: str, main: SQLObject):
         main.values.setdefault(SELECT, []).append(
@@ -2009,7 +2024,7 @@ class CTE(Select):
         # ---------------------------------------------------------
         return 'WITH {}{} AS (\n    {}\n){}'.format(
             self.prefix, self.table_name, 
-            '\nUNION ALL\n    '.join(
+            '\n\tUNION ALL\n    '.join(
                 justify(q) for q in self.query_list
             ), super().__str__() if self.show_query else ''
         )
@@ -2081,18 +2096,33 @@ class CTEFactory:
         self.cte_list = [
             CTE(alias, [
                 Select.parse(query)[0]
+                for query in elements
             ])
-            for alias, query in summary.items()
+            for alias, elements in summary.items()
         ]
 
     def __str__(self):
         CTE.show_query = False
         lines = [str(cte) for cte in self.cte_list]
-        return ',\n'.join(lines) + '\n' + str(self.main)
+        result = ',\n'.join(lines) + '\n' + str(self.main)
+        CTE.show_query = True
+        return result    
 
     @staticmethod
-    def extract_subqueries(txt: str) -> dict:
+    def extract_subqueries(txt: str) -> dict:        
         result = {}
+        # ---------------------------------------------------
+        def clean_subquery(source: list) -> str:
+            while source:
+                if source[0].upper() == 'SELECT':
+                    break
+                word = source.pop(0)
+                if word.upper() in ('FROM', 'JOIN'):
+                    result[MAIN_TAG] += f' {word}'
+            return ' '.join(source)
+        def balanced_parentheses(expr: str) -> bool:
+            return expr.count('(') == expr.count(')')
+        # ---------------------------------------------------
         for found in re.finditer(r'(FROM|JOIN)\s*[(]\s*SELECT', txt, re.IGNORECASE):
             start = found.start()
             alias = ''
@@ -2102,16 +2132,23 @@ class CTEFactory:
                 if not found:
                     break
                 end = found.end() + pos
-                elements = txt[start: end].split()
-                if '(' not in elements[-3]:
-                    _, alias = elements[-2:]
+                last = end
+                if balanced_parentheses(txt[start: end]):
+                    pos += found.start()
+                    alias = re.findall(r'\s*(\w+)$', txt[pos: end])[0]
+                    end = pos
                 pos = end
-            first_word = elements.pop(0)
             if not result:
                 result[MAIN_TAG] = txt[:start]
-            result[MAIN_TAG] += f' {first_word} {alias} {alias}'
-            result[alias] = ' '.join(elements[1: -3])
-        result[MAIN_TAG] += txt[end:]
+            query_list = [
+                clean_subquery( expr.split() )
+                for expr in  re.split(
+                    r'\bUNION\b', txt[start: end], re.IGNORECASE
+                )
+            ]
+            result[MAIN_TAG] += f' {alias} {alias}'
+            result[alias] = query_list
+        result[MAIN_TAG] += txt[last:]
         return result
 
 
@@ -2259,3 +2296,12 @@ def detect(text: str, join_queries: bool = True, format: str='') -> Select | lis
         result += query
     return result
 # ===========================================================================================//
+
+if __name__ == "__main__":
+    
+    print(
+        Select(
+            'Emprestimo e',
+            _=Sum(Case('atraso').when(gt(60), 25).when(lt(15), 5).else_value(10)).As('multa', OrderBy)
+        )
+    )

{sql_blocks-1.2025.625.dist-info → sql_blocks-1.2025.628.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sql_blocks
-Version: 1.2025.625
+Version: 1.2025.628
 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

sql_blocks-1.2025.628.dist-info/RECORD

@@ -0,0 +1,7 @@
+sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
+sql_blocks/sql_blocks.py,sha256=gbihrdYwzwGUMDO2pXSeIOmxxaOX9A-hXoz-qZv_4Ls,78290
+sql_blocks-1.2025.628.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+sql_blocks-1.2025.628.dist-info/METADATA,sha256=mhDyYy5hxk3A-I7CKjekvhTBizZVXpkUA5RtKMHcIks,23328
+sql_blocks-1.2025.628.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+sql_blocks-1.2025.628.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
+sql_blocks-1.2025.628.dist-info/RECORD,,

sql_blocks-1.2025.625.dist-info/RECORD

@@ -1,7 +0,0 @@
-sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
-sql_blocks/sql_blocks.py,sha256=J7zEJ5JNRxI3F-7TIypb0myb5OgdE5Stv4boZCTVBLM,76610
-sql_blocks-1.2025.625.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-sql_blocks-1.2025.625.dist-info/METADATA,sha256=I6KTi5S_usvCjJEQl6lm8LG4DaxHhX0NlGRwhHpKBT8,23328
-sql_blocks-1.2025.625.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sql_blocks-1.2025.625.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
-sql_blocks-1.2025.625.dist-info/RECORD,,