Move PegdownMarkdownFormatter to docs class

Signed-off-by: Jason T. Brown <jason@astraea.earth>

Author: vpipkt

pyrasterframes/src/main/python/docs/__init__.py

@@ -21,6 +21,8 @@
 import os
 from pyspark.sql import DataFrame
 from pyrasterframes import RFContext
+from pweave import PwebPandocFormatter
+
 
 def docs_dir():
     return os.path.dirname(os.path.realpath(__file__))
@@ -38,3 +40,12 @@ def resource_dir():
 
 def resource_dir_uri():
     return 'file://' + resource_dir()
+
+
+class PegdownMarkdownFormatter(PwebPandocFormatter):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    # Pegdown doesn't support the width and label options.
+    def make_figure_string(self, figname, width, label, caption=""):
+        return "![%s](%s)" % (caption, figname)

pyrasterframes/src/main/python/setup.py

@@ -57,15 +57,6 @@ class PweaveDocs(distutils.cmd.Command):
         ('quick=', 'q', 'Check to see if the source file is newer than existing output before building. Defaults to `False`.')
     ]
 
-    from pweave import PwebPandocFormatter
-
-    class PegdownMarkdownFormatter(PwebPandocFormatter):
-        def __init__(self, *args, **kwargs):
-            super().__init__(*args, **kwargs)
-
-        # Pegdown doesn't support the width and label options.
-        def make_figure_string(self, figname, width, label, caption=""):
-            return "![%s](%s)" % (caption, figname)
 
     def initialize_options(self):
         """Set default values for options."""
@@ -96,6 +87,8 @@ def run(self):
         """Run pweave."""
         import traceback
         import pweave
+        from docs import PegdownMarkdownFormatter
+
         bad_words = ["Error"]
         pweave.rcParams["chunk"]["defaultoptions"].update({'wrap': False, 'dpi': 100})
         if self.format == 'markdown':