ormlambda 0.1.0__py3-none-any.whl

ormlambda/utils/lambda_disassembler/disassembler.py

@@ -0,0 +1,69 @@
+import dis
+from typing import Any, Callable, overload
+
+from .nested_element import NestedElement
+from .tree_instruction import TreeInstruction
+
+
+class Disassembler[TProp1, TProp2: Any]:
+    """
+    class to dissambler lambda function to detach information from left to right of compare symbol.
+
+    >>> dis = Disassembler[DtoC, None](lambda d: d.c.b.b_data == "asdf")
+
+    >>> dis.cond_1.name # b_data
+    >>> dis.cond_1.parent.parent.parent.name # d
+    >>> dis.cond_1.parent.parent.name # c
+
+    >>> dis.cond_2.name, "asdf"
+
+    >>> dis.compare_op, "="
+
+    """
+
+    __slots__ = (
+        "_function",
+        "_bytecode_function",
+        "_cond_1",
+        "_cond_2",
+        "_compare_op",
+    )
+
+    @overload
+    def __init__(self, function: Callable[[], bool]) -> None: ...
+
+    @overload
+    def __init__(self, function: Callable[[TProp1], bool]) -> None: ...
+
+    @overload
+    def __init__(self, function: Callable[[TProp1, TProp2], bool]) -> None: ...
+
+    def __init__(self, function: Callable[[], bool] | Callable[[TProp1], bool] | Callable[[TProp1, TProp2], bool]) -> None:
+        self._function: Callable[[], bool] | Callable[[TProp1], bool] | Callable[[TProp1, TProp2], bool] = function
+        self._bytecode_function: dis.Bytecode = dis.Bytecode(function)
+        self.__init_custom__()
+
+    def __repr__(self) -> str:
+        return f"<{self.__class__.__name__}>: {self._cond_1.name} {self._cond_2.name} {self._compare_op}"
+
+    def __init_custom__(self):
+        tree = TreeInstruction(self._function)
+        dicc = tree.to_list()
+        self._compare_op: str = tree.compare_op[0]
+
+        self._cond_1: NestedElement[TProp1] = dicc[0].nested_element
+        self._cond_2: NestedElement[TProp2] = dicc[1].nested_element
+
+        return None
+
+    @property
+    def cond_1(self) -> NestedElement[str]:
+        return self._cond_1
+
+    @property
+    def cond_2(self) -> NestedElement[str]:
+        return self._cond_2
+
+    @property
+    def compare_op(self) -> str:
+        return self._compare_op

ormlambda/utils/lambda_disassembler/dtypes.py

@@ -0,0 +1,103 @@
+"""
+String Data Types
+Data type	Description
+CHAR(size)	A FIXED length string (can contain letters, numbers, and special characters). The size parameter specifies the column length in characters - can be from 0 to 255. Default is 1
+VARCHAR(size)	A VARIABLE length string (can contain letters, numbers, and special characters). The size parameter specifies the maximum column length in characters - can be from 0 to 65535
+BINARY(size)	Equal to CHAR(), but stores binary byte strings. The size parameter specifies the column length in bytes. Default is 1
+VARBINARY(size)	Equal to VARCHAR(), but stores binary byte strings. The size parameter specifies the maximum column length in bytes.
+TINYBLOB	For BLOBs (Binary Large OBjects). Max length: 255 bytes
+TINYTEXT	Holds a string with a maximum length of 255 characters
+TEXT(size)	Holds a string with a maximum length of 65,535 bytes
+BLOB(size)	For BLOBs (Binary Large OBjects). Holds up to 65,535 bytes of data
+MEDIUMTEXT	Holds a string with a maximum length of 16,777,215 characters
+MEDIUMBLOB	For BLOBs (Binary Large OBjects). Holds up to 16,777,215 bytes of data
+LONGTEXT	Holds a string with a maximum length of 4,294,967,295 characters
+LONGBLOB	For BLOBs (Binary Large OBjects). Holds up to 4,294,967,295 bytes of data
+ENUM(val1, val2, val3, ...)	A string object that can have only one value, chosen from a list of possible values. You can list up to 65535 values in an ENUM list. If a value is inserted that is not in the list, a blank value will be inserted. The values are sorted in the order you enter them
+SET(val1, val2, val3, ...)	A string object that can have 0 or more values, chosen from a list of possible values. You can list up to 64 values in a SET list
+Numeric Data Types
+Data type	Description
+BIT(size)	A bit-value type. The number of bits per value is specified in size. The size parameter can hold a value from 1 to 64. The default value for size is 1.
+TINYINT(size)	A very small integer. Signed range is from -128 to 127. Unsigned range is from 0 to 255. The size parameter specifies the maximum display width (which is 255)
+BOOL	Zero is considered as false, nonzero values are considered as true.
+BOOLEAN	Equal to BOOL
+SMALLINT(size)	A small integer. Signed range is from -32768 to 32767. Unsigned range is from 0 to 65535. The size parameter specifies the maximum display width (which is 255)
+MEDIUMINT(size)	A medium integer. Signed range is from -8388608 to 8388607. Unsigned range is from 0 to 16777215. The size parameter specifies the maximum display width (which is 255)
+INT(size)	A medium integer. Signed range is from -2147483648 to 2147483647. Unsigned range is from 0 to 4294967295. The size parameter specifies the maximum display width (which is 255)
+INTEGER(size)	Equal to INT(size)
+BIGINT(size)	A large integer. Signed range is from -9223372036854775808 to 9223372036854775807. Unsigned range is from 0 to 18446744073709551615. The size parameter specifies the maximum display width (which is 255)
+FLOAT(size, d)	A floating point number. The total number of digits is specified in size. The number of digits after the decimal point is specified in the d parameter. This syntax is deprecated in MySQL 8.0.17, and it will be removed in future MySQL versions
+FLOAT(p)	A floating point number. MySQL uses the p value to determine whether to use FLOAT or DOUBLE for the resulting data type. If p is from 0 to 24, the data type becomes FLOAT(). If p is from 25 to 53, the data type becomes DOUBLE()
+DOUBLE(size, d)	A normal-size floating point number. The total number of digits is specified in size. The number of digits after the decimal point is specified in the d parameter
+DOUBLE PRECISION(size, d)
+DECIMAL(size, d)	An exact fixed-point number. The total number of digits is specified in size. The number of digits after the decimal point is specified in the d parameter. The maximum number for size is 65. The maximum number for d is 30. The default value for size is 10. The default value for d is 0.
+DEC(size, d)	Equal to DECIMAL(size,d)
+Note: All the numeric data types may have an extra option: UNSIGNED or ZEROFILL. If you add the UNSIGNED option, MySQL disallows negative values for the column. If you add the ZEROFILL option, MySQL automatically also adds the UNSIGNED attribute to the column.
+
+Date and Time Data Types
+Data type	Description
+DATE	A date. Format: YYYY-MM-DD. The supported range is from '1000-01-01' to '9999-12-31'
+DATETIME(fsp)	A date and time combination. Format: YYYY-MM-DD hh:mm:ss. The supported range is from '1000-01-01 00:00:00' to '9999-12-31 23:59:59'. Adding DEFAULT and ON UPDATE in the column definition to get automatic initialization and updating to the current date and time
+TIMESTAMP(fsp)	A timestamp. TIMESTAMP values are stored as the number of seconds since the Unix epoch ('1970-01-01 00:00:00' UTC). Format: YYYY-MM-DD hh:mm:ss. The supported range is from '1970-01-01 00:00:01' UTC to '2038-01-09 03:14:07' UTC. Automatic initialization and updating to the current date and time can be specified using DEFAULT CURRENT_TIMESTAMP and ON UPDATE CURRENT_TIMESTAMP in the column definition
+TIME(fsp)	A time. Format: hh:mm:ss. The supported range is from '-838:59:59' to '838:59:59'
+YEAR	A year in four-digit format. Values allowed in four-digit format: 1901 to 2155, and 0000.
+MySQL 8.0 does not support year in two-digit format.
+"""
+
+from typing import Literal
+
+
+STRING = Literal[
+    "CHAR(size)",
+    "VARCHAR(size)",
+    "BINARY(size)",
+    "VARBINARY(size)",
+    "TINYBLOB",
+    "TINYTEXT",
+    "TEXT(size)",
+    "BLOB(size)",
+    "MEDIUMTEXT",
+    "MEDIUMBLOB",
+    "LONGTEXT",
+    "LONGBLOB",
+    "ENUM(val1, val2, val3, ...)",
+    "SET(val1, val2, val3, ...)",
+]
+
+NUMERIC_SIGNED = Literal[
+    "BIT(size)",
+    "TINYINT(size)",
+    "BOOL",
+    "BOOLEAN",
+    "SMALLINT(size)",
+    "MEDIUMINT(size)",
+    "INT(size)",
+    "INTEGER(size)",
+    "BIGINT(size)",
+    "FLOAT(size, d)",
+    "FLOAT(p)",
+    "DOUBLE(size, d)",
+    "DOUBLE PRECISION(size, d)",
+    "DECIMAL(size, d)",
+    "DEC(size, d)",
+]
+
+NUMERIC_UNSIGNED = Literal[
+    "BIT(size)",
+    "TINYINT(size)",
+    "BOOL",
+    "BOOLEAN",
+    "SMALLINT(size)",
+    "MEDIUMINT(size)",
+    "INT(size)",
+    "INTEGER(size)",
+    "BIGINT(size)",
+    "FLOAT(size, d)",
+    "FLOAT(p)",
+    "DOUBLE(size, d)",
+    "DOUBLE PRECISION(size, d)",
+    "DECIMAL(size, d)",
+    "DEC(size, d)",
+]
+
+DATE = Literal["DATE", "DATETIME(fsp)", "TIMESTAMP(fsp)", "TIME(fsp)", "YEAR"]

ormlambda/utils/lambda_disassembler/name_of.py

@@ -0,0 +1,41 @@
+import inspect
+import re
+
+
+def nameof(_, full_name=False) -> str:
+    """
+    Return the name of the variable passed as argument
+
+    >>> class C:
+    ...     member = 5
+    >>> my_var = C()
+    >>> nameof(my_var)
+    'my_var'
+    >>> nameof(my_var.member)
+    'member'
+    >>> nameof(my_var.member, full_name=True)
+    'my_var.member'
+
+    >>> nameof(4 + (my_var.member) + 11)
+    Traceback (most recent call last):
+      ...
+    RuntimeError: Invalid nameof invocation: nameof(4 + (my_var.member) + 11)
+    >>> nameof(4)
+    Traceback (most recent call last):
+      ...
+    RuntimeError: Invalid nameof invocation: nameof(4)
+    """
+    s = inspect.stack()
+    func_name = s[0].frame.f_code.co_name
+    call = s[1].code_context[0].strip()
+    # Simple search for first closing bracket or comma
+    arg = re.search(rf"{func_name}\(([^,)]+)[,)]", call)
+    if arg is not None:
+        var_parts = arg.group(1).split(".")
+    if arg is None or not all(map(str.isidentifier, var_parts)):
+        raise RuntimeError(f"Invalid {func_name} invocation: {call}")
+
+    if full_name:
+        return arg.group(1)
+    else:
+        return var_parts[-1]

ormlambda/utils/lambda_disassembler/nested_element.py

@@ -0,0 +1,44 @@
+from typing import Iterable
+
+
+class NestedElement[T]:
+    def __init__(self, cond: T):
+        self._cond: T = cond
+        self._parent_list: list[str] = self._create_parent_list(cond)
+
+    def __repr__(self) -> str:
+        return f"{NestedElement.__name__}: {'.'.join(self._parent_list)}"
+
+    @property
+    def name(self) -> T:
+        if self._parent_list:
+            return self._parent_list[-1]
+        else:
+            return self._cond
+
+    @property
+    def parent(self) -> "NestedElement":
+        if not self._parent_list:
+            raise ValueError(f"Attribute '{self._cond}' has not parent values")
+        new_cond = self._parent_list[:-1]
+        return self.__get_parent(new_cond)
+
+    @property
+    def parents(self) -> list[T]:
+        return self._parent_list
+
+    def _create_parent_list(self, condition: T) -> Iterable[str]:
+        if isinstance(condition, Iterable) and not isinstance(condition, str):
+            return condition
+
+        try:
+            _list: list[str] = condition.split(".")
+
+        except Exception:
+            return []
+        else:
+            return _list
+
+    @staticmethod
+    def __get_parent(constructor: T) -> "NestedElement":
+        return NestedElement[T](constructor)

ormlambda/utils/lambda_disassembler/tree_instruction.py

@@ -0,0 +1,145 @@
+from collections import defaultdict
+from typing import Any, Callable, NamedTuple, Self, Optional
+from dis import Instruction, Bytecode
+from ...common.enums.condition_types import ConditionType
+from .dis_types import OpName
+from .nested_element import NestedElement
+
+
+class Node[T]:
+    def __init__(self, data: T):
+        self.data: T = data
+        self.children: list[Self[T]] = []
+
+    def __repr__(self) -> str:
+        return f"{Node.__name__}: data={self.data} children={self.children}"
+
+
+class TupleInstruction(NamedTuple):
+    var: str
+    nested_element: NestedElement[str]
+
+
+class TreeInstruction:
+    _FIRST_LEVEL: tuple[OpName] = (
+        OpName.LOAD_FAST,
+        OpName.LOAD_GLOBAL,
+        OpName.LOAD_CONST,
+        OpName.LOAD_DEREF,
+    )
+
+    def __init__[T, *Ts](self, lambda_: Callable[[T, *Ts], None]):
+        self._root: Node[Instruction] = Node[Instruction](None)
+
+        self._bytecode: Bytecode = Bytecode(lambda_)
+        self._compare_op: Optional[list[str]] = []
+        self._set_root()
+
+    def __repr__(self) -> str:
+        return f"{TreeInstruction.__name__} < at {hex(id(self))}>"
+
+    @staticmethod
+    def _transform__compare_op(compare_sybmol: ConditionType) -> str:
+        dicc_symbols: dict[ConditionType, str] = {
+            ConditionType.EQUAL: "=",
+        }
+        return dicc_symbols.get(compare_sybmol, compare_sybmol.value)
+
+    @staticmethod
+    def _transform_is_is_not(instr: Instruction) -> str:
+        if instr.argval == 1:
+            return ConditionType.IS_NOT.value
+        elif instr.argval == 0:
+            return ConditionType.IS.value
+        else:
+            raise Exception(f"argval value '{instr.argval}' not expected.")
+
+    def _set_root(self) -> None:
+        """
+        add instructions into self._root
+        """
+
+        def is_instruction_comparable(instr: Instruction) -> bool:
+            opname = OpName(instr.opname)
+            if opname in (OpName.COMPARE_OP, OpName.IS_OP):
+                return True
+            return False
+
+        def set_comparable(instr: Instruction) -> None:
+            opname = OpName(instr.opname)
+            if opname == OpName.COMPARE_OP:
+                return self._compare_op.append(self._transform__compare_op(ConditionType(x.argval)))
+            elif opname == OpName.IS_OP:
+                return self._compare_op.append(self._transform_is_is_not(x))
+            return None
+
+        for x in self._bytecode:
+            if is_instruction_comparable(x):
+                set_comparable(x)
+            else:
+                self.add_instruction(x)
+
+    def to_dict(self) -> dict[str, list[NestedElement[str]]]:
+        _dict: dict[str, list[NestedElement[str]]] = defaultdict(list)
+        for node in self.root.children:
+            argval = self._join_data_attributes_from_node(node)
+            _dict[node.data.argval].append(NestedElement[str](argval))
+        return _dict
+
+    def add_instruction(self, ins: Instruction) -> None:
+        opname = OpName(ins.opname)
+        if opname in TreeInstruction._FIRST_LEVEL:
+            self._root.children.append(Node[Instruction](ins))
+            return None
+
+        if OpName(ins.opname) == OpName.LOAD_ATTR:
+            last_fast_node = self._root.children[-1]
+            new_node = Node[Instruction](ins)
+            self._add_node(last_fast_node, new_node)
+        return None
+
+    def _add_node(self, parent_node: Node[Instruction], new_node: Node[Instruction]):
+        parent_node.children.append(new_node)
+        return None
+
+    def get_all_nodes_of(self, node: Node[Instruction]) -> list[Node[Optional[Instruction]]]:
+        def get_all_nodes(node: Node[Instruction], list: list[Node[Instruction]]):
+            if not node.children:
+                return list
+            for subnode in node.children:
+                if not subnode.children:
+                    data = subnode
+                else:
+                    data = self.get_all_nodes_of(subnode)
+                list.append(data)
+
+        lista = []
+        get_all_nodes(node, lista)
+        return lista
+
+    def to_list(self) -> list[TupleInstruction]:
+        """
+        Return list with tuple as data.
+        This tuple contains the variable name as the first element and all children as the second one, all inside a TupleInstruction NamedTuple.
+        """
+        _list: list[list[NestedElement[str]]] = []
+        for node in self.root.children:
+            argval = self._join_data_attributes_from_node(node)
+            _list.append(TupleInstruction(node.data.argval, NestedElement[str](argval)))
+        return _list
+
+    def _join_data_attributes_from_node(self, _node: Node[Instruction]) -> list[Any]:
+        lista = self.get_all_nodes_of(_node)
+        if not lista:
+            return _node.data.argval
+        else:
+            # We just want to retrieve argval not Instruction object itself
+            return [_node.data.argval] + [x.data.argval for x in lista]
+
+    @property
+    def root(self) -> Node[Instruction]:
+        return self._root
+
+    @property
+    def compare_op(self) -> Optional[list[str]]:
+        return self._compare_op

ormlambda/utils/module_tree/dfs_traversal.py

@@ -0,0 +1,60 @@
+from typing import Iterable
+from collections import deque, defaultdict
+
+
+class DFSTraversal:
+    """
+    Depth first search with Kahn's algorithm
+    """
+
+    @staticmethod
+    def sort[T](graph: dict[T, Iterable[T]], raise_: bool = False) -> tuple[T, ...]:
+        indegree: dict[T, int] = defaultdict(int)
+        for key, val in graph.items():
+            for node in val:
+                indegree[node] += 1
+
+        q: deque[T] = deque()
+        for key in graph:
+            if indegree[key] == 0:
+                q.append(key)
+
+        topo: list[T] = []
+        while q:
+            node: T = q.popleft()
+            topo.append(node)
+
+            for neighbor in graph[node]:
+                indegree[neighbor] -= 1
+                if indegree[neighbor] == 0:
+                    q.append(neighbor)
+
+        if len(graph) != len(topo):
+            difference = set(graph).difference(set(topo))
+            msg = f"Cycle found near of nodes '{difference}'"
+            if raise_:
+                raise Exception(msg)
+            print(msg)
+        return tuple(topo[::-1])
+
+
+if __name__ == "__main__":
+    adj = {
+        0: [2, 3, 6],
+        1: [4],
+        2: [6],
+        3: [1, 4],
+        4: [5, 8],
+        5: [],
+        6: [7, 11],
+        7: [4, 12],
+        8: [0],
+        9: [10],
+        10: [6],
+        11: [12],
+        12: [8],
+        13: [],
+    }
+
+    ans = DFSTraversal.sort(adj)
+    print(ans)