Python反射によるExcelとオブジェクトの変換

Python反射によるExcelとオブジェクトの変換
コードは一番下にあり、手を伸ばして党を直接最後までスクロールします.
シーン
Excelからメモリにロードし、classオブジェクトに変換して操作する必要があります
環境

Python3.8

openpyxl==3.0.5

先行知識
反射(このヘルプクラスで使用されるいくつかの反射方法のみを説明します)
setattr、getattr

class Person():
    name = None

    def __init__(self, name):
        self.name = name


p = Person("laowang")
n = getattr(p, "name")
print(n)
#     ：    laowang

setattr(p, "name", "laoxu")
n2 = getattr(p, "name")
print(n2)
#     ：   laoxu

実体化オブジェクトの反射(Reflection Materialized Objects)

class Person():
    name = None

    def print_sth(self):
        print("       ", self.name)


def test(clazz):
    """
         
    :param clazz:         
    """
    x = clazz()
    setattr(x, "name", "  ")
    x.print_sth()
    #   ：              


test(Person)

Excel操作クラスライブラリ-openpyxl
Excelの作成

from openpyxl import Workbook


wb = Workbook()
ws1 = wb.active

ws1.append(['name', 'age', 'isManager', 'remark'])
ws1.append(['', '', '', ' '])
ws1.append(['  ', '18', 'false', '  '])
ws1.append(['laoxu', '28.4', 'TRUE', 'zaoricaifuziyou'])
ws1.append(['', '', '', ' '])

ws2 = wb.create_sheet("ws2")

ws2.append(['name', 'age', 'isManager', 'remark'])
ws2.append(['  ', '50', 'TRuE', 'fly knife'])
ws2.append(['', '', '', ' '])
ws2.append(['aaa', '11', 'false', 'hehehe'])

wb.save("test_convert_2_class.xlsx")

Excelの読み込み

from openpyxl import Workbook

def print_row(arr):
    """      ，   """
    for item in arr:
        print(item,end="\t\t|")
    print()

#            Excel  
work_book = load_workbook("test_convert_2_class.xlsx")
result = []
for sheet_name in work_book.sheetnames:
    print("-----------------",sheet_name,"-----------------")
    ws = work_book[sheet_name]

    #     
    table_heads = []
    for title_row in ws.iter_rows(max_row=1):
        for cell in title_row:
            table_heads.append(cell.value)
    print_row(table_heads)
    #      
    table = []
    for row in ws.iter_rows(min_row=2):
        row_data=[]
        for column_index in range(len(row)):
            row_data.append(row[column_index].value)
        print_row(row_data)
#       ：
# ----------------- Sheet -----------------
# name        |age        |isManager        |remark        |
# None        |None        |None        |         |
#           |18        |false        |          |
# laoxu        |28.4        |TRUE        |zaoricaifuziyou        |
# None        |None        |None        |         |
# ----------------- ws2 -----------------
# name        |age        |isManager        |remark        |
#           |50        |TRuE        |fly knife        |
# None        |None        |None        |         |
# aaa        |11        |false        |hehehe        |

手を伸ばす党コード
excel_helper.py

import os
import re
from os.path import isfile

from openpyxl import load_workbook, Workbook


def _convert_value(value):
    """
           ，      
      "true"/"false"(      )   bool 
             float  
      （  、  ）   None
    :param value:      
    :return:       
    """
    value_str = str(value).lower()
    if value_str == 'true':
        return True
    elif value_str == 'false':
        return False
    elif re.match(r"^[+|-]?\d+.?\d*$", value_str):
        return float(value_str)
    elif re.match(r"^\s*$", value_str):
        return None
    else:
        return value


class ExcelHelper:
    """
    Excel   
    """

    @classmethod
    def convert_2_class(cls, file_path, clazz):
        """
           class，     sheet ，     clazz  
              
        :param file_path:Excel    
        :param clazz:     clazz  
        :return:        ，   [[clazz(),clazz()],[clazz()]]
        """
        if not file_path.endswith(".xlsx"):
            raise ValueError("     .xlsx   Excel  ")
        if not isfile(file_path):
            raise FileNotFoundError("     {0}    ".format(file_path))
        work_book = load_workbook(file_path)
        result = []
        for sheet_name in work_book.sheetnames:
            ws = work_book[sheet_name]

            #     
            table_heads = []
            for title_row in ws.iter_rows(max_row=1):
                for cell in title_row:
                    table_heads.append(cell.value)
            #      
            table = []
            for row in ws.iter_rows(min_row=2):
                #      
                instance = clazz()
                for column_index in range(len(row)):
                    setattr(instance, table_heads[column_index], _convert_value(row[column_index].value))

                #     （      None   ）
                is_valid = False
                for attr in instance.__dict__:
                    if not attr.startswith("_") and instance.__dict__[attr] is not None:
                        is_valid = True
                        break
                if is_valid:
                    table.append(instance)
            result.append(table)
        return result

    @classmethod
    def save(cls, file_path, tables):
        if not file_path.endswith(".xlsx"):
            raise ValueError("     .xlsx   Excel  ")
        work_book = Workbook()
        is_first = True
        for table in tables:
            if is_first:
                ws = work_book.active
                is_first = False
            else:
                ws = work_book.create_sheet()
            #     
            table_heads = []
            for attr in table[0].__dict__:
                #   "_"     
                if not attr.startswith("_"):
                    table_heads.append(attr)
            ws.append(table_heads)

            #     
            for row in table:
                data = []
                for head in table_heads:
                    data.append(getattr(row, head))
                ws.append(data)
        try:
            #          
            folder_index = max(file_path.rfind("\\"), file_path.rfind("/"))
            if folder_index != -1:
                folder_path = file_path[0:folder_index]
                if not os.path.exists(folder_path):
                    os.mkdir(folder_path)
            work_book.save(file_path)
        except Exception:
            raise OSError("  Excel  ")

使用方法

#    
from excel_helper import ExcelHelper

#     
class A:
    name=None
    age=None
    isManager=None

#   Excel  ，            
tables = ExcelHelper.convert_2_class("123.xlsx", A)

#    Excel
a1=A()
table=[a1]
ExcelHelper.save("456.xls", [table])

に注意

このヘルプクラスは@classmethod

このヘルプクラスは反射を使用して実装されるため、ヘッダ名はオブジェクトのフィールド名に1つ1つ対応する必要がある(コードのclass Aと下表「表1-1」)

Excelには複数のテーブル(sheet tab)がありますので、パラメータは対象リストのリストですので、対応関係

現在の読み込みExcelは1つのclassタイプにしか変換できませんので、複数のテーブル構造は複数のテーブル

を使用してください.
表1-1
name
age
isManager
王さん
18
True
Tom
28
FALSE
参考資料
PYTHONでの反射(自己学習)
以上