Add Python Dependency Deck

flytekit/core/context_manager.py

@@ -741,6 +741,17 @@
 
         return _get_deck(typing.cast(ExecutionState, self.execution_state).user_space_params)
 
+    def add_deck(self, deck: Deck) -> None:
+        """
+        Adds a deck to the current context.
+        """
+        if (
+            self.execution_state is not None
+            and self.execution_state.user_space_params is not None
+            and hasattr(self.execution_state.user_space_params, "decks")
+        ):
+            self.execution_state.user_space_params.decks.append(deck)
+
     @dataclass
     class Builder(object):
         file_access: FileAccessProvider

flytekit/deck/deck.py

@@ -174,3 +174,39 @@
         autoescape=select_autoescape(enabled_extensions=("html",)),
     )
     return env.get_template("template.html")
+
+
+class PythonDependencyDeck(Deck):
+    """
+    PythonDependencyDeck is a deck that contains information about the library name and version.
+    """
+
+    def __init__(self, name: Optional[str] = "Python Dependency", html: Optional[str] = ""):
+        super().__init__(name, html)
+
+    @property
+    def html(self) -> str:
+        try:
+            from flytekitplugins.deck.renderer import TableRenderer
+        except ImportError:
+            warning_info = "Plugin 'flytekit-deck-standard' is not installed. To display python dependency, install the plugin in the image."
+            logger.warning(warning_info)
+            return warning_info
+
+        import subprocess
+
+        import pandas as pd
+
+        try:
+            installed_packages = subprocess.check_output(["pip", "freeze"]).decode().split("\n")
+        except subprocess.CalledProcessError as e:
+            logger.error(f"Error occurred while fetching installed packages: {e}")
+            return ""
+
+        columns_name = ["Library", "Version"]
+        installed_packages = [package.split("==") for package in installed_packages if package]
+        df = pd.DataFrame(installed_packages, columns=columns_name)
+        html = TableRenderer().to_html(df, header_labels=columns_name)
+        # Add CSS style to center align the table content
+        html = f"<div style='text-align: center;'>{html}</div>"
+        return html