import os
import sys
sys.path.append(os.getcwd())
from common.mysql_db import MysqlHelper


class sqlCollect():



    @classmethod
    def insert_machine_making_data(cls, name: str, task_mark: str, channel_id: str, url: str, v_id: str, piaoquan_id: str, new_title: str, code: str, formatted_time, old_title: str, oss_object_key: str):
        insert_sql = f"""INSERT INTO machine_making_data (name, task_mark, channel, user, v_id, pq_uid, title, pq_vid, data_time, old_title, oss_object_key) values ("{name}", "{task_mark}", "{channel_id}", "{url}", "{v_id}" , "{piaoquan_id}", "{new_title}", "{code}", "{formatted_time}", "{old_title}", "{oss_object_key}")"""
        MysqlHelper.update_values(
            sql=insert_sql
        )

    @classmethod
    def get_channle_id(cls, pq_id):
        """
        从数据库表中读取 id
        """
        sql = f"""select v_id,channel from machine_making_data where pq_vid = %s limit 1"""
        data = MysqlHelper.get_values(sql, (pq_id,))
        if data:
            return data[0][0],data[0][1]
        else:
            return None, None