from file.file_db import FIND_FILE


def get_pdf_file_impl(file_type, fid):
    """
    根据文件类型、文件ID,获取pdf文件
    """
    # ftm: file_type_map
    ftm = {
        "cc_report": "综信报告",
        "cc_certificate": "综信证书",
        "esg_report": "ESG报告",
        "esg_certificate": "ESG证书"
    }
    filestream = FIND_FILE(ftm[file_type], fid)
    return filestream


def company_esg_rating_report_by_fid(fid):
    """
    根据fid返回文件
    Args:
        fid:
    Returns:
        filestream
    """
    filestream = find_file('esg报告', fid)
    return filestream


def company_esg_rating_certification_by_fid(fid):
    """
    根据fid返回文件
    Args:
        fid:
    Returns:
        filestream
    """
    filestream = find_file('ESG评价证书', fid)
    return filestream