sql-blocks 1.25.37022003__py3-none-any.whl → 1.25.312022057__py3-none-any.whl

sql_blocks/sql_blocks.py

@@ -196,17 +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
-    need_params = True
+    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 = list(params)
+        self.params = [set_func_types(p) for p in params]
         self.field_class = Field
         self.pattern = self.get_pattern()
         self.extra = {}
@@ -226,17 +244,28 @@ class Function:
             params=self.separator.join(str(p) for p in self.params)
         )
 
+    @classmethod
+    def help(cls) -> str:
+        descr = ' '.join(B.__name__ for B in cls.__bases__)
+        params = cls.inputs or ''
+        return cls().get_pattern().format(
+            func_name=f'{descr} {cls.__name__}',
+            params=cls.separator.join(str(p) for p in params)
+        ) + f'  Return {cls.output}'
+
     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.need_params:
+            if func.inputs:
                 func.set_main_param(name, main)
                 return
-        self.params = [
-            Field.format(name, main)
-        ] + self.params
+        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 '*_':
@@ -260,6 +289,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})'
@@ -267,33 +299,54 @@ 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()
-        params = [str(p) for p in self.params]
+            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 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):
-    need_params = False
+    output = DATE
 
     def get_pattern(self) -> str:
         database_type = {
@@ -334,7 +387,8 @@ class Frame:
 
 
 class Aggregate(Frame):
-    ...
+    inputs = [FLOAT]
+    output = FLOAT
 
 class Window(Frame):
     ...
@@ -353,19 +407,26 @@ class Count(Aggregate, Function):
 
 # ---- Window Functions: -----------------------------------
 class Row_Number(Window, Function):
-    ... 
+    output = INT
+
 class Rank(Window, Function):
-    ... 
+    output = INT
+
 class Lag(Window, Function):
-    ...
+    output = ANY
+
 class Lead(Window, Function):
-    ...
+    output = ANY
 
 
 # ---- Conversions and other Functions: ---------------------
 class Coalesce(Function):
-    ...
+    inputs = [ANY]
+    output = ANY
+    
 class Cast(Function):
+    inputs = [ANY]
+    output = ANY
     separator = ' As '
 
 
@@ -1393,15 +1454,18 @@ class Select(SQLObject):
 
     def add(self, name: str, main: SQLObject):
         old_tables = main.values.get(FROM, [])
-        new_tables = set([
-            '{jt}JOIN {tb} {a2} ON ({a1}.{f1} = {a2}.{f2})'.format(
+        if len(self.values[FROM]) > 1:
+            old_tables += self.values[FROM][1:]
+        new_tables = []
+        row = '{jt}JOIN {tb} {a2} ON ({a1}.{f1} = {a2}.{f2})'.format(
                 jt=self.join_type.value,
                 tb=self.aka(),
                 a1=main.alias, f1=name,
                 a2=self.alias, f2=self.key_field
             )
-        ] + old_tables[1:])
-        main.values[FROM] = old_tables[:1] + list(new_tables)
+        if row not in old_tables[1:]:
+            new_tables.append(row)
+        main.values[FROM] = old_tables[:1] + new_tables + old_tables[1:]
         for key in USUAL_KEYS:
             main.update_values(key, self.values.get(key, []))
 
@@ -1521,7 +1585,7 @@ class CTE(Select):
             result, line = [], ''
             keywords = '|'.join(KEYWORD)
             for word in re.split(fr'({keywords}|AND|OR|,)', str(query)):
-                if len(line) >= 60:
+                if len(line) >= 50:
                     result.append(line)
                     line = ''
                 line += word
@@ -1728,4 +1792,31 @@ 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__":
+    print('='*50)
+    visitante = Select(
+        'Pessoa p1',
+        nome=NamedField('nome_visitante'),
+        tipo=eq('V'), cpf=PrimaryKey
+    )
+    dono_do_apto = Select(
+        'Apartamento A',
+        dono=Select(
+            'Pessoa p2',
+            cpf=PrimaryKey,
+            nome=NamedField('nome_morador')
+        ), 
+        id=PrimaryKey
+    )
+    movimento = Select(
+        'Movimento M',
+        entrada_saida=eq('E'),
+        apartamento=dono_do_apto,
+        pessoa=visitante
+    )
+    print(dono_do_apto)
+    print('-'*50)
+    print(movimento)
+    print('='*50)

{sql_blocks-1.25.37022003.dist-info → sql_blocks-1.25.312022057.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sql_blocks
-Version: 1.25.37022003
+Version: 1.25.312022057
 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
@@ -693,6 +693,22 @@ 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.312022057.dist-info/RECORD

@@ -0,0 +1,7 @@
+sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
+sql_blocks/sql_blocks.py,sha256=ICqvEtrB1xnCqquTyTgDX-O9hkeeysQLcLz9pdkFBYs,60671
+sql_blocks-1.25.312022057.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
+sql_blocks-1.25.312022057.dist-info/METADATA,sha256=VCXJmWhUPrpSd5dD_Jj5JeXnr2D5XLQNnXMYkWRsN0g,20512
+sql_blocks-1.25.312022057.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+sql_blocks-1.25.312022057.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
+sql_blocks-1.25.312022057.dist-info/RECORD,,

sql_blocks-1.25.37022003.dist-info/RECORD

@@ -1,7 +0,0 @@
-sql_blocks/__init__.py,sha256=5ItzGCyqqa6kwY8wvF9kapyHsAiWJ7KEXCcC-OtdXKg,37
-sql_blocks/sql_blocks.py,sha256=kaI6nCcq1X3CenGFP4HkKkk3rsDt_jts8xErg-ml7dk,58217
-sql_blocks-1.25.37022003.dist-info/LICENSE,sha256=6kbiFSfobTZ7beWiKnHpN902HgBx-Jzgcme0SvKqhKY,1091
-sql_blocks-1.25.37022003.dist-info/METADATA,sha256=AUa01HeqYtgwVObYWMTkqixy0RNoxwNQXurTMEWjjQ8,20146
-sql_blocks-1.25.37022003.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sql_blocks-1.25.37022003.dist-info/top_level.txt,sha256=57AbUvUjYNy4m1EqDaU3WHeP-uyIAfV0n8GAUp1a1YQ,11
-sql_blocks-1.25.37022003.dist-info/RECORD,,