MNPC/RasadDam_Backend
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Mojtaba-z
2025-08-02 10:17:37 +03:30
parent 817afb1be2 ca5f050179
commit 62b7098486
9 changed files with 165 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
apps/product/services/excel/excel_processing.py
View File

@@ -4,9 +4,10 @@ from io import BytesIO
from django.db.models import Q
from django.http import HttpResponse
from openpyxl import Workbook
from rest_framework import viewsets
from rest_framework import viewsets, filters
from rest_framework.decorators import action
from apps.core.mixins.search_mixin import DynamicSearchMixin
from apps.product import models as product_models
from apps.product.web.api.v1.serializers import quota_distribution_serializers as distribution_serializers
from apps.product.web.api.v1.serializers.product_serializers import IncentivePlanSerializer
@@ -15,9 +16,10 @@ from common.helper_excel import create_header, excel_description, create_header_
from common.helpers import get_organization_by_user
class ProductExcelViewSet(viewsets.ModelViewSet):
class ProductExcelViewSet(viewsets.ModelViewSet, DynamicSearchMixin):
queryset = product_models.QuotaDistribution.objects.all()
serializer_class = distribution_serializers.QuotaDistributionSerializer
filter_backends = [filters.SearchFilter]
# noqa # سهمیه و توزیع
@action(
@@ -34,22 +36,23 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
worksheet.sheet_view.rightToLeft = True
worksheet.insert_rows(1)
product = self.queryset
query = request.query_params
product = self.filter_query(self.queryset) # return by search param or all objects
organization = get_organization_by_user(request.user)
query = self.request.query_params
description_name = ''
if query.get('param') == 'assigned':
product = product.filter(
Q(assigned_organization=organization)
).order_by('-modify_date')
description_name = 'سهمیه'
description_name = 'سهمیه' # noqa
elif query.get('param') == 'assigner':
product = product.filter(
Q(assigner_organization=organization)
).order_by('-modify_date')
description_name = 'توزیع'
description_name = 'توزیع' # noqa
elif query.get('param') == 'all':
product = product.filter(
@@ -86,11 +89,11 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
]
create_header(worksheet, header_list, 5, 2, height=20, border_style='thin')
create_header(worksheet, header_list, 5, 2, height=25, border_style='thin')
excel_description(worksheet, 'B1', f'{description_name}', row2='C3')
create_header_freez(worksheet, excel_options, 1, 6, 7, height=22, width=20)
create_header_freez(worksheet, excel_options, 1, 6, 7, height=25, width=20)
l = 6
m = 1
@@ -102,10 +105,10 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
str(data['quota']['quota_id']) if data.get('quota') and data['quota'].get('quota_id') else '',
str(shamsi_date(convert_str_to_date(data['create_date']), in_value=True)) if data.get(
'create_date') else '',
data[
'assigner_organization'].get('organization') or '-',
data[
'assigned_organization'].get('organization') or '-',
(data[
'assigner_organization'] or {}).get('organization') or '-',
(data[
'assigned_organization'] or {}).get('organization') or '-',
data.get('weight') or 0,
data.get('distributed') or 0,
data.get('remaining_weight') or 0,
@@ -114,7 +117,7 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
data.get('warehouse_entry') or 0,
data.get('description') or '-'
]
create_value(worksheet, list1, l + 1, 1, m=m)
create_value(worksheet, list1, l + 1, 1, height=24, m=m)
m += 1
l += 1
@@ -152,19 +155,18 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
'',
]
create_value(worksheet, list2, l + 3, 1, color='gray')
create_value(worksheet, list2, l + 3, 1, color='gray') # noqa
workbook.save(output)
output.seek(0)
response = HttpResponse(
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet') # noqa
response[
'Content-Disposition'] = f'attachment; filename="سهمیه.xlsx"'.encode(
'Content-Disposition'] = f'attachment; filename="سهمیه.xlsx"'.encode( # noqa
'utf-8')
response.write(output.getvalue())
return response \
return response
# noqa # طرح های تشویقی
@action(
@@ -204,7 +206,7 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
excel_description(worksheet, 'B1', f'طرح های تشویقی', row2='C3')
create_header_freez(worksheet, excel_options, 1, 6, 7, height=22, width=20)
create_header_freez(worksheet, excel_options, 1, 6, 7, height=25, width=20)
l = 6
m = 1
@@ -226,7 +228,7 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
str(shamsi_date(convert_str_to_date(data.get('start_date_limit')), in_value=True)),
str(shamsi_date(convert_str_to_date(data.get('end_date_limit')), in_value=True)),
]
create_value(worksheet, list1, l + 1, 1, m=m)
create_value(worksheet, list1, l + 1, 1, height=24, m=m)
m += 1
l += 1
@@ -242,7 +244,7 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
'',
]
create_value(worksheet, list2, l + 3, 1, color='gray')
create_value(worksheet, list2, l + 3, 1, height=24, color='gray') # noqa
workbook.save(output)
output.seek(0)
@@ -250,7 +252,7 @@ class ProductExcelViewSet(viewsets.ModelViewSet):
response = HttpResponse(
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response[
'Content-Disposition'] = f'attachment; filename="طرح های تشویقی.xlsx"'.encode(
'Content-Disposition'] = f'attachment; filename="طرح های تشویقی.xlsx"'.encode( # noqa
'utf-8')
response.write(output.getvalue())
return response