RasadDam_Backend/common/helper_excel.py

from datetime import datetime
from io import BytesIO

import jdatetime
import openpyxl
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

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")
GRAY_CELL = PatternFill(start_color="B0B0B0", fill_type="solid")
CREAM_CELL = PatternFill(start_color="D8AA72", fill_type="solid")
LIGHT_CREAM_CELL = PatternFill(start_color="E8C6A0", fill_type="solid")


def shamsi_date(date, in_value=None):
    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,
                  different_color=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
        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,
            'gray': GRAY_CELL
        }
        if color in color_dict:
            cell.fill = color_dict[color]
        else:
            cell.fill = CREAM_CELL
        if different_color:
            cell.fill = PatternFill(start_color=different_color, fill_type="solid")
        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 = LIGHT_CREAM_CELL
        cell.font = Font(size=10, bold=True, color='404040')

        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, bold=True)
    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,
                 m_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,
        'gray': GRAY_CELL
    }

    for item in range(len(list)):
        cell = worksheet.cell(row=l, column=item + num, value=list[item])
        cell.alignment = Alignment_CELL
        if height is not None:
            worksheet.row_dimensions[l].height = height

        if width is not None:
            worksheet.column_dimensions[openpyxl.utils.get_column_letter(item + num)].width = width

        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:
            if m_color:
                cell.fill = PatternFill(start_color=m_color, fill_type="solid")
            else:
                cell.fill = PatternFill(start_color="FAF0E7", fill_type="solid")

        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


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 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 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):
    string = str(string).strip()
    try:
        return datetime.strptime(string, '%Y-%m-%dT%H:%M:%S.%fZ').date()
    except ValueError:
        try:
            return datetime.strptime(string, '%Y-%m-%dT%H:%M:%SZ').date()  # Added format without milliseconds
        except ValueError:
            try:
                return datetime.strptime(string, '%Y-%m-%d').date()
            except ValueError:
                return None