excel表数据的导入导出

从excel表导入MySQL数据库

import xlrd
import pymysql


# 写入excel表数据
def insert_data():
    data = xlrd.open_workbook("excel/test.xlsx")
    sheet_names = data.sheet_names()
    # print(sheet_names)
    # 连接数据库
    db = pymysql.connect(host='localhost', user='root', passwd='123456',
                         db='excel', charset='utf8')
    cursor = db.cursor()
    for i in range(len(sheet_names)):
        # 当前sheet的名字
        table_name = sheet_names[i]
        # 当前的sheet
        now_table = data.sheet_by_index(i)
        # 获得当前sheet的列数就是属性数
        cols_num = now_table.ncols
        # 获得当前表格的行数,就是有多少的数据量
        rows_num = now_table.nrows
        attrs = now_table.row_values(0)
        # 判断表格是否存在
        flag = cursor.execute("show tables like '%s';" % table_name)
        # print(flag)
        # 表存在
        if flag:
            print("-------")
            # 将当前的sheet插入到数据库
            for j in range(1, rows_num):
                row_value = now_table.row_values(j)
                # 处理要插入的数据,把非字符串的数据转换成字符串类型,同时将字符串变成 sql语句需要的类型
                for k in range(1, len(row_value)):
                    ctype = now_table.cell(j, k).ctype
                    # print('ctype', ctype)
                    # ctype: 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error
                    if ctype == 2 and row_value[k] % 1 == 0:
                        tmp = int(row_value[k])
                        row_value[k] = str(tmp)
                    c = row_value[k]
                    row_value[k] = "'" + c + "'"
                # print(row_value)
                sql = "insert into %s(%s) values(%s);" % (table_name, ','.join(attrs[1:]), ','.join(row_value[1:]))
                cursor.execute(sql)
                db.commit()
        else:
            # 为sheet建表
            cursor.execute(f"create table {table_name} (id int primary key auto_increment);")
            db.commit()
            for l in range(1, cols_num):
                cursor.execute(f"alter table {table_name} add column {attrs[l]} varchar(200);")
                db.commit()
            # 将当前的sheet插入到数据库
            for j in range(1, rows_num):
                row_value = now_table.row_values(j)
                # 处理要插入的数据,把非字符串的数据转换成字符串类型,同时将字符串变成 sql语句需要的类型
                for k in range(1, len(row_value)):
                    ctype = now_table.cell(j, k).ctype
                    print('ctype', ctype)
                    # ctype: 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error
                    if ctype == 2 and row_value[k] % 1 == 0:
                        tmp = int(row_value[k])
                        row_value[k] = str(tmp)
                    c = row_value[k]
                    row_value[k] = "'" + c + "'"
                sql = "insert into %s(%s) values (%s);" % (table_name, ','.join(attrs[1:]), ','.join(row_value[1:]))
                cursor.execute(sql)
                db.commit()
    db.close()


if __name__ == '__main__':
    insert_data()

从MySQL数据库写入excel表

import xlwt
import pymysql
import os
import datetime


# 从数据库读取数据写入excel,生成xls or xlsx格式文件
def table_export_to_xls(sql):
    table_name = 'sheet1'
    # 连接数据库
    db = pymysql.connect(host='localhost', user='root', passwd='123456',
                         db='excel', charset='utf8')
    cursor = db.cursor()
    cursor.execute(sql)
    # 重置游标位置
    cursor.scroll(0, mode='absolute')
    # 搜取所有的结果
    results = cursor.fetchall()
    # print(result)
    # 获取属性名
    attrs = cursor.description
    print(attrs)

    workbook = xlwt.Workbook()
    sheet1 = workbook.add_sheet(table_name, cell_overwrite_ok=True)
    # 写入表格的属性值
    for i in range(1, len(attrs)):
        sheet1.write(0, i-1, attrs[i][0])
    # 将数据库的数据导入表格
    row = 1
    col = 0
    for row in range(1, len(results)+1):
        for col in range(0, len(attrs)-1):
            sheet1.write(row, col, results[row-1][col+1])
    # 保存路径
    now_path = os.path.dirname(__file__)
    print(now_path)
    file_path = os.path.join(now_path, 'excel')
    print(file_path)
    export_time = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
    file_name = 'test-{0}.xlsx'.format(export_time)
    workbook.save(os.path.join(file_path, file_name))

    if os.path.isfile(os.path.join(file_path, file_name)):
        print('数据库重成功导出数据!')
        return {'path': file_path, 'name': file_name}
    else:
        print('导出失败!')
        return 'error'


if __name__ == '__main__':
    sql = "select * from sheet1"
    table_export_to_xls(sql)

PostgreSQL可参考:

https://blog.csdn.net/huangmengfeng/article/details/81005505
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,222评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,455评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,720评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,568评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,696评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,879评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,028评论 3 409
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,773评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,220评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,550评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,697评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,360评论 4 332
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,002评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,782评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,010评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,433评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,587评论 2 350

推荐阅读更多精彩内容