sql-blocks 1.25.33022258__py3-none-any.whl → 1.25.38021329__py3-none-any.whl

sql_blocks/sql_blocks.py

@@ -82,8 +82,14 @@ class SQLObject:
         return KEYWORD[key][0].format(appendix.get(key, ''))
 
     @staticmethod
-    def is_named_field(fld: str, key: str) -> bool:
-        return key == SELECT and re.search(r'\s+as\s+|\s+AS\s+', fld)
+    def is_named_field(fld: str, name: str='') -> bool:
+        return re.search(fr'(\s+as\s+|\s+AS\s+){name}', fld)
+
+    def has_named_field(self, name: str) -> bool:
+        return any(
+            self.is_named_field(fld, name)
+            for fld in self.values.get(SELECT, [])
+        )
 
     def diff(self, key: str, search_list: list, exact: bool=False) -> set:
         def disassemble(source: list) -> list:
@@ -100,7 +106,9 @@ class SQLObject:
         def field_set(source: list) -> set:
             return set(
                 (
-                    fld if self.is_named_field(fld, key) else
+                    fld 
+                    if key == SELECT and self.is_named_field(fld, key) 
+                    else
                     re.sub(pattern, '', cleanup(fld))
                 )
                 for string in disassemble(source)
@@ -150,7 +158,7 @@ class Field:
         name = name.strip()
         if name in ('_', '*'):
             name = '*'
-        elif not is_const():
+        elif not is_const() and not main.has_named_field(name):
             name = f'{main.alias}.{name}'
         if Function in cls.__bases__:
             name = f'{cls.__name__}({name})'
@@ -188,15 +196,35 @@ class Dialect(Enum):
     POSTGRESQL = 3
     MYSQL = 4
 
+SQL_TYPES = 'CHAR INT DATE FLOAT ANY'.split()
+CHAR, INT, DATE, FLOAT, ANY  =  SQL_TYPES
+
 class Function:
     dialect = Dialect.ANSI
+    inputs = None
+    output = None
+    separator = ', '
+    auto_convert = True
+    append_param = False
 
     def __init__(self, *params: list):
+        def set_func_types(param):
+            if self.auto_convert and isinstance(param, Function):
+                func = param
+                main_param = self.inputs[0]
+                unfriendly = all([
+                    func.output != main_param,
+                    func.output != ANY,
+                    main_param  != ANY
+                ])
+                if unfriendly:
+                    return Cast(func, main_param)
+            return param
         # --- Replace class methods by instance methods: ------
         self.add = self.__add
         self.format = self.__format
         # -----------------------------------------------------
-        self.params = [str(p) for p in params]
+        self.params = [set_func_types(p) for p in params]
         self.field_class = Field
         self.pattern = self.get_pattern()
         self.extra = {}
@@ -213,14 +241,26 @@ class Function:
     def __str__(self) -> str:
         return self.pattern.format(
             func_name=self.__class__.__name__,
-            params=', '.join(self.params)
+            params=self.separator.join(str(p) for p in self.params)
         )
 
+    def set_main_param(self, name: str, main: SQLObject) -> bool:
+        nested_functions = [
+            param for param in self.params if isinstance(param, Function)
+        ]
+        for func in nested_functions:
+            if func.inputs:
+                func.set_main_param(name, main)
+                return
+        new_params = [Field.format(name, main)]
+        if self.append_param:
+            self.params += new_params
+        else:
+            self.params = new_params + self.params         
+
     def __format(self, name: str, main: SQLObject) -> str:
         if name not in '*_':
-            self.params = [
-                Field.format(name, main)
-            ] + self.params
+            self.set_main_param(name, main)
         return str(self)
 
     @classmethod
@@ -240,6 +280,9 @@ class Function:
 
 # ---- String Functions: ---------------------------------
 class SubString(Function):
+    inputs = [CHAR, INT, INT]
+    output = CHAR
+
     def get_pattern(self) -> str:
         if self.dialect in (Dialect.ORACLE, Dialect.MYSQL):
             return 'Substr({params})'
@@ -247,31 +290,55 @@ class SubString(Function):
 
 # ---- Numeric Functions: --------------------------------
 class Round(Function):
-    ...
+    inputs = [FLOAT]
+    output = FLOAT
 
 # --- Date Functions: ------------------------------------
 class DateDiff(Function):
-    def get_pattern(self) -> str:
+    inputs = [DATE]
+    output = DATE
+    append_param = True
+
+    def __str__(self) -> str:
         def is_field_or_func(name: str) -> bool:
-            return re.sub('[()]', '', name).isidentifier()
+            candidate = re.sub(
+                '[()]', '', name.split('.')[-1]
+            )
+            return candidate.isidentifier()
         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 self.params
+                for p in params
             )  # <====  Date subtract
-        return super().get_pattern()
+        return super().__str__()
+
+
+class DatePart(Function):
+    inputs = [DATE]
+    output = INT
 
-class Year(Function):
     def get_pattern(self) -> str:
+        interval = self.__class__.__name__
         database_type = {
-            Dialect.ORACLE: 'Extract(YEAR FROM {params})',
-            Dialect.POSTGRESQL: "Date_Part('year', {params})",
+            Dialect.ORACLE: 'Extract('+interval+' FROM {params})',
+            Dialect.POSTGRESQL: "Date_Part('"+interval+"', {params})",
         }
         if self.dialect in database_type:
             return database_type[self.dialect]
         return super().get_pattern()
 
+class Year(DatePart):
+    ...
+class Month(DatePart):
+    ...
+class Day(DatePart):
+    ...
+
+
 class Current_Date(Function):
+    output = DATE
+
     def get_pattern(self) -> str:
         database_type = {
             Dialect.ORACLE: SQL_CONST_SYSDATE,
@@ -341,9 +408,13 @@ class Lead(Window, Function):
 
 # ---- Conversions and other Functions: ---------------------
 class Coalesce(Function):
-    ...
+    inputs = [ANY]
+    output = ANY
+    
 class Cast(Function):
-    ...
+    inputs = [ANY]
+    output = ANY
+    separator = ' As '
 
 
 FUNCTION_CLASS = {f.__name__.lower(): f for f in Function.__subclasses__()}
@@ -516,14 +587,9 @@ class Where:
 
     def add(self, name: str, main: SQLObject):
         func_type = FUNCTION_CLASS.get(name.lower())
-        exists = any(
-            main.is_named_field(fld, SELECT)
-            for fld in main.values.get(SELECT, [])
-            if name in fld
-        )
         if func_type:
             name = func_type.format('*', main)
-        elif not exists:
+        elif not main.has_named_field(name):
             name = Field.format(name, main)
         main.values.setdefault(WHERE, []).append('{}{} {}'.format(
             self.prefix, name, self.content
@@ -604,6 +670,14 @@ class Between:
         Where.gte(self.start).add(name, main),
         Where.lte(self.end).add(name, main)
 
+class SameDay(Between):
+    def __init__(self, date: str):
+        super().__init__(
+            f'{date} 00:00:00',
+            f'{date} 23:59:59',
+        )
+
+
 
 class Clause:
     @classmethod
@@ -1438,11 +1512,7 @@ class Select(SQLObject):
         Recognizes if the field is from the current table
         '''
         if key in (ORDER_BY, GROUP_BY) and '.' not in field:
-            return any(
-                self.is_named_field(fld, SELECT)
-                for fld in self.values[SELECT]
-                if field in fld
-            )
+            return main.has_named_field(field)
         return re.findall(f'\b*{self.alias}[.]', field) != []
 
     @classmethod
@@ -1455,12 +1525,10 @@ class Select(SQLObject):
         for rule in rules:
             rule.apply(self)
 
-    def add_fields(self, fields: list, order_by: bool=False, group_by:bool=False):
-        class_types = [Field]
-        if order_by:
-            class_types += [OrderBy]
-        if group_by:
-            class_types += [GroupBy]
+    def add_fields(self, fields: list, class_types=None):
+        if not class_types:
+            class_types = []
+        class_types += [Field]
         FieldList(fields, class_types).add('', self)
 
     def translate_to(self, language: QueryLanguage) -> str:
@@ -1491,12 +1559,17 @@ class CTE(Select):
         self.break_lines = False
 
     def __str__(self) -> str:
+        size = 0
+        for key in USUAL_KEYS:
+            size += sum(len(v) for v in self.values.get(key, []) if '\n' not in v)
+        if size > 70:
+            self.break_lines = True
         # ---------------------------------------------------------
         def justify(query: Select) -> str:
             result, line = [], ''
             keywords = '|'.join(KEYWORD)
             for word in re.split(fr'({keywords}|AND|OR|,)', str(query)):
-                if len(line) >= 65:
+                if len(line) >= 60:
                     result.append(line)
                     line = ''
                 line += word
@@ -1510,6 +1583,7 @@ class CTE(Select):
                 justify(q) for q in self.query_list
             ), super().__str__()
         )
+
     def join(self, pattern: str, fields: list | str, format: str=''):
         if isinstance(fields, str):
             count = len( fields.split(',') )
@@ -1702,11 +1776,4 @@ def detect(text: str, join_queries: bool = True, format: str='') -> Select | lis
     for query in query_list[1:]:
         result += query
     return result
-
-
-if __name__ == "__main__":
-    CAMPO_MEDIA = 'MEDIA_SALARIAL_DEPTO'
-    employees = detect(
-        f'Employees@department_id(avg$salary:{CAMPO_MEDIA})'
-    )
-    print(employees)
+# ===========================================================================================//

{sql_blocks-1.25.33022258.dist-info → sql_blocks-1.25.38021329.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sql_blocks
-Version: 1.25.33022258
+Version: 1.25.38021329
 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
@@ -67,6 +67,16 @@ You can specify your own alias:  `a = Select('Actor a')`
 3.1 -- If you want to filter the field on a range of values:
 
 `a = Select( 'Actor a', age=Between(45, 69) )`
+...but if it is a time slot within the same day, you can do it like this:
+`Select(..., event_date=SameDay("2024-10-03"))`
+This results in
+```
+    SELECT ...
+    WHERE
+        event_date >= '2024-10-03 00:00:00' AND
+        event_date <= '2024-10-03 23:59:59'
+```
+---
 
 3.2 -- Sub-queries:
 ```
@@ -668,6 +678,37 @@ For example, if your query is going to run on Oracle, do the following:
 
 `Function.dialect = Dialect.ORACLE`
 
+
+>  Most of this functions you can use nested inside each other.
+*Example:*
+```
+    Select(...
+        event_date=Substring(
+            Cast("CHAR"), 12, 19 
+        ).As('time')
+    )
+```
+Results...
+```
+    SELECT ...
+    SubString(Cast(event_date As char), 12, 19) as time
+```
+
+>> `Function.auto_convert` option (default: True)
+
+- Put Cast(...) when there is a difference between the types of the parameter and the return of the nested function
+```
+birth=Round( DateDiff(Current_Date()) ).As('age')
+```
+...Returns...
+```
+SELECT
+    Round(
+        Cast(Current_Date() - p.birth As FLOAT)
+       /* ^^^  */
+    ) as age
+...
+```
 ---
 
 ### 17 - CTE and Recursive classes

sql_blocks-1.25.38021329.dist-info/RECORD

@@ -0,0 +1,7 @@
+sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
+sql_blocks/sql_blocks.py,sha256=5LQPEOkvMylJJxoU2m1P6pAhfcMHVdsnTiYgfLLJ5bQ,59477
+sql_blocks-1.25.38021329.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+sql_blocks-1.25.38021329.dist-info/METADATA,sha256=un6TFbAfNbyhIrjLzHpvG1Yy77jsyZCw2xy_i3tMw2k,20511
+sql_blocks-1.25.38021329.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+sql_blocks-1.25.38021329.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
+sql_blocks-1.25.38021329.dist-info/RECORD,,

sql_blocks-1.25.33022258.dist-info/RECORD

@@ -1,7 +0,0 @@
-sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
-sql_blocks/sql_blocks.py,sha256=dGmMtLWiyluRp1ZwPplmBqPs1nDOL-vemRR0gi2htWA,57513
-sql_blocks-1.25.33022258.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-sql_blocks-1.25.33022258.dist-info/METADATA,sha256=GsTRa0fLkerPn-D-ourudj9Pve0oP69IzQ-YGHMtRnI,19574
-sql_blocks-1.25.33022258.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sql_blocks-1.25.33022258.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
-sql_blocks-1.25.33022258.dist-info/RECORD,,