import os
from settings import conf
from .named_enum import DocStatus


class DocHandler:

    @staticmethod
    def get_link(doc_id, file='pdf'):
        data_path = os.path.join(conf.DATA_DIR, str(doc_id))
        if file == 'pdf':
            return os.path.join(data_path, '{0}.pdf'.format(str(doc_id)))
        elif file == 'img':
            return os.path.join(data_path, '{0}_img.zip'.format(str(doc_id)))
        else:
            return os.path.join(data_path, '{0}.xlsx'.format(str(doc_id)))

    def get_doc_list(self, doc_queryset):
        for doc_dict in doc_queryset:
            if doc_dict['status'] != DocStatus.COMPLETE.value:
                continue
            doc_id = doc_dict.get('id')
            doc_dict['pdf_link'] = self.get_link(doc_id)
            doc_dict['img_link'] = self.get_link(doc_id, 'img')
            doc_dict['excel_link'] = self.get_link(doc_id, 'excel')
        return list(doc_queryset)