Passed args from render() to _render_alt_profile_html(), included templates folder in setup.py

Author: jamiefeiss

pyldapi/renderer.py

@@ -443,10 +443,10 @@ def _make_rdf_response(self, graph, mimetype=None, headers=None, delete_graph=Tr
         )
 
     def _render_alt_profile_html(
-            self,
-            alt_template: str = "alt.html",
-            additional_alt_template_context=None,
-            alt_template_context_replace=False
+        self,
+        alt_template: str = "alt.html",
+        additional_alt_template_context=None,
+        alt_template_context_replace=False
     ):
         """Renders the Alternates Profile in HTML
 
@@ -509,7 +509,12 @@ def _render_alt_profile_json(self):
             headers=self.headers
         )
 
-    def _render_alt_profile(self):
+    def _render_alt_profile(
+        self,
+        alt_template: str = "alt.html",
+        additional_alt_template_context=None,
+        alt_template_context_replace=False
+    ):
         """
         Return a Flask Response object depending on the value assigned to :code:`self.mediatype`.
 
@@ -519,13 +524,22 @@ def _render_alt_profile(self):
         if self.mediatype == '_internal':
             return self
         if self.mediatype == 'text/html':
-            return self._render_alt_profile_html()
+            return self._render_alt_profile_html(
+                alt_template,
+                additional_alt_template_context,
+                alt_template_context_replace
+            )
         elif self.mediatype in RDF_MEDIATYPES:
             return self._render_alt_profile_rdf()
         else:  # application/json
             return self._render_alt_profile_json()
 
-    def render(self):
+    def render(
+        self,
+        alt_template: str = "alt.html",
+        additional_alt_template_context=None,
+        alt_template_context_replace=False
+    ):
         """
         Use the received profile and mediatype to create a response back to the client.
 
@@ -542,7 +556,11 @@ def render(self):
         if self.vf_error is not None:
             return Response(self.vf_error, status=400, media_type='text/plain')
         elif self.profile == 'alt' or self.profile == 'alternates':
-            return self._render_alt_profile()
+            return self._render_alt_profile(
+                alt_template,
+                additional_alt_template_context,
+                alt_template_context_replace
+            )
         return None
 
     # end making response content

pyldapi/renderer_container.py

@@ -64,7 +64,7 @@ def __init__(self,
         :param super_register: A super-Register URI for this register. Can be within this API or external.
         :type super_register: str
         :param members_template: The Jinja2 template to use for rendering the HTML profile of the register. If None,
-        then it will default to try and use a template called :code:`alt.html`.
+        then it will default to try and use a template called :code:`mem.html`.
         :type members_template: str or None
         :param per_page: Number of items to show per page if not specified in request. If None, then it will default to
         RegisterRenderer.DEFAULT_ITEMS_PER_PAGE.
@@ -195,7 +195,12 @@ def _paging(self):
 
         return None
 
-    def render(self):
+    def render(
+        self,
+        mem_template: str = "mem.html",
+        additional_mem_template_context=None,
+        mem_template_context_replace=False
+    ):
         """
         Renders the register profile.
 
@@ -206,7 +211,11 @@ def render(self):
         if response is None and self.profile == 'mem':
             if self.paging_error is None:
                 if self.mediatype == 'text/html':
-                    return self._render_mem_profile_html()
+                    return self._render_mem_profile_html(
+                        mem_template,
+                        additional_mem_template_context,
+                        mem_template_context_replace
+                    )
                 elif self.mediatype in Renderer.RDF_MEDIA_TYPES:
                     return self._render_mem_profile_rdf()
                 else:
@@ -216,10 +225,10 @@ def render(self):
         return response
 
     def _render_mem_profile_html(
-            self,
-            mem_template: str = "mem.html",
-            additional_mem_template_context=None,
-            mem_template_context_replace=False
+        self,
+        mem_template: str = "mem.html",
+        additional_mem_template_context=None,
+        mem_template_context_replace=False
     ):
         _template_context = {
             'uri': self.instance_uri,

setup.py

@@ -3,7 +3,7 @@
 import codecs
 import re
 import os
-from setuptools import setup
+from setuptools import setup, find_packages
 
 
 def open_local(paths, mode='r', encoding='utf8'):
@@ -29,7 +29,12 @@ def open_local(paths, mode='r', encoding='utf8'):
 
 setup(
     name='pyldapi',
-    packages=['pyldapi'],
+    packages=find_packages(),
+    package_dir={"pyldapi": "pyldapi"},
+    package_data={
+        "pyldapi": ["templates/*"],
+    },
+    include_package_data=True,
     version=version,
     description='A very small module to add Linked Data API functionality to '
                 'a Python FastAPI or Flask (v3.x) installation',