fix: only import when there are imports (#5)

dylibso · Oct 9, 2024 · d53e56c · d53e56c
1 parent 21fbea6
commit d53e56c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/template/plugin/__init__.py.ejs b/template/plugin/__init__.py.ejs
@@ -2,7 +2,10 @@
 
 import extism # pyright: ignore
 import plugin
+
+<% if (Object.values(schema.schemas).length > 0){ %>
 from pdk_types import <%- Object.values(schema.schemas).map(schema => schema.name).join(", ") %> # noqa: F401 
+<% } %>
 
 # Imports
 

diff --git a/template/plugin/pdk_imports.py.ejs b/template/plugin/pdk_imports.py.ejs
@@ -1,7 +1,10 @@
 # THIS FILE WAS GENERATED BY `xtp-python-bindgen`. DO NOT EDIT.
 
 import extism # noqa: F401 # pyright: ignore
+
+<% if (Object.values(schema.schemas).length > 0) { %>
 from pdk_types import <%- Object.values(schema.schemas).map(schema => schema.name).join(", ") %> # noqa: F401
+<% } %>
 
 <% schema.imports.forEach(imp => { %>
 <% if (hasComment(imp)) -%>

diff --git a/template/plugin/plugin.py.ejs b/template/plugin/plugin.py.ejs
@@ -1,6 +1,10 @@
 import extism # noqa: F401 # pyright: ignore
+<% if (Object.values(schema.schemas).length > 0) { %>
 from pdk_types import <%- Object.values(schema.schemas).map(schema => schema.name).join(", ") %> # noqa: F401
+<% } %>
+<% if (schema.imports.length > 0) { %>
 from pdk_imports import <%- schema.imports.map(schema => camelToSnakeCase(schema.name)).join(", ") %> # noqa: F401
+<% } %>
 from typing import List, Optional # noqa: F401
 
 <% schema.exports.forEach(ex => { -%>