Merge pull request #136 from XpressAI/adry/add-import-component

✨Add component that enable to add import
XpressAI · Apr 4, 2022 · 3c573a9 · 3c573a9
2 parents aabafe2 + 6c15c66
commit 3c573a9
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/components/xircuitBodyWidget.tsx b/src/components/xircuitBodyWidget.tsx
@@ -401,6 +401,17 @@ export const BodyWidget: FC<BodyWidgetProps> = ({
 			let current_node: any;
 			let package_name: string = "components";
 
+			const addImportNode = getNodeModelByName(nodeModels, 'AddImport');
+			if (componentName == 'AddImport') {
+				const importPortName = addImportNode['portsIn'][1].getOptions()['name']
+				const getImportPortLinks = addImportNode.getPorts()[importPortName].getLinks();
+				for (let portLink in getImportPortLinks) {
+					// Add value of import_str port for importing
+					const importLabel = getImportPortLinks[portLink].getSourcePort().getOptions()["label"];
+					pythonCode += importLabel + "\n";
+				}
+			}
+
 			if (component_exist != -1) {
 				current_node = componentList[component_exist];
 				package_name = current_node["package_name"];

diff --git a/xai_components/xai_template/example_components.py b/xai_components/xai_template/example_components.py
@@ -103,4 +103,16 @@ def execute(self, ctx) -> None:
 
         print(f"After Adding Context:\n{ctx}")
 
-        self.done = True
+        self.done = True
+
+@xai_component(color="red")
+class AddImport(Component):
+    import_str: InArg[str]
+
+    def __init__(self):
+        self.import_str = InArg.empty()
+        self.done = False
+
+    def execute(self, ctx) -> None:
+
+        self.done = True