revisit 0.0.5__tar.gz → 0.0.7__tar.gz

{revisit-0.0.5 → revisit-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: revisit
-Version: 0.0.5
+Version: 0.0.7
 Requires-Dist: anywidget
 Requires-Dist: ipykernel>=6.29.5
 Requires-Dist: pydantic>=2.10.5
@@ -58,3 +58,17 @@ yarn run dev
 Open `example.ipynb` in JupyterLab, VS Code, or your favorite editor
 to start developing. Changes made in `js/` will be reflected
 in the notebook.
+
+
+## CODE GEN
+
+```bash
+datamodel-codegen --input StudyConfigSchema.json --output models.py  --custom-template-dir custom_templates --output-model-type pydantic_v2.BaseModel --additional-imports typing.TypedDict --input-file-type jsonschema --special-field-name-prefix we_are_going_to_replace_this && sed -i '' 's/we_are_going_to_replace_this_//g'  src/revisit/models.py
+```
+
+## TESTS
+
+```bash
+cd revisit-py
+python -m unittest tests.test_module_one
+```

{revisit-0.0.5 → revisit-0.0.7}/README.md

@@ -47,3 +47,17 @@ yarn run dev
 Open `example.ipynb` in JupyterLab, VS Code, or your favorite editor
 to start developing. Changes made in `js/` will be reflected
 in the notebook.
+
+
+## CODE GEN
+
+```bash
+datamodel-codegen --input StudyConfigSchema.json --output models.py  --custom-template-dir custom_templates --output-model-type pydantic_v2.BaseModel --additional-imports typing.TypedDict --input-file-type jsonschema --special-field-name-prefix we_are_going_to_replace_this && sed -i '' 's/we_are_going_to_replace_this_//g'  src/revisit/models.py
+```
+
+## TESTS
+
+```bash
+cd revisit-py
+python -m unittest tests.test_module_one
+```

{revisit-0.0.5 → revisit-0.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "revisit"
-version = "0.0.5"
+version = "0.0.7"
 dependencies = [
     "anywidget",
     "ipykernel>=6.29.5",

{revisit-0.0.5 → revisit-0.0.7}/src/revisit/__init__.py

@@ -1,6 +1,10 @@
 # Import main functions from core.py
 from . import revisit
 
+# Import all symbols
+from .revisit import *
+
+
 # Import the Widget class from widget.py
 from .widget import Widget
 

{revisit-0.0.5 → revisit-0.0.7}/src/revisit/revisit.py

@@ -7,6 +7,21 @@ from enum import Enum
 import csv
 from dataclasses import make_dataclass
 import re
+import os
+import shutil
+from . import widget as _widget
+
+
+__all__ = [
+    "component",
+    "sequence",
+    "response",
+    "uiConfig",
+    "studyMetadata",
+    "studyConfig",
+    "data",
+    "widget"
+]
 
 
 class _JSONableBaseModel(BaseModel):
@@ -105,13 +120,24 @@ class _WrappedComponentBlock(_JSONableBaseModel):
 
     def __add__(self, other):
         """Allows addition operator to append to sequence components list."""
-        if isinstance(other, _WrappedComponentBlock) or isinstance(other, _WrappedComponent):
+        if isinstance(other, _WrappedComponent):
             self.component_objects__.append(other)
             self.root.components.append(other.component_name__)
             return self
+        elif isinstance(other, _WrappedComponentBlock):
+            # Extend existing list of components with new set of components for tracking
+            self.component_objects__.extend(other.component_objects__)
+
+            # Add root object to components
+            self.root.components.append(other.root)
+            return self
         return NotImplemented
 
-    def from_data(self, data_list: list):
+    def from_data(self, data_list) -> DataIterator:
+        if not isinstance(data_list, list):
+            raise RevisitError(
+                message="'from_data' must take in a list of data rows. Use reVISit's 'data' method to parse a CSV file into a valid input."
+            )
         return DataIterator(data_list, self)
 
 
@@ -356,6 +382,67 @@ def data(file_path: str) -> List[Any]:
     return data_rows
 
 
+def widget(study: _WrappedStudyConfig, revisitPath: str):
+    if not os.path.isdir(revisitPath):
+        raise RevisitError(message=f'"{revisitPath}" does not exist.')
+
+    extracted_paths = []
+
+    for component in study.root.components.values():
+        if hasattr(component.root, 'path'):
+
+            fileName = component.root.path.split('/')[-1]
+
+            if component.root.type == 'react-component':
+                dest = f"{revisitPath}/src/public/__revisit-widget/assets/{fileName}"
+            else:
+                dest = f"{revisitPath}/public/__revisit-widget/assets/{fileName}"
+
+            extracted_paths.append({
+                "src": component.root.path,
+                "dest": dest
+            })
+
+            newPath = f"__revisit-widget/assets/{fileName}"
+            component.root.path = newPath
+
+    uiConfig = study.root.uiConfig
+    if uiConfig.helpTextPath is not None:
+
+        fileName = uiConfig.helpTextPath.split('/')[-1]
+        dest = f"{revisitPath}/public/__revisit-widget/assets/{fileName}"
+
+        extracted_paths.append({
+            "src": uiConfig.helpTextPath,
+            "dest": dest
+        })
+
+        newPath = f"__revisit-widget/assets/{fileName}"
+        uiConfig.helpTextPath = newPath
+
+    if uiConfig.logoPath is not None:
+
+        fileName = uiConfig.logoPath.split('/')[-1]
+
+        dest = f"{revisitPath}/public/__revisit-widget/assets/{fileName}"
+
+        extracted_paths.append({
+            "src": uiConfig.logoPath,
+            "dest": dest
+        })
+
+        newPath = f"__revisit-widget/assets/{fileName}"
+        uiConfig.logoPath = newPath
+
+    # Copy all files
+    for item in extracted_paths:
+        _copy_file(item['src'], item['dest'])
+
+    w = _widget.Widget()
+    w.config = json.loads(study.__str__())
+    return w
+
+
 # ------- PRIVATE FUNCTIONS ------------ #
 
 def _validate_component(kwargs: dict):
@@ -486,7 +573,7 @@ def pretty_error(errors):
 
 def _get_filtered_kwargs(class_type: Any, kwargs):
     try:
-        possible_items = get_args(class_type.__fields__.get('root').annotation)
+        possible_items = get_args(class_type.model_fields.get('root').annotation)
     except AttributeError:
         possible_items = [class_type]
 
@@ -519,3 +606,14 @@ def _extract_datum_value(text: str) -> str:
     if match:
         return match.group(1)  # Return the captured part (i.e., 'thing')
     return None  # Return None if the pattern doesn't match
+
+
+def _copy_file(src: str, dest: str):
+    # Check if file exists
+    if not os.path.exists(src):
+        raise RevisitError(message=f'File "{src}" not found.')
+
+    os.makedirs(os.path.dirname(dest), exist_ok=True)
+
+    print(f'Copying file from {src} to {dest}')
+    shutil.copyfile(src, dest)

{revisit-0.0.5 → revisit-0.0.7}/src/revisit/widget.py

@@ -47,3 +47,7 @@ class Widget(anywidget.AnyWidget):
         self.internalWidget.value += 1
         # internalWidget.value += 1
         # print("{name} changed from {old} to {new}".format(**change))
+
+    # def set(self, study: rvt._WrappedStudyConfig):
+    #     self.config = json.loads(study.__str__())
+    #     return self