from django.shortcuts import render, redirect, get_object_or_404
from openpyxl import Workbook
from django.http import HttpResponse
from django.contrib import messages

from .models import Category

# User dari finance
from finance.models import User


def scm_home(request):

    return render(
        request,
        'scm/home.html'
    )


def export_to_excel_category(request):

    categories = Category.objects.all().order_by('category')

    wb = Workbook()
    ws = wb.active
    ws.title = "Category"

    # Header
    ws.append([
        "No",
        "Category",
    ])

    # Data
    for no, item in enumerate(categories, start=1):
        ws.append([
            no,
            item.category or "",
        ])

    # Column width
    ws.column_dimensions['A'].width = 10
    ws.column_dimensions['B'].width = 40

    response = HttpResponse(
        content_type=(
            'application/vnd.openxmlformats-officedocument.'
            'spreadsheetml.sheet'
        )
    )

    response['Content-Disposition'] = (
        'attachment; filename="category.xlsx"'
    )

    wb.save(response)

    return response