Files
Rasadyar_Kurdestan/panel/helper_excel.py
2026-01-18 11:59:48 +03:30

539 lines
24 KiB
Python

from datetime import datetime
from io import BytesIO
import jdatetime
import openpyxl
from django.db.models import Q, Sum
from django.http import HttpResponse
from openpyxl import Workbook
from openpyxl.chart import LineChart, Reference, BarChart
from openpyxl.styles import PatternFill, Alignment, Font
from openpyxl.utils import get_column_letter
from deposit_id import wage_counting_type
from panel.models import WageType, ProvinceKillRequest, KillHouseRequest, KillHouseFreeBarInformation, \
KillHouseFreeSaleBarInformation, InternalTransaction, KillHouse, StewardAllocation
blue_fill = PatternFill(start_color="277358", fill_type="solid")
Alignment_CELL = Alignment(horizontal='center', vertical='center', wrap_text=True)
red_font = Font(color="C00000", bold=True)
GREEN_CELL = PatternFill(start_color="00B050", fill_type="solid")
RED_CELL = PatternFill(start_color="FCDFDC", fill_type="solid")
YELLOW_CELL = PatternFill(start_color="FFFF00", fill_type="solid")
ORANGE_CELL = PatternFill(start_color="FFC000", fill_type="solid")
BLUE_CELL = PatternFill(start_color="538DD5", fill_type="solid")
LIGHT_GREEN_CELL = PatternFill(start_color="92D050", fill_type="solid")
VERY_LIGHT_GREEN_CELL = PatternFill(start_color="5AFC56", fill_type="solid")
percent_of_losses = 0.11
def shamsi_date(date, in_value=None, set_datetime=None):
if set_datetime:
sh_dt = jdatetime.datetime.fromgregorian(datetime=date)
return sh_dt.strftime('%Y/%m/%d (%H:%M)')
if in_value:
sh_date = jdatetime.date.fromgregorian(
year=date.year,
month=date.month,
day=date.day
)
else:
gh_date = jdatetime.date.fromgregorian(
year=date.year,
month=date.month,
day=date.day
).strftime('%Y-%m-%d')
reversed_date = reversed(gh_date.split("-"))
separate = "-"
sh_date = separate.join(reversed_date)
return sh_date
def create_header(worksheet, list, num, row, height=None, width=None, color=None, text_color=None, border_style=None):
for col_num, option in enumerate(list, num):
cell = worksheet.cell(row=row, column=col_num, value=option)
col_letter = get_column_letter(col_num)
cell.alignment = Alignment_CELL
if color is not None:
if color == 'green':
cell.fill = GREEN_CELL
elif color == 'orange':
cell.fill = ORANGE_CELL
elif color == 'blue':
cell.fill = BLUE_CELL
else:
cell.fill = PatternFill(start_color=color, fill_type="solid")
else:
cell.fill = blue_fill
if text_color is not None:
cell.font = Font(size=9, bold=True, color=text_color)
else:
cell.font = Font(size=9, bold=True, color='D9FFFFFF')
if height is not None:
worksheet.row_dimensions[row].height = height
if width is not None:
worksheet.column_dimensions[col_letter].width = width
if border_style is not None:
cell.border = openpyxl.styles.Border(
left=openpyxl.styles.Side(style=border_style),
right=openpyxl.styles.Side(style=border_style),
top=openpyxl.styles.Side(style=border_style),
bottom=openpyxl.styles.Side(style=border_style)
)
def create_header_freez(worksheet, list, num, row, header_row, height=None, width=None, len_with=None,
different_cell=None):
for col_num, option in enumerate(list, num):
col_letter = get_column_letter(col_num)
cell = worksheet.cell(row=row, column=col_num, value=option)
cell.alignment = Alignment_CELL
cell.fill = blue_fill
cell.font = Font(size=10, bold=True, color='D9FFFFFF')
if height is not None:
worksheet.row_dimensions[row].height = height
if len(option) > worksheet.column_dimensions[col_letter].width:
worksheet.column_dimensions[col_letter].width = len(option) + 2
if width is not None:
worksheet.column_dimensions[col_letter].width = width
if len_with is not None:
if len(option) > worksheet.column_dimensions[col_letter].width:
worksheet.column_dimensions[col_letter].width = len(option) + 3
if different_cell is not None:
if option == different_cell:
cell.fill = PatternFill(start_color="C00000", fill_type="solid")
worksheet.freeze_panes = worksheet[f'A{header_row}']
max_col = worksheet.max_column
range_str = f'A{header_row - 1}:{get_column_letter(max_col)}{worksheet.max_row}'
worksheet.auto_filter.ref = range_str
def excel_description(worksheet, row1, description, size=None, color=None, my_color=None, row2=None):
worksheet[row1] = description
worksheet[row1].alignment = Alignment_CELL
if size is not None:
worksheet[row1].font = Font(size=size)
if color is not None:
worksheet[row1].font = red_font
if my_color is not None:
worksheet[row1].font = PatternFill(start_color=my_color, fill_type="solid")
if row2 is not None:
merge_range = f'{row1}:{row2}'
worksheet.merge_cells(merge_range)
def create_value(worksheet, list, l, num, border_style=None, m=None, height=None, color=None, width=None,
different_cell=None, different_value=None, item_num=None, item_color=None):
color_dict = {
'green': GREEN_CELL,
'yellow': YELLOW_CELL,
'blue': BLUE_CELL,
'red': RED_CELL,
'light_green': LIGHT_GREEN_CELL,
'very_light_green': VERY_LIGHT_GREEN_CELL
}
for item in range(len(list)):
cell = worksheet.cell(row=l, column=item + num, value=list[item])
cell.alignment = Alignment_CELL
if border_style:
cell.border = openpyxl.styles.Border(
left=openpyxl.styles.Side(style=border_style),
right=openpyxl.styles.Side(style=border_style),
top=openpyxl.styles.Side(style=border_style),
bottom=openpyxl.styles.Side(style=border_style)
)
value = list[item]
if isinstance(value, (int, float)) and value != 0:
cell.number_format = '#,###'
else:
cell.value = value
cell.font = Font(size=10, bold=True)
if m is not None and m % 2 != 0:
cell.fill = PatternFill(start_color="D6F6FE", fill_type="solid")
if height is not None:
worksheet.row_dimensions[l + 1].height = height
if item_num is not None and item == item_num:
if item_color:
cell.fill = item_color
elif color in color_dict:
cell.fill = color_dict[color]
if different_cell is not None and list[different_cell] == different_value:
cell.fill = RED_CELL
if width is not None:
worksheet.column_dimensions[openpyxl.utils.get_column_letter(item + num)].width = width
def merge_cells(worksheet, l, s, cell1=None, cell2=None, lst=None):
if lst is not None:
for col in lst:
rng = f'{col}{l}:{col}{l + s}'
worksheet.merge_cells(rng)
worksheet[col + f'{l}'].alignment = Alignment_CELL
else:
for col in range(ord(f'{cell1}'), ord(f'{cell2}') + 1):
rng = f'{chr(col)}{l}:{chr(col)}{l + s}'
worksheet.merge_cells(rng)
worksheet[chr(col) + f'{l}'].alignment = Alignment_CELL
def add_header(worksheet):
worksheet.oddHeader.center.text = "سامانه رصدیار"
worksheet.oddHeader.center.size = 14 # تنظیم اندازه فونت
worksheet.oddHeader.center.font = "Arial,Bold" # تنظیم فونت و ضخامت
# همچنین می‌توانید از هدرهای چپ و راست هم استفاده کنید
# worksheet.oddHeader.right.text = f"تاریخ: {shamsi_now_date}"
def cell_color_changer(worksheet, row, start_index, end_index, custom_color):
for item in range(start_index, end_index):
cell = worksheet.cell(row=row, column=item)
cell.fill = PatternFill(start_color=custom_color, fill_type="solid")
def get_kill_house_finance_info_by_date(kill_house, date1, date2):
total_wage_type = WageType.objects.filter(trash=False)
province_live_wage_type = total_wage_type.filter(en_name='province-kill-request', trash=False).first()
free_buying_live_weight_wage_type = total_wage_type.filter(en_name='live-buy', trash=False).first()
free_buying_carcesses_weight_wage_type = total_wage_type.filter(en_name='carcasse-buy', trash=False).first()
free_sell_carcesses_weight_wage_type = total_wage_type.filter(en_name='carcasse-sell', trash=False).first()
province_live_wage_type_amount = province_live_wage_type.amount if province_live_wage_type.status == True else 0
free_buying_live_weight_wage_type_amount = free_buying_live_weight_wage_type.amount if free_buying_live_weight_wage_type.status == True else 0
free_buying_carcesses_weight_wage_type_amount = free_buying_carcesses_weight_wage_type.amount if free_buying_carcesses_weight_wage_type.status == True else 0
free_sell_carcesses_weight_wage_type_amount = free_sell_carcesses_weight_wage_type.amount if free_sell_carcesses_weight_wage_type.status == True else 0
province_kill_requests = ProvinceKillRequest.objects.filter(killhouse_user=kill_house, trash=False,
archive_wage=False, return_to_province=False,
state__in=('pending', 'accepted'),
first_car_allocated_quantity=0,
kill_request__recive_date__date__gte=date1,
kill_request__recive_date__date__lte=date2).order_by(
'id')
kill_house_requests = KillHouseRequest.objects.filter(
Q(Q(killhouse_user=kill_house) & Q(killer=kill_house)) | Q(
Q(killhouse_user=kill_house) & Q(killer__isnull=True)) | Q(
Q(killhouse_user=kill_house) | Q(killer=kill_house)), archive_wage=False,
trash=False, calculate_status=True, kill_request__recive_date__date__gte=date1,
kill_request__recive_date__date__lte=date2
)
kill_house_requests = kill_house_requests.filter(
Q(killhouse_user=kill_house, killer__isnull=True) | Q(killhouse_user=kill_house, killer=kill_house) | Q(
killer=kill_house))
kill_house_free_bar_info = KillHouseFreeBarInformation.objects.filter(
Q(kill_house=kill_house, exclusive_killer__isnull=True) | Q(exclusive_killer=kill_house), date__date__gte=date1,
date__date__lte=date2,
archive_wage=False,
calculate_status=True,
trash=False)
kill_house_free_sale_bar_info = KillHouseFreeSaleBarInformation.objects.filter(date__date__gte=date1,
date__date__lte=date2,
kill_house=kill_house,
archive_wage=False,
calculate_status=True,
trash=False)
slaughter_transactions = InternalTransaction.objects.filter(date__date__gte=date1,
date__date__lte=date2,
kill_house=kill_house, status='completed',
trash=False)
total_paid_wage = slaughter_transactions.aggregate(total=Sum('amount'))[
'total'] or 0
total_province_live_weight = province_kill_requests.aggregate(total=Sum('total_killed_weight'))['total'] or 0
total_province_live_weight += kill_house_requests.aggregate(total=Sum('accepted_real_weight'))['total'] or 0
total_out_selling_province_carcasses_weight = \
kill_house_free_sale_bar_info.aggregate(total=Sum('real_weight_of_carcasses'))['total'] or 0
if wage_counting_type == 'live':
total_province_carcasses_weight = total_province_live_weight
total_pure_province_carcasses_weight = total_province_carcasses_weight
else:
total_province_carcasses_weight = total_province_live_weight * 0.75
total_out_carcasses_buying_for_pure_province_carcasses_weight = \
kill_house_free_bar_info.aggregate(total=Sum('weight_of_carcasses'))['total'] or 0
out_selling_out_carcasses_buying_difference = total_out_selling_province_carcasses_weight - total_out_carcasses_buying_for_pure_province_carcasses_weight if (
total_out_selling_province_carcasses_weight - total_out_carcasses_buying_for_pure_province_carcasses_weight) > 0 else 0
total_pure_province_carcasses_weight = total_province_carcasses_weight - out_selling_out_carcasses_buying_difference
total_out_live_buying_province_carcasses_weight = \
kill_house_free_bar_info.filter(buy_type='live').aggregate(total=Sum('live_weight'))['total'] or 0
total_out_carcasses_buying_province_carcasses_weight = \
kill_house_free_bar_info.filter(buy_type='carcass').aggregate(total=Sum('weight_of_carcasses'))['total'] or 0
total_pure_province_carcasses_price = total_pure_province_carcasses_weight * province_live_wage_type_amount
total_out_selling_province_carcasses_price = total_out_selling_province_carcasses_weight * free_sell_carcesses_weight_wage_type_amount
total_out_live_buying_province_carcasses_price = total_out_live_buying_province_carcasses_weight * free_buying_live_weight_wage_type_amount
total_out_carcasses_buying_province_carcasses_price = total_out_carcasses_buying_province_carcasses_weight * free_buying_carcesses_weight_wage_type_amount
total_price = total_pure_province_carcasses_price + total_out_selling_province_carcasses_price + total_out_live_buying_province_carcasses_price + total_out_carcasses_buying_province_carcasses_price
return {
"total_province_live_weight": total_province_live_weight,
"total_province_carcasses_weight": total_province_carcasses_weight,
"total_out_selling_province_carcasses_weight": total_out_selling_province_carcasses_weight,
"total_pure_province_carcasses_weight": total_pure_province_carcasses_weight,
"total_pure_province_carcasses_price": total_pure_province_carcasses_price,
"total_out_selling_province_carcasses_price": total_out_selling_province_carcasses_price,
"total_out_carcasses_buying_province_carcasses_weight": total_out_carcasses_buying_province_carcasses_weight,
"total_out_carcasses_buying_province_carcasses_price": total_out_carcasses_buying_province_carcasses_price,
"total_out_live_buying_province_carcasses_weight": total_out_live_buying_province_carcasses_weight,
"total_out_live_buying_province_carcasses_price": total_out_live_buying_province_carcasses_price,
"total_paid_wage": total_paid_wage,
"total_price": total_price,
"province_live_wage_amount": province_live_wage_type_amount,
"free_buying_live_weight_amount": free_buying_live_weight_wage_type_amount,
"free_buying_carcesses_weight_amount": free_buying_carcesses_weight_wage_type_amount,
"free_sell_carcesses_weight_amount": free_sell_carcesses_weight_wage_type_amount,
}
def start_excel():
output = BytesIO()
workbook = Workbook()
worksheet = workbook.active
worksheet.sheet_view.rightToLeft = True
worksheet.insert_rows(1)
return workbook, worksheet, output
def close_excel(name):
workbook, worksheet, output = start_excel()
workbook.save(output)
output.seek(0)
response = HttpResponse(
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
response[
'Content-Disposition'] = f'attachment; filename="{name}.xlsx"'.encode(
'utf-8')
response.write(output.getvalue())
return response
def get_kill_house_distribution_info(city, date1=None, date2=None):
kill_house_name = KillHouse.objects.filter(trash=False, out_province=False).order_by('id')
list1 = [city.name]
for kill_house in kill_house_name:
if date1:
allocations = StewardAllocation.objects.filter(
Q(to_steward__user__city=city) | Q(to_guilds__user__city=city),
allocation_type__in=('killhouse_steward', 'killhouse_guild'),
trash=False, date__date__gte=date1, receiver_state__in=('pending', 'accepted'),
date__date__lte=date2, calculate_status=True, kill_house=kill_house).order_by('id')
else:
allocations = StewardAllocation.objects.filter(
Q(to_steward__user__city=city) | Q(to_guilds__user__city=city),
allocation_type__in=('killhouse_steward', 'killhouse_guild'),
receiver_state__in=('pending', 'accepted'),
trash=False, calculate_status=True, kill_house=kill_house).order_by('id')
weight = allocations.aggregate(
total_quantity=Sum('real_weight_of_carcasses')).get(
'total_quantity') or 0
list1.append(weight
)
return list1
def get_kill_house_distribution_out_province_info(date1=None, date2=None):
kill_house_name = KillHouse.objects.filter(trash=False, out_province=False).order_by('id')
list1 = ['خارج استان']
for kill_house in kill_house_name:
if date1:
kill_house_free_sale_bars = KillHouseFreeSaleBarInformation.objects.filter(
kill_house=kill_house,
trash=False, calculate_status=True, date__date__gte=date1,
date__date__lte=date2)
else:
kill_house_free_sale_bars = KillHouseFreeSaleBarInformation.objects.filter(
kill_house=kill_house,
trash=False, calculate_status=True)
weight_kill_house_free_sale_bars = kill_house_free_sale_bars.aggregate(
total_quantity=Sum('real_weight_of_carcasses')).get(
'total_quantity') or 0
list1.append(weight_kill_house_free_sale_bars
)
if date1:
kill_house_free_sale_bars = KillHouseFreeSaleBarInformation.objects.filter(
trash=False, calculate_status=True, date__date__gte=date1,
date__date__lte=date2)
else:
kill_house_free_sale_bars = KillHouseFreeSaleBarInformation.objects.filter(
trash=False, calculate_status=True)
weight_kill_house_free_sale_bars = kill_house_free_sale_bars.aggregate(
total_quantity=Sum('real_weight_of_carcasses')).get(
'total_quantity') or 0
list1.append(weight_kill_house_free_sale_bars
)
return list1
def all_get_kill_house_distribution_info(city, date1=None, date2=None):
if date1:
allocations = StewardAllocation.objects.filter(
Q(to_steward__user__city=city) | Q(to_guilds__user__city=city),
allocation_type__in=('killhouse_steward', 'killhouse_guild'),
trash=False, date__date__gte=date1, receiver_state__in=('pending', 'accepted'),
date__date__lte=date2, calculate_status=True).order_by('id')
else:
allocations = StewardAllocation.objects.filter(
Q(to_steward__user__city=city) | Q(to_guilds__user__city=city),
allocation_type__in=('killhouse_steward', 'killhouse_guild'),
receiver_state__in=('pending', 'accepted'),
trash=False, calculate_status=True).order_by('id')
weight = allocations.aggregate(
total_quantity=Sum('real_weight_of_carcasses')).get(
'total_quantity') or 0
return weight
def add_chart(
worksheet,
chart_type,
data_columns,
category_column,
start_row,
end_row,
chart_position,
chart_title,
x_axis_title,
y_axis_title,
chart_width=25, # عرض نمودار پیش‌فرض (واحد: cm)
chart_height=15
):
"""
افزودن نمودار به صفحه اکسل.
ورودی:
worksheet (openpyxl.Worksheet): صفحه اکسل.
chart_type (str): نوع نمودار ("line" یا "bar").
data_columns (list): لیستی از ستون‌های داده.
category_column (int): ستون دسته‌بندی‌ها.
start_row (int): ردیف شروع داده‌ها.
end_row (int): ردیف پایان داده‌ها.
chart_position (str): محل قرار گرفتن نمودار.
chart_title (str): عنوان نمودار.
x_axis_title (str): عنوان محور X.
y_axis_title (str): عنوان محور Y.
chart_width (float): عرض نمودار (واحد: cm).
chart_height (float): ارتفاع نمودار (واحد: cm).
"""
if chart_type == 'line':
chart = LineChart()
chart.style = 20
elif chart_type == 'bar':
chart = BarChart()
else:
raise ValueError("chart_type باید 'line' یا 'bar' باشد.")
chart.title = chart_title
chart.y_axis.title = y_axis_title
chart.x_axis.title = x_axis_title
chart.width = chart_width
chart.height = chart_height
categories = Reference(worksheet, min_col=category_column, min_row=start_row, max_row=end_row)
data = Reference(worksheet, min_col=data_columns, min_row=start_row - 1, max_row=end_row)
chart.add_data(data, titles_from_data=True)
chart.set_categories(categories)
for series in chart.series:
series.graphicalProperties.line.solidFill = "277358"
series.graphicalProperties.line.width = 30000
worksheet.add_chart(chart, chart_position)
# example
# add_chart(
# worksheet=worksheet,
# chart_type='line',
# data_columns=7, # ستون وزن وارد شده
# category_column=2, # ستون نام سردخانه‌ها
# start_row=7,
# end_row=l + 1,
# chart_position="A12",
# chart_title="نمودار تغییرات وزن در سردخانه‌ها",
# x_axis_title="سردخانه‌ها",
# y_axis_title="وزن (کیلوگرم)"
# )
def to_locale_str(a):
return "{:,}".format(int(a))
def convert_str_to_date(string, with_datetime=None):
"""
Convert a string to a datetime.date object.
This function tries multiple common date formats, including ISO 8601 with or
without milliseconds, and plain 'YYYY-MM-DD'. If the string cannot be parsed,
it returns None.
Parameters:
-----------
string : str
The date string to convert.
Returns:
--------
datetime.date or None
A datetime.date object if conversion succeeds, otherwise None.
Supported formats:
------------------
- 'YYYY-MM-DDTHH:MM:SS.sssZ' (ISO 8601 with milliseconds)
- 'YYYY-MM-DDTHH:MM:SSZ' (ISO 8601 without milliseconds)
- 'YYYY-MM-DD' (Simple date)
"""
string = str(string).strip()
# فرمت‌های مختلف تاریخ
date_formats = [
'%Y-%m-%dT%H:%M:%S.%fZ',
'%Y-%m-%dT%H:%M:%SZ',
'%Y-%m-%dT%H:%M:%S.%f%z', # ✅ با میلی‌ثانیه و تایم‌زون
'%Y-%m-%dT%H:%M:%S%z', # ✅ مثل: 2025-02-26T03:30:00+03:30
'%Y-%m-%dT%H:%M:%S.%f',
'%Y-%m-%dT%H:%M:%S',
'%Y-%m-%d %H:%M:%S.%f',
'%Y-%m-%d %H:%M:%S',
'%Y-%m-%d'
]
for fmt in date_formats:
try:
if with_datetime:
date = datetime.strptime(string, fmt)
else:
date = datetime.strptime(string, fmt).date()
return date
except ValueError:
continue
return None