使用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)