Skip to content

Commit 427b12e

Browse files
author
Joel Collins
committed
Added set_extension method to View class
1 parent 15c364a commit 427b12e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

src/labthings/extensions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def add_view(self, view_class, *urls, endpoint=None, **kwargs):
110110
self._rules[url] = self._views[endpoint]
111111

112112
# Store this extension name as the View owner
113-
view_class._parent_extension_name = self.name
113+
view_class.set_extension(self.name)
114114

115115
def on_register(self, function, args=None, kwargs=None):
116116
"""

src/labthings/views/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,10 @@ def extension(self):
6161
return find_extension(self._parent_extension_name)
6262
return None
6363

64+
@classmethod
65+
def set_extension(cls, extension_name: str):
66+
cls._parent_extension_name = extension_name
67+
6468
@classmethod
6569
def get_tags(cls):
6670
""" """

0 commit comments

Comments
 (0)