RasadDam_Backend/apps/warehouse/services/excel/excel_processing.py

from io import BytesIO

from django.http import HttpResponse
from openpyxl import Workbook
from openpyxl.styles import Font
from rest_framework import viewsets
from rest_framework.decorators import action

from apps.core.mixins.search_mixin import ExcelDynamicSearchMixin
from apps.warehouse import models as warehouse_models
from apps.warehouse.web.api.v1 import serializers as warehouse_serializers
from apps.authentication.services.visibility_services import apply_visibility_filter
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

TRANSACTION_STATUS_MAP = {
    'success': 'موفق',
    'waiting': 'در انتظار',
    'failed': 'ناموفق',
}


def process_shares(shares):
    """
    پردازش سهم‌ها: سهم "حساب اصلی" رو با سهم‌هایی که shaba یکسان دارن ادغام می‌کنه
    """
    if not shares or len(shares) == 0:
        return []

    # کپی از سهم‌ها
    processed_shares = [dict(share) for share in shares]

    # پیدا کردن سهم‌های حساب اصلی
    main_account_shares = [
        share for share in processed_shares
        if share.get('name') == 'حساب اصلی'
    ]

    for main_share in main_account_shares:
        # پیدا کردن سهم متناظر با همون shaba ولی اسم متفاوت
        matching_share = next(
            (share for share in processed_shares
             if share.get('name') != 'حساب اصلی'
             and share.get('shaba') == main_share.get('shaba')
             and share.get('shaba')),
            None
        )

        if matching_share:
            matching_share['price'] = (matching_share.get('price') or 0) + (main_share.get('price') or 0)

    # برگردوندن سهم‌ها بدون حساب اصلی
    return [share for share in processed_shares if share.get('name') != 'حساب اصلی']


def calculate_share_totals(items):
    """
    محاسبه جمع کل سهم‌ها از تمام آیتم‌ها
    """
    share_totals = {}

    for item in items:
        item_share = item.get('item_share', [])
        if item_share and len(item_share) > 0:
            processed_shares = process_shares(item_share)
            for share in processed_shares:
                key = share.get('shaba') or share.get('name') or 'unknown'
                if key not in share_totals:
                    share_totals[key] = {
                        'name': share.get('name') or '-',
                        'shaba': share.get('shaba') or '-',
                        'total': 0,
                    }
                share_totals[key]['total'] += share.get('price') or 0

    return list(share_totals.values())


class WareHouseExcelViewSet(viewsets.ModelViewSet, ExcelDynamicSearchMixin):
    queryset = warehouse_models.InventoryEntry.objects.all()
    serializer_class = warehouse_serializers.InventoryEntrySerializer
    search_fields = [
        "distribution__distribution_id",
        "organization__name",
        "weight",
        "balance",
        "lading_number",
        "is_confirmed",
    ]
    date_field = "create_date"

    # 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)
        queryset = self.filter_query(self.queryset)

        entries = queryset.filter(organization=get_organization_by_user(request.user))
        ser_data = self.serializer_class(entries, many=True).data

        excel_options = [
            "ردیف",
            "تاریخ ورود به انبار",
            "شماره سهمیه",
            "وزن",
            "بارنامه",
            "محل دریافت",
            "سند",
            "توضیحات",
        ]

        header_list = [
            "وزن",

        ]

        create_header(worksheet, header_list, 5, 2, height=25, border_style='thin')
        excel_description(worksheet, 'B1', f'ورودی به انبار', row2='C3')
        create_header_freez(worksheet, excel_options, 1, 6, 7, height=25, width=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}", "دانلود")'

                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, height=23, m=m)
                if document:
                    worksheet.cell(row=l + 1, column=7).font = Font(color="0563C1", underline='single', bold=True)
                m += 1
                l += 1

        weight = sum((data['weight'] or 0) for data in ser_data)

        value_list = [
            weight
        ]

        create_value(worksheet, value_list, 3, 5, border_style='thin')

        list2 = [
            'مجموع==>',
            '',
            '',
            weight,
            '',
            '',
            '',
            ''
        ]
        create_value(worksheet, list2, l + 3, 1, color='gray', height=23)
        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

    # noqa # اکسل تراکنش‌ها
    @action(
        methods=['get'],
        detail=False,
        url_path='inventory_sale_transaction_excel',
        url_name='inventory_sale_transaction_excel',
        name='inventory_sale_transaction_excel'
    )
    def inventory_sale_transaction_excel(self, request):
        self.search_fields = [
            'rancher_fullname', 'rancher_mobile', 'pos_device__device_identity',
            'pos_device__acceptor', 'pos_device__terminal', 'pos_device__serial',
            'transaction_id', 'seller_organization__name',
            'quota_distribution__distribution_id', 'weight', 'delivery_address', 'transaction_price',
            'price_paid', 'price_type', 'product_type', 'transactions_number', 'transaction_status',
            'transaction_status_code', 'ref_num', 'terminal', 'payer_cart', 'transaction_date',
        ]
        output = BytesIO()
        workbook = Workbook()
        worksheet = workbook.active
        worksheet.sheet_view.rightToLeft = True
        worksheet.insert_rows(1)

        org = get_organization_by_user(request.user)
        queryset = warehouse_models.InventoryQuotaSaleTransaction.objects.all()
        queryset = apply_visibility_filter(queryset, org)

        if 'status' in request.GET.keys():
            status_param = self.request.query_params.get('status')  # noqa

            if status_param == 'waiting':
                queryset = queryset.filter(transaction_status='waiting').order_by('-create_date')
            elif status_param == 'success':
                queryset = queryset.filter(transaction_status='success').order_by('-create_date')
            elif status_param == 'failed':
                queryset = queryset.filter(transaction_status='failed').order_by('-create_date')
            else:
                queryset = queryset.order_by('-create_date')
        else:
            queryset = queryset.order_by('-create_date')

        queryset = self.filter_query(queryset)

        ser_data = warehouse_serializers.InventoryQuotaSaleTransactionSerializer(queryset, many=True).data

        # جمع‌آوری تمام آیتم‌ها برای پیدا کردن سهم‌های یونیک
        all_items = []
        for data in ser_data:
            all_items.extend(data.get('items', []))

        # محاسبه سهم‌های یونیک (برای ستون‌های داینامیک)
        all_share_totals = calculate_share_totals(all_items)
        share_names = [share['name'] for share in all_share_totals]

        excel_options = [
            "ردیف",
            "تعاونی دامدار",
            "کد ملی دامدار",
            "تاریخ",
            "محصولات",
            "شناسه تراکنش",
            "شماره کارت",
            "مبلغ",
            "وضعیت",
        ]
        # اضافه کردن ستون‌های داینامیک سهم‌ها
        excel_options.extend(share_names)

        header_list = [
            "مبلغ کل",
            "تعداد تراکنش‌ها",
        ]
        # اضافه کردن سهم‌ها به هدر
        header_list.extend(share_names)

        # محاسبه height داینامیک بر اساس تعداد آیتم‌ها
        header_height = max(25, 15 + len(header_list) * 3)
        options_height = max(25, 15 + len(excel_options) * 2)

        create_header(worksheet, header_list, 5, 2, height=header_height, border_style='thin')

        # ساخت عنوان با بازه تاریخ
        start_date = request.query_params.get('start')
        end_date = request.query_params.get('end')

        title = 'تراکنش‌ها'
        if start_date and end_date:
            start_shamsi = shamsi_date(convert_str_to_date(start_date))
            end_shamsi = shamsi_date(convert_str_to_date(end_date))
            title = f'تراکنش‌ها از {start_shamsi} تا {end_shamsi}'
        elif start_date:
            start_shamsi = shamsi_date(convert_str_to_date(start_date))
            title = f'تراکنش‌ها از {start_shamsi}'
        elif end_date:
            end_shamsi = shamsi_date(convert_str_to_date(end_date))
            title = f'تراکنش‌ها تا {end_shamsi}'

        excel_description(worksheet, 'B1', title, row2='C3')
        create_header_freez(worksheet, excel_options, 1, 6, 7, height=options_height, width=20)

        l = 6
        m = 1
        # دیکشنری برای نگهداری جمع سهم‌ها
        share_column_totals = {name: 0 for name in share_names}

        if ser_data:
            for data in ser_data:
                items = data.get('items', [])
                products_list = []
                for item in items:
                    product_name = item.get('name', '')
                    if product_name:
                        products_list.append(product_name)
                products_str = '، '.join(products_list) if products_list else '-'

                rancher_data = data.get('rancher')
                national_code = rancher_data.get('national_code', '-') if rancher_data else '-'

                seller_org = data.get('seller_organization')
                org_name = seller_org.get('name', '-') if seller_org else '-'

                status = TRANSACTION_STATUS_MAP.get(data.get('transaction_status'), '-')

                # محاسبه سهم‌های این تراکنش
                transaction_shares = calculate_share_totals(items)
                share_values = []
                for share_name in share_names:
                    share_value = next(
                        (s['total'] for s in transaction_shares if s['name'] == share_name),
                        0
                    )
                    share_values.append(share_value)
                    share_column_totals[share_name] += share_value

                list1 = [
                    m,
                    org_name,
                    national_code,
                    str(shamsi_date(convert_str_to_date(data['transaction_date']), in_value=True)) if data.get(
                        'transaction_date') else '',
                    products_str,
                    data.get('transaction_id') or '-',
                    data.get('payer_cart') or '-',
                    data.get('price_paid') or 0,
                    status,
                ]
                # اضافه کردن مقادیر سهم‌ها
                list1.extend(share_values)

                create_value(worksheet, list1, l + 1, 1, height=23, m=m)
                m += 1
                l += 1

        total_price = sum((data['price_paid'] or 0) for data in ser_data)
        transaction_count = len(ser_data)

        value_list = [
            total_price,
            transaction_count,
        ]
        # اضافه کردن جمع سهم‌ها به مقادیر هدر
        value_list.extend([share_column_totals[name] for name in share_names])

        create_value(worksheet, value_list, 3, 5, border_style='thin')

        list2 = [
            'مجموع==>',
            '',
            '',
            '',
            '',
            '',
            '',
            total_price,
            '',
        ]
        # اضافه کردن جمع سهم‌ها به ردیف مجموع
        list2.extend([share_column_totals[name] for name in share_names])

        create_value(worksheet, list2, l + 3, 1, color='gray', height=23)
        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