import pymysql
import xlwt as xlwt


def Toexcel(path, sql, title):
    conn = pymysql.connect(host='47.106.183.36', port=3306,
                           user='fuchuang', password='fuchuang',
                           database='fuchuang', charset='utf8mb4')
    curs = conn.cursor()
    curs.execute(sql)
    rows = curs.fetchall()
    w = xlwt.Workbook(encoding='utf-8')
    style = xlwt.XFStyle()  # 初始化样式
    font = xlwt.Font()  # 为样式创建字体
    font.name = "微软雅黑"  # 如果是 python2 ，需要这样写 u"微软雅黑"
    style.font = font  # 为样式设置字体
    ws = w.add_sheet("视频信息", cell_overwrite_ok=True)
    # 将 title 作为 Excel 的列名
    title = title.split(",")
    for i in range(len(title)):
        ws.write(0, i, title[i], style)
    # 开始写入数据库查询到的数据
    for i in range(len(rows)):
        row = rows[i]
        for j in range(len(row)):
            if row[j]:
                item = row[j]
                ws.write(i + 1, j, item, style)
    # 写文件完成，开始保存xls文件
    w.save(path)
    conn.close()


sql_1 = '''select * from video'''
Toexcel('视频信息.xls', sql_1, "id,标题,播放量,弹幕数,发布者")
sql_2 = '''select * from bangumi'''
Toexcel('番剧信息.xls', sql_2, "id,番名,播放量,评分,弹幕数")