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


class sqlCollect():
    """
    视频信息写入库中
    """
    @classmethod
    def insert_task(cls, video_id):
        insert_sql = f"""INSERT INTO guanggao_video (video_id) values ('{video_id}')"""
        MysqlHelper.update_values(
            sql=insert_sql
        )

    """
    判断该任务id是否用过
    """
    @classmethod
    def is_video_id(cls):
        sql = """
            SELECT video_id FROM guanggao_video
        """
        data = MysqlHelper.get_values(sql)
        return data