# -*- coding: utf-8 -*-
"""
create by caijinxu on 2019/4/1
"""
import re

import pymysql

from settings import *

__author__ = 'caijinxu'


getwimgsql = "SELECT imgurl FROM dailywhitiwebimg UNION SELECT imgurl FROM retentionimg WHERE status = 0 UNION " \
             "SELECT imgurl FROM whitechannelimg WHERE status = 0 UNION SELECT imgurl FROM whitewebimg WHERE " \
             "status = 0 UNION SELECT imgurl FROM whiteimage WHERE status = 0;"


def get_imgurlinfo():
    """查询白名单数据库中所有图片表"""
    bdb_connection = pymysql.connect(host=MYSQLHOST, user=MYSQLUSER, db=MYSQLDB, passwd=MYSQLPASSWD, charset="utf8",
                                     init_command='SET NAMES UTF8', cursorclass=pymysql.cursors.DictCursor)
    b_cursor = bdb_connection.cursor()
    b_cursor.execute(getwimgsql)
    result = b_cursor.fetchall()
    bdb_connection.close()
    return result


def get_whiteimg():
    allwhiteimg = get_imgurlinfo()
    if not allwhiteimg:
        print("查询mysql数据库失败")
        raise Exception("查询mysql数据库失败")
    white_imgs = set()
    for img in allwhiteimg:
        # print(img)
        for k, v in WEBROOT.items():
            # print(k, img)
            if re.match(k, img['imgurl']):
                img['imgurl'] = img['imgurl'].rstrip('\n').rstrip('\r').replace(' ', '')
                whitefile = re.sub(k, v[0], img['imgurl'])
                # print(white_imgs)
                white_imgs.add(whitefile)
                break
    return white_imgs

# if __name__ == '__main__':
#     get_whiteimg()