Functionality for retrieving test templates associated with a given part

InvenTree/part/models.py

@@ -41,6 +41,7 @@ from InvenTree.helpers import decimal2string, normalize
 
 from InvenTree.status_codes import BuildStatus, PurchaseOrderStatus
 
+from report import models as ReportModels
 from build import models as BuildModels
 from order import models as OrderModels
 from company.models import SupplierPart
@@ -358,6 +359,20 @@ class Part(MPTTModel):
         self.category = category
         self.save()
 
+    def get_test_report_templates(self):
+        """
+        Return all the TestReport template objects which map to this Part.
+        """
+
+        templates = []
+
+        for report in ReportModels.TestReport.objects.all():
+            if report.matches_part(self):
+                templates.append(report)
+
+        return templates
+
+
     def get_absolute_url(self):
         """ Return the web URL for viewing this part """
         return reverse('part-detail', kwargs={'pk': self.id})