ripple-down-rules 0.1.69__tar.gz → 0.2.0__tar.gz

{ripple_down_rules-0.1.69 → ripple_down_rules-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ripple_down_rules
-Version: 0.1.69
+Version: 0.2.0
 Summary: Implements the various versions of Ripple Down Rules (RDR) for knowledge representation and reasoning.
 Author-email: Abdelrhman Bassiouny <abassiou@uni-bremen.de>
 License:                     GNU GENERAL PUBLIC LICENSE

{ripple_down_rules-0.1.69 → ripple_down_rules-0.2.0}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ripple_down_rules"
-version = "0.1.69"
+version = "0.2.0"
 description = "Implements the various versions of Ripple Down Rules (RDR) for knowledge representation and reasoning."
 readme = "README.md"
 authors = [{ name = "Abdelrhman Bassiouny", email = "abassiou@uni-bremen.de" }]

{ripple_down_rules-0.1.69 → ripple_down_rules-0.2.0}/src/ripple_down_rules/datastructures/callable_expression.py

@@ -9,7 +9,7 @@ from typing_extensions import Type, Optional, Any, List, Union, Tuple, Dict, Set
 
 from .case import create_case, Case
 from ..utils import SubclassJSONSerializer, get_full_class_name, get_type_from_string, conclusion_to_json, is_iterable, \
-    build_user_input_from_conclusion, encapsulate_user_input
+    build_user_input_from_conclusion, encapsulate_user_input, extract_function_source
 
 
 class VariableVisitor(ast.NodeVisitor):
@@ -109,7 +109,7 @@ class CallableExpression(SubclassJSONSerializer):
         if user_input is None:
             user_input = build_user_input_from_conclusion(conclusion)
         self.conclusion: Optional[Any] = conclusion
-        self.user_input: str = encapsulate_user_input(user_input, self.encapsulating_function)
+        self._user_input: str = encapsulate_user_input(user_input, self.encapsulating_function)
         if conclusion_type is not None:
             if is_iterable(conclusion_type):
                 conclusion_type = tuple(conclusion_type)
@@ -156,6 +156,35 @@ class CallableExpression(SubclassJSONSerializer):
                           f"return _cond1(case) and _cond2(case)")
         return CallableExpression(new_user_input, conclusion_type=self.conclusion_type)
 
+    def update_user_input_from_file(self, file_path: str, function_name: str):
+        """
+        Update the user input from a file.
+        """
+        new_function_body = extract_function_source(file_path, [function_name])[function_name]
+        if new_function_body is None:
+            return
+        self.user_input = self.encapsulating_function + '\n' + new_function_body
+
+    @property
+    def user_input(self):
+        """
+        Get the user input.
+        """
+        return self._user_input
+
+    @user_input.setter
+    def user_input(self, value: str):
+        """
+        Set the user input.
+        """
+        if value is not None:
+            self._user_input = encapsulate_user_input(value, self.encapsulating_function)
+            self.scope = get_used_scope(self.user_input, self.scope)
+            self.expression_tree = parse_string_to_expression(self.user_input)
+            self.code = compile_expression_to_code(self.expression_tree)
+            self.visitor = VariableVisitor()
+            self.visitor.visit(self.expression_tree)
+
     def __eq__(self, other):
         """
         Check if two callable expressions are equal.
@@ -225,7 +254,7 @@ def parse_string_to_expression(expression_str: str) -> AST:
     :param expression_str: The string which will be parsed.
     :return: The parsed expression.
     """
-    if not expression_str.startswith('def'):
+    if not expression_str.startswith(CallableExpression.encapsulating_function):
         expression_str = encapsulate_user_input(expression_str, CallableExpression.encapsulating_function)
     mode = 'exec' if expression_str.startswith('def') else 'eval'
     tree = ast.parse(expression_str, mode=mode)

{ripple_down_rules-0.1.69 → ripple_down_rules-0.2.0}/src/ripple_down_rules/datastructures/enums.py

@@ -59,7 +59,7 @@ class PromptFor(Enum):
     """
     Prompt for rule conditions about a case.
     """
-    Conclusion: str = "conclusion"
+    Conclusion: str = "value"
     """
     Prompt for rule conclusion about a case.
     """

ripple_down_rules-0.2.0/src/ripple_down_rules/prompt.py

@@ -0,0 +1,354 @@
+import ast
+import logging
+import os
+import subprocess
+import tempfile
+from _ast import AST
+from functools import cached_property
+from textwrap import indent, dedent
+
+from IPython.core.magic import register_line_magic, line_magic, Magics, magics_class
+from IPython.terminal.embed import InteractiveShellEmbed
+from traitlets.config import Config
+from typing_extensions import List, Optional, Tuple, Dict, Type, Union
+
+from .datastructures.enums import PromptFor
+from .datastructures.case import Case
+from .datastructures.callable_expression import CallableExpression, parse_string_to_expression
+from .datastructures.dataclasses import CaseQuery
+from .utils import extract_dependencies, contains_return_statement, make_set, get_imports_from_scope, make_list, \
+    get_import_from_type, get_imports_from_types, is_iterable, extract_function_source, encapsulate_user_input
+
+
+@magics_class
+class MyMagics(Magics):
+    def __init__(self, shell, scope, output_type: Optional[Type] = None, func_name: str = "user_case",
+                 func_doc: str = "User defined function to be executed on the case.",
+                 code_to_modify: Optional[str] = None):
+        super().__init__(shell)
+        self.scope = scope
+        self.temp_file_path = None
+        self.func_name = func_name
+        self.func_doc = func_doc
+        self.code_to_modify = code_to_modify
+        self.output_type = make_list(output_type) if output_type is not None else None
+        self.user_edit_line = 0
+        self.function_signature: Optional[str] = None
+        self.build_function_signature()
+
+    @line_magic
+    def edit_case(self, line):
+
+        boilerplate_code = self.build_boilerplate_code()
+
+        self.write_to_file(boilerplate_code)
+
+        print(f"Opening {self.temp_file_path} in PyCharm...")
+        subprocess.Popen(["pycharm", "--line", str(self.user_edit_line), self.temp_file_path])
+
+    def build_boilerplate_code(self):
+        imports = self.get_imports()
+        self.build_function_signature()
+        if self.code_to_modify is not None:
+            body = indent(dedent(self.code_to_modify), '    ')
+        else:
+            body = "    # Write your code here\n    pass"
+        boilerplate = f"""{imports}\n\n{self.function_signature}\n    \"\"\"{self.func_doc}\"\"\"\n{body}"""
+        self.user_edit_line = imports.count('\n')+6
+        return boilerplate
+
+    def build_function_signature(self):
+        if self.output_type is None:
+            output_type_hint = ""
+        elif len(self.output_type) == 1:
+            output_type_hint = f" -> {self.output_type[0].__name__}"
+        else:
+            output_type_hint = f" -> Union[{', '.join([t.__name__ for t in self.output_type])}]"
+        self.function_signature = f"def {self.func_name}(case: {self.case_type.__name__}){output_type_hint}:"
+
+    def write_to_file(self, code: str):
+        tmp = tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix=".py",
+                                          dir=os.path.dirname(self.scope['__file__']))
+        tmp.write(code)
+        tmp.flush()
+        self.temp_file_path = tmp.name
+        tmp.close()
+
+    def get_imports(self):
+        case_type_import = f"from {self.case_type.__module__} import {self.case_type.__name__}"
+        if self.output_type is None:
+            output_type_imports = [f"from typing_extensions import Any"]
+        else:
+            output_type_imports = get_imports_from_types(self.output_type)
+            if len(self.output_type) > 1:
+                output_type_imports.append("from typing_extensions import Union")
+        print(output_type_imports)
+        imports = get_imports_from_scope(self.scope)
+        imports = [i for i in imports if ("get_ipython" not in i)]
+        if case_type_import not in imports:
+            imports.append(case_type_import)
+        imports.extend([oti for oti in output_type_imports if oti not in imports])
+        imports = set(imports)
+        return '\n'.join(imports)
+
+    @cached_property
+    def case_type(self) -> Type:
+        """
+        Get the type of the case object in the current scope.
+
+        :return: The type of the case object.
+        """
+        case = self.scope['case']
+        return case._obj_type if isinstance(case, Case) else type(case)
+
+    @line_magic
+    def load_case(self, line):
+        if not self.temp_file_path:
+            print("No file to load. Run %edit_case first.")
+            return
+
+        with open(self.temp_file_path, 'r') as f:
+            source = f.read()
+
+        tree = ast.parse(source)
+        for node in tree.body:
+            if isinstance(node, ast.FunctionDef) and node.name == self.func_name:
+                exec_globals = {}
+                exec(source, self.scope, exec_globals)
+                user_function = exec_globals[self.func_name]
+                self.shell.user_ns[self.func_name] = user_function
+                print(f"Loaded `{self.func_name}` function into user namespace.")
+                return
+
+        print(f"Function `{self.func_name}` not found.")
+
+
+class CustomInteractiveShell(InteractiveShellEmbed):
+    def __init__(self, output_type: Union[Type, Tuple[Type], None] = None, func_name: Optional[str] = None,
+                 func_doc: Optional[str] = None, code_to_modify: Optional[str] = None, **kwargs):
+        super().__init__(**kwargs)
+        keys = ['output_type', 'func_name', 'func_doc', 'code_to_modify']
+        values = [output_type, func_name, func_doc, code_to_modify]
+        magics_kwargs = {key: value for key, value in zip(keys, values) if value is not None}
+        self.my_magics = MyMagics(self, self.user_ns, **magics_kwargs)
+        self.register_magics(self.my_magics)
+        self.all_lines = []
+
+    def run_cell(self, raw_cell: str, **kwargs):
+        """
+        Override the run_cell method to capture return statements.
+        """
+        if contains_return_statement(raw_cell) and 'def ' not in raw_cell:
+            if self.my_magics.func_name in raw_cell:
+                self.all_lines = extract_function_source(self.my_magics.temp_file_path,
+                                                         self.my_magics.func_name,
+                                                         join_lines=False)[self.my_magics.func_name]
+            self.all_lines.append(raw_cell)
+            print("Exiting shell on `return` statement.")
+            self.history_manager.store_inputs(line_num=self.execution_count, source=raw_cell)
+            self.ask_exit()
+            return None
+        result = super().run_cell(raw_cell, **kwargs)
+        if result.error_in_exec is None and result.error_before_exec is None:
+            self.all_lines.append(raw_cell)
+        return result
+
+
+class IPythonShell:
+    """
+    Create an embedded Ipython shell that can be used to prompt the user for input.
+    """
+
+    def __init__(self, scope: Optional[Dict] = None, header: Optional[str] = None,
+                 output_type: Optional[Type] = None, prompt_for: Optional[PromptFor] = None,
+                 attribute_name: Optional[str] = None, attribute_type: Optional[Type] = None,
+                 code_to_modify: Optional[str] = None):
+        """
+        Initialize the Ipython shell with the given scope and header.
+
+        :param scope: The scope to use for the shell.
+        :param header: The header to display when the shell is started.
+        :param output_type: The type of the output from user input.
+        :param prompt_for: The type of information to ask the user about.
+        :param attribute_name: The name of the attribute of the case.
+        :param attribute_type: The type of the attribute of the case.
+        :param code_to_modify: The code to modify. If given, will be used as a start for user to modify.
+        """
+        self.scope: Dict = scope or {}
+        self.header: str = header or ">>> Embedded Ipython Shell"
+        self.output_type: Optional[Type] = output_type
+        self.prompt_for: Optional[PromptFor] = prompt_for
+        self.attribute_name: Optional[str] = attribute_name
+        self.attribute_type: Optional[Type] = attribute_type
+        self.code_to_modify: Optional[str] = code_to_modify
+        self.user_input: Optional[str] = None
+        self.func_name: str = ""
+        self.func_doc: str = ""
+        self.shell: CustomInteractiveShell = self._init_shell()
+        self.all_code_lines: List[str] = []
+
+    def _init_shell(self):
+        """
+        Initialize the Ipython shell with a custom configuration.
+        """
+        cfg = Config()
+        self.build_func_name_and_doc()
+        shell = CustomInteractiveShell(config=cfg, user_ns=self.scope, banner1=self.header,
+                                       output_type=self.output_type, func_name=self.func_name, func_doc=self.func_doc,
+                                       code_to_modify=self.code_to_modify)
+        return shell
+
+    def build_func_name_and_doc(self) -> Tuple[str, str]:
+        """
+        Build the function name and docstring for the user-defined function.
+
+        :return: A tuple containing the function name and docstring.
+        """
+        case = self.scope['case']
+        case_type = case._obj_type if isinstance(case, Case) else type(case)
+        self.func_name = self.build_func_name(case_type)
+        self.func_doc = self.build_func_doc(case_type)
+
+    def build_func_doc(self, case_type: Type):
+        if self.prompt_for == PromptFor.Conditions:
+            func_doc = (f"Get conditions on whether it's possible to conclude a value"
+                        f" for {case_type.__name__}.{self.attribute_name}")
+        else:
+            func_doc = f"Get possible value(s) for {case_type.__name__}.{self.attribute_name}"
+        if is_iterable(self.attribute_type):
+            possible_types = [t.__name__ for t in self.attribute_type if t not in [list, set]]
+            func_doc += f" of types list/set of {' and/or '.join(possible_types)}"
+        else:
+            func_doc += f" of type {self.attribute_type.__name__}"
+        return func_doc
+
+    def build_func_name(self, case_type: Type):
+        func_name = f"get_{self.prompt_for.value.lower()}_for"
+        func_name += f"_{case_type.__name__}"
+        if self.attribute_name is not None:
+            func_name += f"_{self.attribute_name}"
+        if is_iterable(self.attribute_type):
+            output_names = [f"{t.__name__}" for t in self.attribute_type if t not in [list, set]]
+        else:
+            output_names = [self.attribute_type.__name__] if self.attribute_type is not None else None
+        if output_names is not None:
+            func_name += '_of_type_' + '_'.join(output_names)
+        return func_name.lower()
+
+    def run(self):
+        """
+        Run the embedded shell.
+        """
+        while True:
+            try:
+                self.shell()
+                self.update_user_input_from_code_lines()
+                break
+            except Exception as e:
+                logging.error(e)
+                print(e)
+
+    def update_user_input_from_code_lines(self):
+        """
+        Update the user input from the code lines captured in the shell.
+        """
+        if len(self.shell.all_lines) == 1 and self.shell.all_lines[0].replace('return', '').strip() == '':
+            self.user_input = None
+        else:
+            self.all_code_lines = extract_dependencies(self.shell.all_lines)
+            if len(self.all_code_lines) == 1 and self.all_code_lines[0].strip() == '':
+                self.user_input = None
+            else:
+                import pdb; pdb.set_trace()
+                self.user_input = '\n'.join(self.all_code_lines)
+                self.user_input = encapsulate_user_input(self.user_input, self.shell.my_magics.function_signature,
+                                                         self.func_doc)
+                if f"return {self.func_name}(case)" not in self.user_input:
+                    self.user_input = self.user_input.strip() + f"\nreturn {self.func_name}(case)"
+
+
+def prompt_user_for_expression(case_query: CaseQuery, prompt_for: PromptFor, prompt_str: Optional[str] = None)\
+        -> Tuple[Optional[str], Optional[CallableExpression]]:
+    """
+    Prompt the user for an executable python expression to the given case query.
+
+    :param case_query: The case query to prompt the user for.
+    :param prompt_for: The type of information ask user about.
+    :param prompt_str: The prompt string to display to the user.
+    :return: A callable expression that takes a case and executes user expression on it.
+    """
+    prev_user_input: Optional[str] = None
+    while True:
+        user_input, expression_tree = prompt_user_about_case(case_query, prompt_for, prompt_str,
+                                                             code_to_modify=prev_user_input)
+        prev_user_input = '\n'.join(user_input.split('\n')[2:-1])
+        if user_input is None:
+            if prompt_for == PromptFor.Conclusion:
+                print("No conclusion provided. Exiting.")
+                return None, None
+            else:
+                print("Conditions must be provided. Please try again.")
+                continue
+        conclusion_type = bool if prompt_for == PromptFor.Conditions else case_query.attribute_type
+        callable_expression = CallableExpression(user_input, conclusion_type, expression_tree=expression_tree,
+                                                 scope=case_query.scope)
+        try:
+            result = callable_expression(case_query.case)
+            result = make_list(result)
+            if len(result) == 0:
+                print(f"The given expression gave an empty result for case {case_query.name}, please modify")
+                continue
+            break
+        except Exception as e:
+            logging.error(e)
+            print(e)
+    return user_input, callable_expression
+
+
+def prompt_user_about_case(case_query: CaseQuery, prompt_for: PromptFor,
+                           prompt_str: Optional[str] = None,
+                           code_to_modify: Optional[str] = None) -> Tuple[Optional[str], Optional[AST]]:
+    """
+    Prompt the user for input.
+
+    :param case_query: The case query to prompt the user for.
+    :param prompt_for: The type of information the user should provide for the given case.
+    :param prompt_str: The prompt string to display to the user.
+    :param code_to_modify: The code to modify. If given will be used as a start for user to modify.
+    :return: The user input, and the executable expression that was parsed from the user input.
+    """
+    if prompt_str is None:
+        prompt_str = f"Give {prompt_for} for {case_query.name}"
+    scope = {'case': case_query.case, **case_query.scope}
+    output_type = case_query.attribute_type if prompt_for == PromptFor.Conclusion else bool
+    shell = IPythonShell(scope=scope, header=prompt_str, output_type=output_type, prompt_for=prompt_for,
+                         attribute_name=case_query.attribute_name, attribute_type=case_query.attribute_type,
+                         code_to_modify=code_to_modify)
+    return prompt_user_input_and_parse_to_expression(shell=shell)
+
+
+def prompt_user_input_and_parse_to_expression(shell: Optional[IPythonShell] = None,
+                                              user_input: Optional[str] = None)\
+        -> Tuple[Optional[str], Optional[ast.AST]]:
+    """
+    Prompt the user for input.
+
+    :param shell: The Ipython shell to use for prompting the user.
+    :param user_input: The user input to use. If given, the user input will be used instead of prompting the user.
+    :return: The user input and the AST tree.
+    """
+    while True:
+        if user_input is None:
+            shell = IPythonShell() if shell is None else shell
+            shell.run()
+            user_input = shell.user_input
+            if user_input is None:
+                return None, None
+            print(user_input)
+        try:
+            return user_input, parse_string_to_expression(user_input)
+        except Exception as e:
+            msg = f"Error parsing expression: {e}"
+            logging.error(msg)
+            print(msg)
+            user_input = None