"""
@author: luojunhui
"""
import pymysql
from config import planConfigDict


class DeNetMysql(object):
    """
     Mysql Server
    """
    connection = pymysql.connect(
        host="rm-t4na9qj85v7790tf84o.mysql.singapore.rds.aliyuncs.com",  # 数据库IP地址，内网地址
        port=3306,  # 端口号
        user="crawler_readonly",  # mysql用户名
        passwd="cyber#crawler_2023",  # mysql用户登录密码
        db="aigc-admin-prod",  # 数据库名
        charset="utf8mb4"  # 如果数据库里面的文本是utf8编码的，charset指定是utf8
    )

    @classmethod
    def getUnEmptyPlan(cls):
        """
        :return:
        """
        sql = "select id, name from crawler_plan where name like '%腾讯互选-%' and crawler_total_num > 0;"
        cursor = cls.connection.cursor()
        cursor.execute(sql)
        data = cursor.fetchall()
        result = [list(line)[0] for line in data]
        # print(result)
        result = [i for i in result if planConfigDict.get(i)]
        return result