80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
"""
|
|
===============================================================================
|
|
FILE: services/reports/styles.py
|
|
ROLE: Стили, палитры цветов, форматирование дат и защита от блокировок Excel.
|
|
===============================================================================
|
|
"""
|
|
|
|
import os
|
|
import time
|
|
from datetime import datetime
|
|
from xlsxwriter.exceptions import FileCreateError
|
|
from config import REPORTS_DIR
|
|
|
|
MONTHS_RU_GENITIVE = {
|
|
1: "января", 2: "февраля", 3: "марта", 4: "апреля",
|
|
5: "мая", 6: "июня", 7: "июля", 8: "августа",
|
|
9: "сентября", 10: "октября", 11: "ноября", 12: "декабря"
|
|
}
|
|
|
|
MONTHS_RU_NOMINATIVE = {
|
|
1: "январь", 2: "февраль", 3: "март", 4: "апрель",
|
|
5: "май", 6: "июнь", 7: "июль", 8: "август",
|
|
9: "сентябрь", 10: "октябрь", 11: "ноябрь", 12: "декабрь"
|
|
}
|
|
|
|
|
|
def format_date_ru(date_str):
|
|
date_clean = str(date_str).replace('_', '.')
|
|
try:
|
|
dt = datetime.strptime(date_clean, "%d.%m.%Y")
|
|
return f"{dt.day} {MONTHS_RU_GENITIVE[dt.month]} {dt.year}"
|
|
except Exception:
|
|
return date_str
|
|
|
|
|
|
def get_dated_reports_dir(date_str):
|
|
date_clean = str(date_str).replace('_', '.')
|
|
try:
|
|
dt = datetime.strptime(date_clean, "%d.%m.%Y")
|
|
year_str = str(dt.year)
|
|
month_name = MONTHS_RU_NOMINATIVE[dt.month]
|
|
except Exception:
|
|
now = datetime.now()
|
|
year_str = str(now.year)
|
|
month_name = MONTHS_RU_NOMINATIVE[now.month]
|
|
|
|
target_dir = os.path.join(REPORTS_DIR, year_str, month_name)
|
|
os.makedirs(target_dir, exist_ok=True)
|
|
return target_dir
|
|
|
|
|
|
def safe_close_workbook(wb, output_path, target_dir, filename):
|
|
try:
|
|
wb.close()
|
|
return output_path
|
|
except (FileCreateError, OSError, PermissionError):
|
|
alt_filename = filename.replace(".xlsx", f"_{int(time.time())}.xlsx")
|
|
alt_path = os.path.join(target_dir, alt_filename)
|
|
try:
|
|
wb.filename = alt_path
|
|
wb._store_workbook()
|
|
return alt_path
|
|
except Exception:
|
|
return output_path
|
|
|
|
|
|
def create_xlsx_format(workbook, font_name="Calibri", font_size=11, bg_color=None, bold=False, align="left", wrap=False):
|
|
fmt_dict = {
|
|
'font_name': font_name,
|
|
'font_size': font_size,
|
|
'bold': bold,
|
|
'align': align,
|
|
'valign': 'vcenter',
|
|
'border': 1,
|
|
'border_color': '#D3D3D3',
|
|
'text_wrap': wrap
|
|
}
|
|
if bg_color:
|
|
fmt_dict['bg_color'] = bg_color
|
|
return workbook.add_format(fmt_dict) |