读取excel表,并写入到本地数据库

使用python读取学生信息Excel表,excel表的路径为C:\stu_excel\test_stu.xlsx,将读取到的数据,插入到本地的students数据库当中的students表中,要保证数据的完整性和准确性。

import xlrd
import pymysql
class test1(object):
    def __init__(self, host, port, user, password, databaseName, excelpath):
        self.host = host #数据库的ip
        self.port = port #数据库的端口
        self.user = user #数据库的用户名
        self.password = password #数据库的密码
        self.databaseName = databaseName  #数据库名
        self.excelpath = excelpath # 本地excel文件的路径
    def read_excel(self):
        excelfile = xlrd.open_workbook(self.excelpath) #打开目标excel文件
        sheet = excelfile.sheet_by_index(0)  #根据索引选择要操作的工作表。  这里选择的是第一个工作表,下标为0
        nrows = sheet.nrows #获取该sheet表中的有效行数
        new_infolist = [] #定义一个大列表,将读取出来的学生数据都存放到该大列表当中
        for row_num in range(1, nrows): # 从第二行开始读取每一行的数据
            row_values = sheet.row_values(row_num) #返回由该行中所有单元格的数据组成的列表。 即获取每一行的数据
            # print(row_values)
            for i in row_values:
                row_values[2] = int(row_values[2]) #年龄转为int类型
                row_values[5] = int(row_values[5]) # 手机号有小数点,转为整型
                new_infolist.append(row_values) #将获取到的每一行的数据添加到定义的大列表中
                break
        # print(new_infolist)
        return new_infolist

    def db_mysql(self,infos):
        try:
            db = pymysql.connect(self.host, self.user, self.password, self.databaseName, self.port)  # 连接数据库
            cursor = db.cursor()  # 创建游标,用来执行sql命令
            
            # sql = 'drop table if exists students;'  # 如果有students表就删除,如果没有这个表,这个sql就不必执行了
            # cursor.execute(sql)  # 执行sql命令
            # db.commit()  # 提交sql命令
            
            sql = "create table if not exists students(" \
                  "sid int primary key auto_increment," \
                  "sname varchar(20) not null unique," \
                  "gender char(5)," \
                  "age int," \
                  "education char(3)," \
                  "place varchar(10)," \
                  "phone char(11)," \
                  "remark char(100) default 'TEST'," \
                  "email varchar(50)" \
                  ");"    # 创建students表
            cursor.execute(sql)
            db.commit()
            for info in infos:  # 向表中插入数据
                sql = f"insert into students(sname,gender,age,education,place,phone,remark,email) VALUES(" \
                      f"'{info[0]}','{info[1]}',{info[2]},'{info[3]}','{info[4]}',{info[5]},'{info[6]}','{info[7]}');"
                cursor.execute(sql)
                db.commit()
            cursor.close()
            db.close()
        except Exception as e:
            print(e)
if __name__ == '__main__':
    t1 = test1('127.0.0.1', 3306, 'root', '123456', 'students', r'C:\stu_excel\test_stu.xlsx')
    infos = t1.read_excel()
    t1.db_mysql(infos)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。