Merge branch 'main' of https://github.com/MNPC-IR/Rasaddam_Backend

apps/warehouse/services/excel/excel_processing.py

@@ -1,85 +1,118 @@
 from io import BytesIO
 
 from django.http import HttpResponse
-from django.views.decorators.csrf import csrf_exempt
 from openpyxl import Workbook
-from openpyxl.styles import Alignment
+from openpyxl.styles import Font
 from rest_framework import viewsets
-from rest_framework.decorators import action, api_view, permission_classes
-from rest_framework.permissions import AllowAny , IsAuthenticated
+from rest_framework.decorators import action
 
 from apps.warehouse import models as warehouse_models
 from apps.warehouse.web.api.v1 import serializers as warehouse_serializers
-from common.helper_excel import create_header, excel_description, create_header_freez, create_value
+from common.helper_excel import create_header, excel_description, create_header_freez, create_value, shamsi_date, \
+    convert_str_to_date
+from common.helpers import get_organization_by_user
 
 
-@api_view(["GET"])
-@permission_classes([IsAuthenticated])
-@csrf_exempt
-def warehouse_excel(request):
+class WareHouseExcelViewSet(viewsets.ModelViewSet):
+    queryset = warehouse_models.InventoryEntry.objects.all()
+    serializer_class = warehouse_serializers.InventoryEntrySerializer
 
-    excel_options = [
-        'ردیف',
-        'وضعیت',
-        'ثبت کننده',
-        'تاریخ ثبت',
-        'مرغدار',
-        'کشتارگاه',
-        'حجم',
-        'حجم کم شده از سالن مرغدار',
-        'وزن تقریبی کشتار(کیلوگرم)',
-        'حجم سفارشات دریافتی توسط کشتارگاه',
-        'اختلاف کشتار(حجم)',
+    # noqa # ورودی به انبار
+    @action(
+        methods=['get'],
+        detail=False,
+        url_path='warehouse_excel',
+        url_name='warehouse_excel',
+        name='warehouse_excel'
+    )
+    def warehouse_excel(self, request):
+        output = BytesIO()
+        workbook = Workbook()
+        worksheet = workbook.active
+        worksheet.sheet_view.rightToLeft = True
+        worksheet.insert_rows(1)
 
-    ]
+        entries = self.queryset.filter(organization=get_organization_by_user(request.user))
+        ser_data = self.serializer_class(entries, many=True).data
 
+        excel_options = [
+            "ردیف",
+            "تاریخ ورود به انبار",
+            "شماره سهمیه",
+            "وزن",
+            "بارنامه",
+            "محل دریافت",
+            "سند",
+            "توضیحات",
+        ]
 
-    output = BytesIO()
-    workbook = Workbook()
-    worksheet = workbook.active
-    worksheet.sheet_view.rightToLeft = True
-    worksheet.insert_rows(1)
-    cell = worksheet.cell(row=1, column=1)
-    cell.alignment = Alignment(horizontal='center', vertical='center')
+        header_list = [
+            "وزن",
 
-    header_list = [
-        'حجم',
-        'حجم کم شده از سالن مرغدار',
-        'وزن تقریبی کشتار(کیلوگرم)',
-        'حجم سفارشات دریافتی توسط کشتارگاه',
-        'اختلاف کشتار(حجم)',
+        ]
 
-    ]
+        create_header(worksheet, header_list, 5, 2, height=20, border_style='thin')
+        excel_description(worksheet, 'B1', f'ورودی به انبار', row2='C3')
+        create_header_freez(worksheet, excel_options, 1, 6, 7, height=22, width=20)
 
-    create_header(worksheet, header_list, 5, 2, height=20)
+        l = 6
+        m = 1
+        if ser_data:
+            for data in ser_data:
+                document = data.get('document')
+                if document:
+                    if str(document).startswith(('http://', 'https://')):
+                        document_value = f'=HYPERLINK("{document}", "دانلود")'
+                    else:
+                        full_path = f"https://yourdomain.com/{document}"
+                        document_value = f'=HYPERLINK("{full_path}", "دانلود")'
 
-    excel_description(worksheet, 'B1', f'اختلاف کشتار ', color='red', row2='C3')
+                else:
+                    document_value = 'ندارد'
+                list1 = [
+                    m,
+                    str(shamsi_date(convert_str_to_date(data['create_date']), in_value=True)) if data.get(
+                        'create_date') else '',
+                    str(data[
+                        'distribution'].get('distribution')) or '-',
+                    data.get('weight') or 0,
+                    data.get('lading_number') or '-',
+                    data.get('delivery_address') or '-',
+                    document_value,
+                    data.get('notes') or '',
+                ]
+                create_value(worksheet, list1, l + 1, 1, m=m)
+                if document:
+                    worksheet.cell(row=l + 1, column=7).font = Font(color="0563C1", underline='single', bold=True)
+                m += 1
+                l += 1
 
-    create_header_freez(worksheet, excel_options, 1, 6, 7, height=22)
+        weight = sum((data['weight'] or 0) for data in ser_data)
 
-    l = 5
-    m = 1
+        value_list = [
+            weight
+        ]
 
+        create_value(worksheet, value_list, 3, 5, border_style='thin')
 
+        list2 = [
+            'مجموع==>',
+            '',
+            '',
+            weight,
+            '',
+            '',
+            '',
+            ''
+        ]
+        create_value(worksheet, list2, l + 3, 1, color='gray')
+        workbook.save(output)
+        output.seek(0)
 
-    list2 = [
-        'مجموع==>',
-        '',
-        '',
-        '',
-        '',
-        '',
-
-
-    ]
-    create_value(worksheet, list2, l + 3, 1, color='green')
-    workbook.save(output)
-    output.seek(0)
-
-    response = HttpResponse(
-        content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
-    response[
-        'Content-Disposition'] = f'attachment; filename="اختلاف کشتار.xlsx"'.encode(
-        'utf-8')
-    response.write(output.getvalue())
-    return response
+        response = HttpResponse(
+            content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+        response[
+            'Content-Disposition'] = f'attachment; filename="ورودی به انبار.xlsx"'.encode(
+            'utf-8')
+        response.write(output.getvalue())
+        return response

apps/warehouse/services/excel/urls.py

@@ -1,11 +1,10 @@
 from django.urls import path, include
 from rest_framework.routers import DefaultRouter
-from apps.warehouse.services.excel import excel_processing as excel
-from apps.warehouse.services.excel.excel_processing import warehouse_excel
+from apps.warehouse.services.excel.excel_processing import WareHouseExcelViewSet
 
 router = DefaultRouter()
+router.register(r'', WareHouseExcelViewSet, basename='warehouse_excel')
 
 urlpatterns = [
     path('', include(router.urls)),
-    path('warehouse_excel/', warehouse_excel),
 ]