From eb6310c774166ec8652801d2ed341a5e336c5ecc Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Thu, 11 Mar 2021 15:01:15 +1100
Subject: [PATCH] Render company image to report
---
InvenTree/report/models.py | 2 ++
InvenTree/report/templatetags/report.py | 34 +++++++++++++++++++++++++
2 files changed, 36 insertions(+)
diff --git a/InvenTree/report/models.py b/InvenTree/report/models.py
index 24bb6da219..a838396bf0 100644
--- a/InvenTree/report/models.py
+++ b/InvenTree/report/models.py
@@ -425,6 +425,7 @@ class PurchaseOrderReport(ReportTemplateBase):
return {
'description': order.description,
+ 'lines': order.lines,
'order': order,
'reference': order.reference,
'supplier': order.supplier,
@@ -459,6 +460,7 @@ class SalesOrderReport(ReportTemplateBase):
return {
'customer': order.customer,
'description': order.description,
+ 'lines': order.lines,
'order': order,
'prefix': common.models.InvenTreeSetting.get_setting('SALESORDER_REFERENCE_PREFIX'),
'reference': order.reference,
diff --git a/InvenTree/report/templatetags/report.py b/InvenTree/report/templatetags/report.py
index 9cd28139f2..8ff15ccdee 100644
--- a/InvenTree/report/templatetags/report.py
+++ b/InvenTree/report/templatetags/report.py
@@ -8,6 +8,7 @@ from django import template
from django.conf import settings
from django.utils.safestring import mark_safe
+from company.models import Company
from part.models import Part
from stock.models import StockItem
@@ -72,6 +73,39 @@ def part_image(part):
return f"file://{path}"
+@register.simple_tag()
+def company_image(company):
+ """
+ Return a fully-qualified path for a company image
+ """
+
+ # If in debug mode, return the URL to the image, not a local file
+ debug_mode = InvenTreeSetting.get_setting('REPORT_DEBUG_MODE')
+
+ if type(company) is Company:
+ img = company.image.name
+ else:
+ img = ''
+
+ if debug_mode:
+ if img:
+ return os.path.join(settings.MEDIA_URL, img)
+ else:
+ return os.path.join(settings.STATIC_URL, 'img', 'blank_image.png')
+
+ else:
+ path = os.path.join(settings.MEDIA_ROOT, img)
+ path = os.path.abspath(path)
+
+ if not os.path.exists(path) or not os.path.isfile(path):
+ # Image does not exist
+ # Return the 'blank' image
+ path = os.path.join(settings.STATIC_ROOT, 'img', 'blank_image.png')
+ path = os.path.abspath(path)
+
+ return f"file://{path}"
+
+
@register.simple_tag()
def internal_link(link, text):
"""