library/module/sql_operate.py

#!/usr/bin/python3.6
# -*- coding: utf-8 -*-
# @Time    : 2021/3/11 9:20
# @Author  : lijunliang
# @Email   : lijunliang@topnet.net.cn
# @File    : sql_operate.py
# @Software: PyCharm

from loguru import logger
from util.hash_file import cal_md5
import uuid
from pymysql.converters import escape_string


def md5_exists(MySql, file_path: str) -> str:
    """
    检查MD5去重接口
    :param file_path:文件路径
    :return:
    """
    md5_number = cal_md5(file_path)
    # 检查返回值
    sql = r"select * from doc where md5='{}'".format(md5_number)
    ret = MySql.search(sql)
    if ret:
        return True, md5_number
    return False, md5_number


def create_insert_sql(value: dict) -> str:
    '''
    插入mysql语句
    :param value:数据字典
    :return:
    '''
    docSummary = escape_string(value["docSummary"])
    docName = escape_string(value["docName"])
    sql = r"""insert into doc (id,userId,docName,docFileType,docFileSuffix,
    docFileSize,docPageSize,docClass,docTags,uploadDate,isDelete,ossDocId,
    md5,ossPdfId,ossTxtId,price,downOrUp,docSummary,previewImgId,appId) values ('{}','{}','{}','{}','{}','{}','{}','{}','{}','{}',
  '{}','{}','{}','{}','{}','{}','{}','{}','{}','{}')""".format(str(uuid.uuid1()),
                                                               value["userId"], docName, value["docFileType"],
                                                               value["docFileSuffix"], value["docFileSize"],
                                                               value["docPageSize"],
                                                               value["docClass"], value["docTags"], value["uploadDate"],
                                                               value["isDelete"], value["ossDocId"], value["md5"],
                                                               value["ossPdfId"],
                                                               value["ossTxtId"], value["price"], value["downOrUp"],
                                                               docSummary, value["previewImgId"],
                                                               value["appId"])
    return sql


def save_field(MySql, value: dict) -> str:
    """
    生成保存字段
    :param kwargs:
    :return:
    """
    # 生成sql
    sql = create_insert_sql(value)

    # 插入
    insert_state = MySql.execute_sql(sql)

    return insert_state