import openpyxl
from openpyxl.drawing.image import Image
import requests
from io import BytesIO

from models import Product

# Отримання товарів із фото
try:
    selected_products = Product.select().where(Product.photo_url.is_null(False)).limit(20)
except Exception as e:
    print(f"Виникла помилка: {e}")


# Створення воркбуку та аркушу
wb = openpyxl.Workbook()
ws = wb.active
ws.title = "Sheet"
# Захист об'єктів
ws.protection.objects = True
ws.protection.password = ''

# Зміна ширини стовпця до 200px
ws.column_dimensions["A"].width = ws.column_dimensions["B"].width  = 200 / 7  # встановлюємо ширину стовпця у символах

# Додавання назви та фото товару до аркушу ws
def add_to_file(product, row_num):

    # Встановлення висоти рядка 200px
    ws.row_dimensions[row_num].height = 200 * 0.75

    try:
        # Отримання зображення
        response = requests.get(product.photo_url)
        img_data = BytesIO(response.content)
        # Об'єкт зображення
        img = Image(img_data)

        # Додаємо зображення на лист
        ws.add_image(img, 'B' + str(row_num))

    except requests.exceptions.HTTPError as http_err:
        print(f"HTTP помилка: {http_err}")
    except  requests.exceptions.RequestException as req_err:
        print(f"Помилка запиту: {req_err}")


    # Координати клітинки назви товару
    name_coordinates = 'A' + str(row_num)
    # Запис назви товару
    ws[name_coordinates] = selected_products[i].product_name
    # Стилізація: перенесення тексту, позиціювання по центру у вертикалі
    ws[name_coordinates].alignment = openpyxl.styles.Alignment(vertical='center', wrap_text=True)

#Запис до файлу
for i in range(len(selected_products)):

    # Виклик функції запису
    add_to_file(selected_products[i], i+1)




# Встановлюємо захист для всього листа з паролем
ws.protection.enable()
# Збережння файлу
wb.save("products.xlsx")