report base dir

app/report/gen_report.py

@@ -16,7 +16,7 @@ from PIL import Image as PILImage
 from openpyxl.drawing.xdr import XDRPoint2D, XDRPositiveSize2D
 from openpyxl.utils.units import pixels_to_EMU
 from openpyxl.drawing.spreadsheet_drawing import AnchorMarker, TwoCellAnchor
-
+from django.conf import settings
 
 def set_image_with_offset_old(sheet, img, cell_coordinate, offset_x=0, offset_y=0):
     """
@@ -149,11 +149,11 @@ def gen_xlsx(template_file, selected_sheets, prefix_filename, data):
     Returns:
         str: Path of the generated Excel file.
     """
-    checked_image_path = "/app/report/checkbox_checked.jpg"  # Path to the checked checkbox image
-    unchecked_image_path = "/app/report/checkbox_unchecked.jpg"  # Path to the unchecked checkbox image
+    checked_image_path = f"{settings.BASE_DIR}/report/checkbox_checked.jpg"  # Path to the checked checkbox image
+    unchecked_image_path = f"{settings.BASE_DIR}/report/checkbox_unchecked.jpg"  # Path to the unchecked checkbox image
 
     # Load the template workbook
-    print("before load workbook {template_file}")
+    print(f"before load workbook {template_file}")
     workbook = load_workbook(template_file)
     print("after load")
     # Remove sheets not in selected_sheets

app/report/views.py

@@ -17,7 +17,7 @@ import json
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.models import User
 from legacy.models import Data
-
+from django.conf import settings
 
 def index(request):
     reports = Report.objects.all()
@@ -170,9 +170,9 @@ def create_coi_file(lot_no, sheets, user, md):
     pprint(merged_data)
 
     output_file = gen_xlsx(
-        template_file="/app/report/coi_templates.xlsx",
+        template_file=f"{settings.BASE_DIR}/report/coi_templates.xlsx",
         selected_sheets=sheets,  # Replace with your actual sheet names
-        prefix_filename="/app/media/coi",
+        prefix_filename=f"{settings.BASE_DIR}/media/coi",
         data=merged_data
     )
     report = Report.objects.create(
@@ -255,8 +255,12 @@ def coi_view(request):
                     pcs = int(first_result.PRO5) - int(first_result.PRO27)
                 except:
                     pcs = 0
-                size_str = f"{first_result.PRO10}x{first_result.PRO11}x{first_result.PRO12}";
-                spec = f"{first_result.PRO13} {first_result.PRO14} {first_result.PRO15} {first_result.PRO16} {first_result.PRO17} {first_result.PRO18}"
+                if first_result:
+                    size_str = f"{first_result.PRO10}x{first_result.PRO11}x{first_result.PRO12}";
+                    spec = f"{first_result.PRO13} {first_result.PRO14} {first_result.PRO15} {first_result.PRO16} {first_result.PRO17} {first_result.PRO18}"
+                else:
+                    size_str = ""
+                    spec = ""
                 return render(request, 'report/coi.html', {'result': first_result, 
                                                            'pcs':pcs, 
                                                            'size_str': size_str,