Нет описания

views.py 9.0KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230
  1. from django.shortcuts import render, redirect, get_object_or_404
  2. from django.core.paginator import Paginator
  3. from django.contrib import messages
  4. from core.models import Report
  5. from core.forms import ReportForm
  6. from core.utils import ConfigurableCRUDView, queryFromMaster
  7. from .filters import ReportFilter
  8. from .forms import ExportOptionsForm
  9. from pprint import pprint
  10. from .gen_report import gen_xlsx
  11. from django.core.files.base import File
  12. from pathlib import Path
  13. from django.views.decorators.csrf import csrf_exempt
  14. from django.http import JsonResponse, HttpResponseBadRequest
  15. import json
  16. from django.contrib.auth.decorators import login_required
  17. from django.contrib.auth.models import User
  18. def index(request):
  19. reports = Report.objects.all()
  20. report_filter = ReportFilter(request.GET, queryset=reports)
  21. # Paginate the filtered queryset
  22. paginator = Paginator(report_filter.qs, 10) # Show 10 reports per page
  23. page_number = request.GET.get('page')
  24. page_obj = paginator.get_page(page_number)
  25. context = {
  26. 'filter': report_filter,
  27. 'page_obj': page_obj,
  28. }
  29. return render(request, 'report/index.html', context)
  30. def report_create_view(request):
  31. if request.method == "POST":
  32. form = ReportForm(request.POST)
  33. if form.is_valid():
  34. form.save()
  35. messages.success(request, "Report created successfully!")
  36. return redirect("report:report_index") # Adjust with your report list view name
  37. else:
  38. form = ReportForm()
  39. return render(request, "report/create.html", {"form": form})
  40. class ReportCRUDView(ConfigurableCRUDView):
  41. model = Report
  42. list_template_name = 'legacy/datacrud_list.html'
  43. detail_template_name = 'legacy/datacrud_detail.html'
  44. form_template_name = 'report/report_form.html'
  45. confirm_delete_template_name = 'legacy/datacrud_confirm_delete.html'
  46. filterset_class = ReportFilter
  47. page_title = "Reports"
  48. # URL name mappings
  49. list_url_name = 'report:report-list'
  50. create_url_name = 'report:report-create'
  51. update_url_name = 'report:report-update'
  52. delete_url_name = 'report:report-delete'
  53. config_fields = ["name", "file", "created_by", "created_at"]
  54. config_field_orders = ["id", "name", "created_by"]
  55. # config_readonly_fields = ["lot_no"]
  56. # config_edit_fields = ["lot_no", "code"]
  57. ordering = ["-created_at", "-id",]
  58. def create_coi_file(lot_no, sheets, user):
  59. data = {
  60. "customer": "Tum Coder",
  61. "inspect_date": "2025-01-15",
  62. "lot_no": "12345",
  63. "staff_name": "Tum 8888",
  64. "man_name": "Tum 999",
  65. "size": "Large",
  66. "pcs": "10 pcs",
  67. "spec": "Spec-A",
  68. "hardness_out.d1_act": "10",
  69. "hardness_out.d2_act": "0[24:28]", # Hide rows 24 to 28 if the prefix is "0"
  70. "hardness_out.acc": True, # Hide rows 24 to 28 if the prefix is "0"
  71. "hardness_out.spe_acc": False, # Hide rows 24 to 28 if the prefix is "0"
  72. "hardness_out.qa1": "Hello world",
  73. "hardness_out.qa2": "Unknow person",
  74. "hardness_out.v11": 5,
  75. "hardness_out.v12": 15,
  76. "hardness_out.v13": 10,
  77. "hardness_out.v21": 9,
  78. "hardness_out.v22": 8,
  79. "hardness_out.v23": 7,
  80. "dimension_app.d1_act": "33",
  81. "dimension_app.d2_act": "0[26:32]", # Hide rows 24 to 28 if the prefix is "0"
  82. "dimension_app.acc": True, # Hide rows 24 to 28 if the prefix is "0"
  83. "dimension_app.spe_acc": True, # Hide rows 24 to 28 if the prefix is "0"
  84. "sign1": user.profile.signed_picture,
  85. "sign2": user.profile.signed_picture,
  86. }
  87. output_file = gen_xlsx(
  88. template_file="/app/report/coi_templates.xlsx",
  89. selected_sheets=sheets, # Replace with your actual sheet names
  90. prefix_filename="/app/media/coi",
  91. data=data
  92. )
  93. report = Report.objects.create(
  94. name=lot_no,
  95. created_by=user,
  96. file=None # Leave this as None or assign a file if required
  97. )
  98. output_file_path = Path(output_file) # Convert to a Path object for convenience
  99. with open(output_file_path, "rb") as f:
  100. report.file.save(output_file_path.name, File(f), save=True)
  101. pprint(f"outputfile = {output_file}")
  102. return report
  103. SHEET_NAMES = {
  104. 'hardness_out': 'Hardness Out',
  105. 'hardness_out_in': 'Hardness Out/In',
  106. 'hardness_both_size': 'Hardness Both Size',
  107. 'dimension': 'Dimension',
  108. 'dimension_app': 'Dimension Appearance',
  109. 'dimension_bal_weight': 'Dimension Balance/Weight',
  110. 'dim_bal_app_hard': 'Dimension Balance/Appearance/Hardness',
  111. 'dim_bal_app_rot_hard': 'Dimension Balance/Appearance/Rotation/Hardness',
  112. 'thickness_8_point': 'Thickness 8 Points',
  113. 'centering': 'Centering',
  114. }
  115. def coi_view(request):
  116. pprint(f"xxxx method = xxx {request.method}")
  117. users = User.objects.all()
  118. if request.method == "POST":
  119. pprint(request.POST)
  120. exports = request.POST.getlist("exports") # Retrieve the list of selected values
  121. pprint(f"Selected Export Options: {exports}")
  122. if 'export' in request.POST:
  123. data = {
  124. "customer": "Tum Coder",
  125. "inspect_date": "2025-01-15",
  126. "lot_no": "12345",
  127. "staff_name": "Tum 8888",
  128. "man_name": "Tum 999",
  129. "size": "Large",
  130. "lot_size": "10 pcs",
  131. "spec": "Spec-A",
  132. "hardness_out.d1_act": "10",
  133. "hardness_out.d2_act": "0[24:28]", # Hide rows 24 to 28 if the prefix is "0"
  134. "hardness_out.acc": True, # Hide rows 24 to 28 if the prefix is "0"
  135. "hardness_out.spe_acc": False, # Hide rows 24 to 28 if the prefix is "0"
  136. "dimension_app.d1_act": "33",
  137. "dimension_app.d2_act": "0[26:32]", # Hide rows 24 to 28 if the prefix is "0"
  138. "dimension_app.acc": True, # Hide rows 24 to 28 if the prefix is "0"
  139. "dimension_app.spe_acc": True, # Hide rows 24 to 28 if the prefix is "0"
  140. }
  141. output_file = gen_xlsx(
  142. template_file="/app/report/coi_templates.xlsx",
  143. selected_sheets=exports, # Replace with your actual sheet names
  144. prefix_filename="/app/media/coi",
  145. data=data
  146. )
  147. report = Report.objects.create(
  148. name=request.POST.get('lot_no','Untitled'),
  149. created_by=request.user,
  150. file=None # Leave this as None or assign a file if required
  151. )
  152. output_file_path = Path(output_file) # Convert to a Path object for convenience
  153. with open(output_file_path, "rb") as f:
  154. report.file.save(output_file_path.name, File(f), save=True)
  155. pprint(f"outputfile = {output_file}")
  156. if 'search_lot' in request.POST:
  157. lot_no = request.POST.get('lot_no', None)
  158. if lot_no:
  159. results = queryFromMaster(lot_no)
  160. first_result = results[0] if results else None
  161. try:
  162. pcs = int(first_result.PRO5) - int(first_result.PRO27)
  163. except:
  164. pcs = 0
  165. size_str = f"{first_result.PRO10}x{first_result.PRO11}x{first_result.PRO12}";
  166. spec = f"{first_result.PRO13} {first_result.PRO14} {first_result.PRO15} {first_result.PRO16} {first_result.PRO17} {first_result.PRO18}"
  167. return render(request, 'report/coi.html', {'result': first_result,
  168. 'pcs':pcs,
  169. 'size_str': size_str,
  170. 'spec': spec})
  171. messages.success(request, "Request Sent")
  172. return redirect(request.path_info)
  173. return render(request, 'report/coi.html', {'SHEET_NAMES': SHEET_NAMES, 'users': users})
  174. @csrf_exempt # Disable CSRF for API requests (ensure this is secure in production)
  175. @login_required
  176. def gen_report_view(request):
  177. if request.method == "POST":
  178. try:
  179. # Parse JSON data from the request body
  180. data = json.loads(request.body)
  181. lot_no = data.get("lot_no")
  182. exports = data.get("exports")
  183. if not lot_no:
  184. return HttpResponseBadRequest("Missing 'lot_no' in request data")
  185. # Call the `create_coi_file` function with the provided lot_no
  186. report = create_coi_file(lot_no, exports, request.user)
  187. # Return a success response with the report details
  188. return JsonResponse({
  189. "message": "Report generated successfully",
  190. "report_id": report.id,
  191. "file_url": report.file.url if report.file else None,
  192. })
  193. except json.JSONDecodeError:
  194. return HttpResponseBadRequest("Invalid JSON data")
  195. except Exception as e:
  196. pprint(e)
  197. return JsonResponse({"error": str(e)}, status=500)
  198. else:
  199. return HttpResponseBadRequest("Only POST requests are allowed")