from django.contrib import admin
from import_export.admin import ImportExportModelAdmin
from import_export import resources
from import_export.fields import Field
from .models import JobData
from django.utils.text import Truncator

# Register your models here.


class JobResource(resources.ModelResource):
    # Map 'csv_column_name' from the file to 'model_variable_name' in your DB
    job_position = Field(attribute='job_position',column_name='Job Position')
    company = Field(attribute='company',column_name='Company')
    URL = Field(attribute='URL',column_name='URL')
    status = Field(attribute='status',column_name='Status')
    date_saved = Field(attribute='date_saved',column_name='Date Saved')
    date_applied = Field(attribute='date_applied',column_name='Date Applied')
    type = Field(attribute='type',column_name='Type')
    resume = Field(attribute='resume',column_name='Resume')
    #cover_letter = Field(attribute='cover_letter',column_name='Cover Letter')
    location = Field(attribute='location',column_name='Location')
    job_description = Field(attribute='job_description',column_name='Job Description')
    salary = Field(attribute='salary',column_name='Salary')
    contact = Field(attribute='contact',column_name='Contact')
    source = Field(attribute='source',column_name='Source')

    class Meta:
        model = JobData



@admin.register(JobData)
class JobAdmin(ImportExportModelAdmin):
    list_display = ('job_position','company','URL','status','date_saved','date_applied','type','resume','location','short_description','salary','contact','source')
    #fields = ['job_position','company','URL','status','date_saved','date_applied','type','resume','cover_letter','location','job_description','salary ','contact','source']
    resource_classes = [JobResource]
    def short_description(self, obj):
        return Truncator(obj.job_description).words(10, truncate=' ...')