From dc70b5ef112764b30e27e8fa0f7e9c5c3a7896d4 Mon Sep 17 00:00:00 2001
From: Matthias <matthias.mair@oewf.org>
Date: Sat, 18 Sep 2021 03:40:21 +0200
Subject: [PATCH] urls as own mixin

---
 InvenTree/plugins/integration/integration.py | 39 +++++++++++++++++++-
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/InvenTree/plugins/integration/integration.py b/InvenTree/plugins/integration/integration.py
index f101bcc1d0..96cac27683 100644
--- a/InvenTree/plugins/integration/integration.py
+++ b/InvenTree/plugins/integration/integration.py
@@ -40,6 +40,43 @@ class SettingsMixin:
         return None
 
 
+class UrlsMixin:
+    """Mixin that enables urls for the plugin"""
+    def __init__(self):
+        super().__init__()
+
+        self.add_mixin('urls')
+        self.urls = self.setup_urls()
+
+    def setup_urls(self):
+        """
+        setup url endpoints for this plugin
+        """
+        if self.urlpatterns:
+            return self.urlpatterns
+        return None
+
+    @property
+    def base_url(self):
+        return f'{settings.PLUGIN_URL}/{self.plugin_name()}/'
+
+    @property
+    def urlpatterns(self):
+        """
+        retruns the urlpatterns for this plugin
+        """
+        if self.has_urls:
+            return url(f'^{self.plugin_name()}/', include(self.urls), name=self.plugin_name())
+        return None
+
+    @property
+    def has_urls(self):
+        """
+        does this plugin use custom urls
+        """
+        return bool(self.urls)
+
+
 class IntegrationPlugin(plugin.InvenTreePlugin):
     """
     The IntegrationPlugin class is used to integrate with 3rd party software
@@ -49,8 +86,6 @@ class IntegrationPlugin(plugin.InvenTreePlugin):
         self.add_mixin('base')
         super().__init__()
 
-        self.urls = self.setup_urls()
-
     def add_mixin(self, key: str):
         if not hasattr(self, 'mixins'):
             self.mixins = {}