revisit 0.0.11__py2.py3-none-any.whl → 0.0.13__py2.py3-none-any.whl

revisit/revisit.py

@@ -10,7 +10,8 @@ import re
 import os
 import shutil
 from . import widget as _widget
-
+import ast
+import time
 
 __all__ = [
     "component",
@@ -59,7 +60,7 @@ class _WrappedResponse(_JSONableBaseModel):
         return self
 
     def clone(self):
-        return response(**self.root.__dict__)
+        return __response__(**self.root.__dict__)
 
 
 # Private
@@ -67,6 +68,7 @@ class _WrappedComponent(_JSONableBaseModel):
     component_name__: str
     base__: Optional[_WrappedComponent] = None
     context__: Optional[dict] = None
+    metadata__: Optional[dict] = None
     root: rvt_models.IndividualComponent
 
     def model_post_init(self, __context: Any) -> None:
@@ -77,7 +79,7 @@ class _WrappedComponent(_JSONableBaseModel):
     def responses(self, responses: List[_WrappedResponse]) -> _WrappedComponent:
         for item in responses:
             if not isinstance(item, _WrappedResponse):
-                raise ValueError(f'Expecting type Response got {type(item)}')
+                raise RevisitError(message=f'Expecting type Response but got {type(item)}')
         self.root.response = responses
         return self
 
@@ -93,7 +95,7 @@ class _WrappedComponent(_JSONableBaseModel):
                 # Get dict 
                 response_dict = r.root.__dict__
                 # Create new response
-                new_response = response(**response_dict)
+                new_response = __response__(**response_dict)
                 # Set with new values
                 new_response.set(**kwargs)
                 # Filter out old response
@@ -119,7 +121,7 @@ class _WrappedComponent(_JSONableBaseModel):
         return self
 
     def clone(self, component_name__):
-        return component(**self.root.__dict__, component_name__=component_name__)
+        return __component__(**self.root.__dict__, component_name__=component_name__)
 
 
 class _WrappedStudyMetadata(_JSONableBaseModel):
@@ -156,6 +158,37 @@ class _WrappedComponentBlock(_JSONableBaseModel):
             )
         return DataIterator(data_list, self)
 
+    def get_component(self, name: str) -> _WrappedComponent:
+        for entry in self.component_objects__:
+            if entry.component_name__ == name:
+                return entry
+
+    def permute(
+        self,
+        factors: List[str],
+        order: rvt_models.Order,
+        numSamples: Optional[int] = None,
+        component_function=None
+    ) -> None:
+        # Convert to JSON
+        self_json = json.loads(self.__str__())
+        # Get all current component dictionaries
+        components_dict = {c.component_name__: c for c in self.component_objects__}
+        # Recursively start permutation function
+        new_permuted_component_block = _recursive_json_permutation(
+            self_json,
+            factors=factors,
+            order=order,
+            numSamples=numSamples,
+            input_components=components_dict,
+            component_function=component_function
+        )
+        # Set new objects
+        self.component_objects__ = new_permuted_component_block.component_objects__
+        # Set new root
+        self.root = new_permuted_component_block.root
+        return self
+
 
 class _WrappedStudyConfig(_JSONableBaseModel):
     root: rvt_models.StudyConfig
@@ -198,7 +231,7 @@ class DataIterator:
                             current_dict[key] = value
                     else:
                         current_dict[key] = value
-            curr_component = component(**current_dict)
+            curr_component = __component__(**current_dict)
             self.parent_class = self.parent_class + curr_component
         # Return the parent class calling iterator when component is finished.
         return self.parent_class
@@ -276,6 +309,10 @@ def component(**kwargs) -> _WrappedComponent:
         raise RevisitError(e.errors())
 
 
+# Shadowing
+__component__ = component
+
+
 # Response factory function
 @overload
 def response(**kwargs: Unpack[rvt_models.NumericalResponseType]) -> _WrappedResponse: ...
@@ -312,6 +349,10 @@ def response(**kwargs) -> _WrappedResponse:
         raise RevisitError(e.errors())
 
 
+# Shadowing
+__response__ = response
+
+
 def studyMetadata(**kwargs: Unpack[rvt_models.StudyMetadataType]):
     filter_kwargs = _get_filtered_kwargs(rvt_models.StudyMetadata, kwargs)
     base_model = rvt_models.StudyMetadata(**filter_kwargs)
@@ -350,6 +391,10 @@ def sequence(**kwargs: Unpack[rvt_models.ComponentBlockType]):
     return _WrappedComponentBlock(**kwargs, root=base_model, component_objects__=valid_components)
 
 
+# Shadowing
+__sequence__ = sequence
+
+
 @overload
 def studyConfig(**kwargs: Unpack[_StudyConfigType]) -> _WrappedStudyConfig: ...
 @overload
@@ -646,3 +691,60 @@ def _copy_file(src: str, dest: str):
 
     print(f'Copying file from {src} to {dest}')
     shutil.copyfile(src, dest)
+
+
+def _recursive_json_permutation(
+    input_json: dict,
+    factors: List[str],
+    order: rvt_models.Order,
+    numSamples: int,
+    input_components: dict,
+    component_function=None
+):
+    new_seq = __sequence__(order=order, numSamples=numSamples)
+    while input_json['components']:
+        c = input_json['components'].pop()
+        # If component name
+        if isinstance(c, str):
+            # Get orig component
+            curr_comp = input_components[c]
+            # Create new comp block for permuting this component across all factors
+            curr_seq = __sequence__(order=order, numSamples=numSamples)
+            # Generate new comp for each
+            for entry in factors:
+                # Assign params
+                metadata = entry
+                if curr_comp.metadata__ is not None:
+                    metadata = {**curr_comp.metadata__, **entry}
+                # Create new component
+                comp_name = ":".join(f"{key}:{value}" for key, value in entry.items())
+                if component_function:
+                    new_comp = component_function(**metadata)
+                else:
+                    new_comp = __component__(
+                        base__=curr_comp,
+                        component_name__=f"{c}__{comp_name}",
+                        metadata__=metadata
+                    )
+                # Add to curr seq block
+                curr_seq = curr_seq + new_comp
+            # Add seq block to outer seq block
+        else:
+            new_input_json = c
+            temp_num_samples = None
+            if new_input_json.get('numSamples') is not None:
+                temp_num_samples = new_input_json['numSamples']
+
+            curr_seq = _recursive_json_permutation(
+                input_json=new_input_json,
+                order=order,
+                numSamples=numSamples,
+                input_components=input_components,
+                factors=factors,
+                component_function=component_function
+            )
+            curr_seq.root.order = new_input_json['order']
+            curr_seq.root.numSamples = temp_num_samples
+        new_seq = new_seq + curr_seq
+    return new_seq
+        

revisit/static/widget.css

@@ -1,2 +1 @@
 .revisit_notebook_widget button{background:#f0f0f0;border-radius:5px;border:1px solid #ccc;cursor:pointer;font-family:Roboto,sans-serif;font-size:1em;margin:5px;padding:5px 10px;transition:box-shadow .25s ease-in-out;&:hover{background:#e0e0e0;box-shadow:0 0 5px #ccc}&:active{background:#d0d0d0;box-shadow:0 0 5px #ccc inset}}.revisit_notebook_widget iframe{border:0;margin:0;width:100%}.revisit_notebook_widget{.sequence{width:100%;display:flex}.sequenceComponent{display:flex;padding-left:4px;padding-right:4px;padding-bottom:2px;&:hover{background:#0000001a}}.sequenceComponentIndividual{background:#000;width:5px;height:20px;margin-top:5px}}
-/*# sourceMappingURL=data:application/json;base64,ewogICJ2ZXJzaW9uIjogMywKICAic291cmNlcyI6IFsiLi4vLi4vLi4vanMvd2lkZ2V0LmNzcyJdLAogICJzb3VyY2VzQ29udGVudCI6IFsiLnJldmlzaXRfbm90ZWJvb2tfd2lkZ2V0IGJ1dHRvbiB7XG5cdGJhY2tncm91bmQ6ICNmMGYwZjA7XG5cdGJvcmRlci1yYWRpdXM6IDVweDtcblx0Ym9yZGVyOiAxcHggc29saWQgI2NjYztcblx0Y3Vyc29yOiBwb2ludGVyO1xuXHRmb250LWZhbWlseTogXCJSb2JvdG9cIiwgc2Fucy1zZXJpZjtcblx0Zm9udC1zaXplOiAxZW07XG5cdG1hcmdpbjogNXB4O1xuXHRwYWRkaW5nOiA1cHggMTBweDtcblx0dHJhbnNpdGlvbjogYm94LXNoYWRvdyAwLjI1cyBlYXNlLWluLW91dDtcblxuXHQmOmhvdmVyIHtcblx0XHRiYWNrZ3JvdW5kOiAjZTBlMGUwO1xuXHRcdGJveC1zaGFkb3c6IDAgMCA1cHggMCAjY2NjO1xuXHR9XG5cblx0JjphY3RpdmUge1xuXHRcdGJhY2tncm91bmQ6ICNkMGQwZDA7XG5cdFx0Ym94LXNoYWRvdzogMCAwIDVweCAwICNjY2MgaW5zZXQ7XG5cblx0fVxufVxuXG4ucmV2aXNpdF9ub3RlYm9va193aWRnZXQgaWZyYW1lIHtcblx0Ym9yZGVyOiAwO1xuXHRtYXJnaW46IDA7XG5cdHdpZHRoOiAxMDAlO1xufVxuXG4ucmV2aXNpdF9ub3RlYm9va193aWRnZXQge1xuXHQuc2VxdWVuY2Uge1xuXHRcdHdpZHRoOiAxMDAlO1xuXHRcdGRpc3BsYXk6IGZsZXg7XG5cdH1cblxuXHQuc2VxdWVuY2VDb21wb25lbnQge1xuXHRcdGRpc3BsYXk6IGZsZXg7XG5cdFx0cGFkZGluZy1sZWZ0OiA0cHg7XG5cdFx0cGFkZGluZy1yaWdodDogNHB4O1xuXHRcdHBhZGRpbmctYm90dG9tOiAycHg7XG5cblx0XHQmOmhvdmVyIHtcblx0XHRcdGJhY2tncm91bmQ6IHJnYmEoMCwwLDAsLjEpO1xuXHRcdH1cblx0fVxuXG5cdC5zZXF1ZW5jZUNvbXBvbmVudEluZGl2aWR1YWwge1xuXHRcdGJhY2tncm91bmQ6ICMwMDAwMDA7XG5cdFx0d2lkdGg6IDVweDtcblx0XHRoZWlnaHQ6IDIwcHg7XG5cdFx0bWFyZ2luLXRvcDogNXB4O1xuXHR9XG59Il0sCiAgIm1hcHBpbmdzIjogIkFBQUEsQ0FBQyx3QkFBd0IsT0FDeEIsV0FBWSxRQURiLGNBRWdCLElBQ2YsT0FBUSxJQUFJLE1BQU0sS0FDbEIsT0FBUSxRQUNSLFlBQWEsTUFBUSxDQUFFLFdBQ3ZCLFVBQVcsSUFOWixPQU9TLElBUFQsUUFRVSxJQUFJLEtBQ2IsV0FBWSxXQUFXLEtBQU0sWUFFN0IsQ0FBQyxPQUNBLFdBQVksUUFDWixXQUFZLEVBQUUsRUFBRSxJQUFNLElBQ3ZCLENBRUEsQ0FBQyxRQUNBLFdBQVksUUFDWixXQUFZLEVBQUUsRUFBRSxJQUFNLEtBQUssS0FFNUIsQ0FDRCxDQUVBLENBdkJDLHdCQXVCd0IsT0FDeEIsT0FBUSxFQXhCVCxPQXlCUyxFQUNSLE1BQU8sSUFDUixDQUVBLENBN0JDLHdCQThCQSxDQUFDLFNBQ0EsTUFBTyxLQUNQLFFBQVMsSUFDVixDQUVBLENBQUMsa0JBQ0EsUUFBUyxLQUNULGFBQWMsSUFDZCxjQUFlLElBQ2YsZUFBZ0IsSUFFaEIsQ0FBQyxPQUNBLFdBQVksU0FDYixDQUNELENBRUEsQ0FBQyw0QkFDQSxXQUFZLEtBQ1osTUFBTyxJQUNQLE9BQVEsS0FDUixXQUFZLEdBQ2IsQ0FDRCIsCiAgIm5hbWVzIjogW10KfQo= */