feat: add source code deck

plugins/flytekit-deck-standard/flytekitplugins/deck/renderer.py

@@ -1,5 +1,9 @@
 from typing import TYPE_CHECKING, List, Optional, Union
 
+from pygments import highlight
+from pygments.formatters.html import HtmlFormatter
+from pygments.lexers.python import PythonLexer
+
 from flytekit import lazy_module
 from flytekit.types.file import FlyteFile
 
@@ -15,6 +19,45 @@
     PIL = lazy_module("PIL")
 
 
+class SourceCodeRenderer:
+    """
+    Convert Python source code to HTML, and return HTML as a unicode string.
+    """
+
+    def __init__(self, title: str = "Source Code"):
+        self._title = title
+
+    def to_html(self, source_code: str) -> str:
+        """
+        Convert the provided Python source code into HTML format using Pygments library.
+
+        This method applies a colorful style and replaces the color "#fff0f0" with "#ffffff" in CSS.
+
+        Args:
+            source_code (str): The Python source code to be converted.
+
+        Returns:
+            str: The resulting HTML as a string, including CSS and highlighted source code.
+        """
+        formatter = HtmlFormatter(style="colorful")
+        css = self._get_css(formatter)
+        html = highlight(source_code, PythonLexer(), formatter)
+        return f"<style>{css}</style>{html}"
+
+    @staticmethod
+    def _get_css(formatter: HtmlFormatter) -> str:
+        """
+        Get the CSS from the provided HtmlFormatter instance and replace the color "#fff0f0" with "#ffffff".
+
+        Args:
+            formatter (HtmlFormatter): An instance of HtmlFormatter.
+
+        Returns:
+            str: The CSS string with "#fff0f0" replaced by "#ffffff".
+        """
+        return formatter.get_style_defs(".highlight").replace("#fff0f0", "#ffffff")
+
+
 class FrameProfilingRenderer:
     """
     Generate a ProfileReport based on a pandas DataFrame

plugins/flytekit-deck-standard/setup.py

@@ -4,7 +4,14 @@
 
 microlib_name = f"flytekitplugins-{PLUGIN_NAME}-standard"
 
-plugin_requires = ["flytekit>=1.3.0b2,<2.0.0", "markdown", "plotly", "ydata-profiling", "ipywidgets"]
+plugin_requires = [
+    "flytekit",
+    "markdown",
+    "plotly",
+    "ydata-profiling",
+    "ipywidgets",
+    "pygments",
+]
 
 __version__ = "0.0.0+develop"
 

plugins/flytekit-deck-standard/tests/test_renderer.py

@@ -4,13 +4,16 @@
 import markdown
 import pandas as pd
 import pytest
+from pygments.formatters.html import HtmlFormatter
+
 from flytekitplugins.deck.renderer import (
     BoxRenderer,
     FrameProfilingRenderer,
     GanttChartRenderer,
     ImageRenderer,
     MarkdownRenderer,
     TableRenderer,
+    SourceCodeRenderer,
 )
 from PIL import Image
 
@@ -80,3 +83,15 @@ def test_table_renderer():
 def test_gantt_chart_renderer():
     renderer = GanttChartRenderer()
     assert "Plotlyconfig = {Mathjaxconfig: 'Local'}" in renderer.to_html(time_info_df).title()
+
+
+def test_get_css():
+    formatter = HtmlFormatter(style="colorful")
+    css = SourceCodeRenderer._get_css(formatter)
+
+    # Check if the returned CSS contains the expected style
+    assert ".highlight" in css
+
+    # Check if the color "#fff0f0" has been replaced with "#ffffff"
+    assert "#fff0f0" not in css
+    assert "#ffffff" in css