futool/tests/test_fu_excel.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2023/4/2 9:33
# @Author  : old tom
# @File    : test_fu_excel.py
# @Project : futool
from unittest import TestCase
from futool.poi.fu_excel import ExcelReader, SimpleExcelWriter

ex_reader = ExcelReader(file_path=r'D:\test\test3.xlsx')


# @Desc    :
class TestExcelReader(TestCase):

    def test_read_sheet(self):
        sheet = ex_reader.load_sheet('Sheet1')
        # for row in sheet:
        #     for cell in row:
        #         print(cell.value)
        for i, row in enumerate(sheet):
            if i == 0:
                for cell in row:
                    print(cell.value)

    def test_read_sheet_by_index(self):
        sheet = ex_reader.load_sheet_by_index(0)
        for row in sheet:
            for cell in row:
                print(cell.value)

    def test_read_row(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_row(0))

    def test_read_range_row(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_range_rows(1, 3))

    def test_read_column(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_column(4))

    def test_read_rows(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_rows(row_index=[0, 3]))

    def test_read_all(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_all())

    def test_read_range_column(self):
        ex_reader.load_sheet('Sheet1')
        print(ex_reader.read_range_column(1, 4))


class TestExcelWriter(TestCase):
    def test_write(self):
        ew = SimpleExcelWriter(r'D:\test\test3.xlsx')
        # ew.write(head=['序号', '姓名', '年龄', '身份证', '住址'], data=[('1', '张三', '1', '101111', '特特特特特')])
        ew.write(head=['序号', '姓名', '年龄', '身份证', '住址'], data=self.gen_big_excel())

    def gen_big_excel(self):
        big_data = []
        for i in range(0, 999999):
            big_data.append(('1', '张三', '1', '101111', '特特特特特'))
        return big_data