From 2bddba42da4485842d8bdff8594faf356fed1d4b Mon Sep 17 00:00:00 2001
From: Oliver <oliver.henry.walters@gmail.com>
Date: Tue, 14 Mar 2023 16:09:04 +1100
Subject: [PATCH] Adds new 'role' specific for return orders

---
 InvenTree/users/admin.py  | 6 +++++-
 InvenTree/users/models.py | 9 +++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/InvenTree/users/admin.py b/InvenTree/users/admin.py
index d26a7f97fc..6ac2cece36 100644
--- a/InvenTree/users/admin.py
+++ b/InvenTree/users/admin.py
@@ -85,7 +85,7 @@ class RoleGroupAdmin(admin.ModelAdmin):  # pragma: no cover
     ]
 
     list_display = ('name', 'admin', 'part_category', 'part', 'stocktake', 'stock_location',
-                    'stock_item', 'build', 'purchase_order', 'sales_order')
+                    'stock_item', 'build', 'purchase_order', 'sales_order', 'return_order')
 
     def get_rule_set(self, obj, rule_set_type):
         """Return list of permissions for the given ruleset."""
@@ -161,6 +161,10 @@ class RoleGroupAdmin(admin.ModelAdmin):  # pragma: no cover
         """Return the ruleset for the SalesOrder role"""
         return self.get_rule_set(obj, 'sales_order')
 
+    def return_order(self, obj):
+        """Return the ruleset ofr the ReturnOrder role"""
+        return self.get_rule_set(obj, 'returns')
+
     def get_formsets_with_inlines(self, request, obj=None):
         """Return all inline formsets"""
         for inline in self.get_inline_instances(request, obj):
diff --git a/InvenTree/users/models.py b/InvenTree/users/models.py
index f5d5ce2148..b9be20d682 100644
--- a/InvenTree/users/models.py
+++ b/InvenTree/users/models.py
@@ -42,6 +42,7 @@ class RuleSet(models.Model):
         ('build', _('Build Orders')),
         ('purchase_order', _('Purchase Orders')),
         ('sales_order', _('Sales Orders')),
+        ('returns', _('Returns')),
     ]
 
     RULESET_NAMES = [
@@ -155,6 +156,14 @@ class RuleSet(models.Model):
             'order_salesorderextraline',
             'order_salesordershipment',
             'report_salesorderreport',
+        ],
+        'returns': [
+            'company_company',
+            'company_companyattachment',
+            'order_returnorder',
+            'order_returnorderlineitem',
+            'order_returnorderextraline',
+            'order_returnorderattachment',
         ]
     }