从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