from django.db import models

from file.validators import validate_version_format

# Create your models here.
class File(models.Model):
    # file content
    filecontent = models.FileField()
    # 파일 이름
    name = models.CharField(max_length=100)
    # 파일 설명
    desc = models.TextField(null=True, blank=True)
    # 파일 확장자
    ext = models.CharField(max_length=10, null=True)
    # 파일 용량
    size = models.IntegerField()
    # 문서 업로드 시간
    upload_datetime = models.DateTimeField(auto_now_add=True)
    # 문서 생성 시간, creation_date 는 문서를 실제 생성한 시간
    creation_date = models.DateField(null=True, blank=True)
    # 문서 버전
    version = models.CharField(max_length=100, validators=[validate_version_format], null=True, blank=True)

# file/validators.py
import re
from django.core.exceptions import ValidationError

def validate_version_format(value):
    # "v.x.x.x"와 같은 형식인지 검사
    pattern = re.compile(r'^v\.\d+\.\d+\.\d+$')
    if not pattern.match(value):
        raise ValidationError('Invalid version format. Use "v.x.x.x" format.')

# fms/file/forms.py
from django import forms
from .models import File

class FileCreationForm(forms.ModelForm):
    class Meta:
        model = File
        fields = ['filecontent',
                  'desc',
                  'creation_date',
                  'version']

        widgets = {
            'creation_date': forms.widgets.DateInput(attrs={'type': 'date'})
        }

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

# Create your views here.
def create(request):
    if request.method == 'POST':
        form = FileCreationForm(request.POST, request.FILES)  # <-- FILES 와 같이 입력이 되어야 한다.
        if form.is_valid():
            inst = form.save(commit=False)
            # 파일 속성 추출 및 생성
            file = form.cleaned_data['filecontent']
            file_ext = os.path.splitext(file.name)[1]

            # 속성 정보 저장
            inst.name = file.name
            inst.size = file.size
            inst.ext = file_ext
            inst = form.save(commit=False)

            # file field 파일 저장
            inst.filecontent.save(file.name, file)

            # 일반 form data 저장
            inst.save()
            return redirect(to='file:index')
        else:

            print('Errors:', form.errors)
            return render(request, template_name='file/create.html', context={'form': form, 'errors': form.errors})

    elif request.method == 'GET':
        form = FileCreationForm()  # <-- FILES 와 같이 입력이 되어야 한다.
        return render(request, template_name='file/create.html', context={'form': form})

    else:
        raise NotImplementedError
    return HttpResponse('200')

# fms/file/urls.py
from django.urls import path
from file.views import index, create

app_name='file'
urlpatterns = [
    path('create/', create, name='create'),
]

def index(request):
	 # 데이터베이스에서 모든 File 객체를 가져옴
    files = File.objects.all()

    # 데이터베이스로부터 DataFrame 생성
    files_df = pd.DataFrame.from_records(files.values())

    # 필요한 column 순서로 DataFrame 열을 선택
    column_order = ['id', 'name', 'filecontent', 'desc', 'ext', 'size', 'upload_datetime', 'creation_date',
                    'version']
    files_df = files_df.loc[:, column_order]

    # 파일 다운로드를 위한 URL 리스트 생성
    urls = [file.filecontent.url for file in files]

    # 파일 이름을 클릭하면 해당 파일을 다운로드할 수 있는 링크로 변환
    target_colname = 'name'
    target_series = files_df.loc[:, target_colname]
    url_tags = [wrap_with_tag(str(ele), tag_name='a', href=url) for ele, url in zip(target_series, urls)]
    files_df.loc[:, target_colname] = url_tags

    # 업로드 시간을 원하는 형식으로 변환
    target_colname = 'upload_datetime'
    files_df[target_colname] = files_df[target_colname].dt.strftime('%y:%m:%d %H:%M:%S')

    # DataFrame을 HTML로 변환
    table_html = files_df.to_html(escape=False)

    # HTML 템플릿 파일의 절대 경로 가져오기
    abspath = get_template_path('file/bootstrap.html')  # {app_name}/{templates 이름}을 입력하면 절대 경로로 반환
    # BeautifulSoup로 HTML 템플릿 파일을 읽어오기
    base_soup = bs4_load(abspath)

    # 템플릿에 DataFrame을 삽입
    soup = insert_html_to_body(soup=base_soup, target=table_html)

    # BeautifulSoup 객체를 HttpResponse로 반환
    return HttpResponse(soup.prettify(formatter=None))