python Excel表格操作封装

"""
Excel表格操作封装
"""
import openpyxl


class ExcelXlsx(object):
	
	"""
	需要以xlsx为后缀名
	默认创建名为 Sheet 的表 
	"""
	def __init__(self, fileName="example.xlsx"):
		self.fileName = fileName
		self.wb = openpyxl.Workbook()

	"""
	@string sheetName
	"""
	def addOneSheet(self, sheetName):
		self.wb.create_sheet(sheetName)

	"""
	@string sheetName
	@tuple(1,2,3) or list[1,2,3] data
	"""
	def insertOneRow(self, sheetName, data):
		sheet = self.wb[sheetName]
		sheet.append(data)

	"""
	close and save xlsx
	"""
	def save(self):
		self.wb.save(self.fileName)


if __name__ == "__main__":
	excelTool = ExcelXlsx("test.xlsx")
	excelTool.insertOneRow("Sheet", (1, 2, 3))
	excelTool.save()