pdf_title_image/api.py

# -*- coding: utf-8 -*-
# @Author: privacy
# @Date:   2024-09-03 11:24:56
# @Last Modified by:   privacy
# @Last Modified time: 2024-09-30 13:41:28
import os
import time

import zipfile
import requests
from pydantic import BaseModel
from celery.result import AsyncResult
from fastapi import FastAPI, UploadFile, File, Body

from celery_tasks import celery_app
from celery_tasks.commonprocess import bidding_factor, test_all_files
from celery_tasks.project_loc import extract_project
from celery_tasks.commonprocess import full_func


app = FastAPI()


@app.get('/result')
def back(taskid):
    result = AsyncResult(id=taskid, app=celery_app)
    if result.successful():
        val = result.get()
        return "执行完成，结果：%s" % val
    else:
        return '正在处理中...'


# 检测编码(已完成)
def decode_path(path):
    '''zipfile解压出现乱码，将乱码的路径编码为UTF8'''
    try:
        path_name = path.decode('utf-8')
    except Exception:
        path_name = path.encode('437').decode('gbk')
        path_name = path_name.encode('utf-8').decode('utf-8')
    return path_name


# 下载招标、投标、专家打分表图片文件
def get_file(path_name="1@/static/bm/bid_check_bidder/20240808151412_投标文件25mb.zip", save_file='./download'):
    url = 'http://192.168.1.111:9999/admin/sys-file/download'  # 要下载的文件的URL
    url = 'http://192.168.1.111:9999/admin/sys-file/preview'
    json = {"path": path_name}

    os.makedirs(save_file, exist_ok=True)
    code = 0
    try:
        response = requests.get(url, data=json, stream=True)  # 发送GET请求，stream参数指定以流的方式下载文件
    except Exception:
        print('下载失败，状态码：', response.status_code)
        code = 1
    file_name = path_name.split('/')[-1]

    save_file_path = save_file + '/new_' + file_name

    if response.status_code == 200:  # 检查响应状态码
        with open(save_file_path, "wb") as fp:
            fp.write(response.content)
            print('文件下载完成！')
    if code != 0:  # 下载失败抓取
        return save_file_path, code
    if os.path.isfile(save_file_path) and save_file_path.endswith('.zip'):
        # 解压方式2：防止乱码
        tempdir = time.strftime("%Y_%m_%dT%H_%M_%S")

        os.makedirs('./cache/' + tempdir, exist_ok=True)
        file_path_list = []
        try:
            with zipfile.ZipFile(save_file_path, allowZip64=True) as zf:
                # 排除目录文件
                file_iter = (filename for filename in zf.filelist if os.path.isfile(save_file_path))
                for filename in file_iter:
                    # 编码文件名称为 utf 格式
                    filename.filename = decode_path(filename.filename)  # 防止乱码的操作
                    zf.extract(filename, "./cache/" + tempdir)
                    if filename.filename[-1] == '/':
                        continue
                    file_path_list.append("./cache/" + tempdir + '/' + filename.filename)
        except Exception as e:
            print(e)

        return file_path_list, code

    return save_file_path, code


# 预审查、清标
@app.post("/file_upload")
async def file_upload(text_list=Body(None)):
    """
    {
        'bidderUnit': '杭州华新机电工程有限公司',
        'bidderFile': '1@/static/bm/bid_pre_check/20240924172133_三峡投标文件新-华新(1).zip',
        'buyFile': '1@/static/bm/project/20240924171822_三峡电站左岸厂房桥机远程智能化操作研发与实施重新招标.pdf',
        'reportFlag': '0',
        'projectName': '华新',
        'projectId': '2024-9-24-0'
    }
    """
    try:
        json_post = eval(text_list)
    except Exception:
        json_post = text_list

    buyFile = json_post['buyFile']  # 采购文件

    bidderFile = json_post['bidderFile']  # 投标文件

    try:
        buy_file_path, code = get_file(buyFile)

        bidder_file_path, code = get_file(bidderFile)

        json_data = {
            "code": 0,
            "name": '',
            "msg": "文件下载成功",
            "data": {},
            "ok": True
        }
    except Exception:
        json_data = {
            "code": 1,
            "name": '',
            "msg": "文件下载失败",
            "data": {},
            "ok": False
        }

    return json_data