getmysqlconninfo.py

Posted 2020-10-09

#!/usr/bin/env python
#-*- encoding: utf8 -*-

import xlrd

"""
此模块作用：从excel文件获取数据库连接信息，第一列是IP，第二列是账号，第三列是密码，第四列是端口号。
返回的csv_data变量是一个列表
"""
class GetConn:
    def __init__(self,fromfile):
        self.fromfile = fromfile

    def get_csv(self):
        csv_file = xlrd.open_workbook(self.fromfile,‘rb‘)
        csv_sheet = csv_file.sheet_by_index(0)
        nrows = csv_sheet.nrows
        csv_data = []
        for i in range(nrows):
            csv_row = csv_sheet.row_values(i)
            csv_data.append(csv_row)
        return csv_data