Python3技术问询：Excel数据导入Shelve及Apple序列号模块改写

针对你的两个问题，我整理了具体的实现方案和代码示例，希望能帮到你：

Shelve是Python自带的轻量级键值存储工具，把Excel数据导入进去的核心思路是先读取Excel数据，再将数据以键值对的形式存入Shelve。这里提供两种常用实现方式：

方式一：用Pandas快速处理（推荐，适合数据量大的情况） #

Pandas能轻松读取各种格式的Excel文件，处理起来更简洁：

1. 先安装依赖库：

pip install pandas openpyxl

（openpyxl用于处理.xlsx格式，如果你是.xls文件，可安装xlrd==1.2.0，新版本xlrd不支持xls格式）

2. 编写导入代码：

import pandas as pd
import shelve

# 读取Excel文件，替换成你的文件路径
df = pd.read_excel("你的设备数据.xlsx", engine="openpyxl")

# 使用with语句自动管理Shelve文件，避免忘记关闭
with shelve.open("device_shelf_db") as db:
    # 假设Excel里有"序列号"列作为唯一键，将每行数据转为字典存入
    for _, row in df.iterrows():
        serial_key = row["序列号"]
        db[serial_key] = row.to_dict()

print("Excel数据已成功导入Shelve！")

方式二：用OpenPyXL手动读取（适合需要精细控制的场景） #

如果不想用Pandas，直接用openpyxl操作Excel单元格：

1. 安装依赖：

pip install openpyxl

2. 实现代码：

import openpyxl
import shelve

# 加载Excel工作簿
wb = openpyxl.load_workbook("你的设备数据.xlsx")
sheet = wb.active  # 获取当前活跃工作表

# 提取表头（第一行）
headers = [cell.value for cell in sheet[1]]

with shelve.open("device_shelf_db") as db:
    # 从第二行开始遍历数据行
    for row_data in sheet.iter_rows(min_row=2, values_only=True):
        # 将行数据转为字典
        row_dict = dict(zip(headers, row_data))
        # 以序列号为键存入Shelve
        db[row_dict["序列号"]] = row_dict

wb.close()
print("数据导入完成！")

你的需求很明确：优先用自己的数据库返回详细信息，原模块作为兜底。不建议直接修改原模块的源码（会影响后续更新），更优雅的方式是写一个「包装函数」或者「子类继承」来扩展功能：

步骤1：准备你的自定义数据库 #

可以用第一个问题的方法，把你的序列号-机型详细信息导入Shelve（比如命名为my_custom_mac_db），数据结构建议是：

# 示例数据，键是序列号前缀，值是详细信息字典
{
    "W88": {"产品类型": "MacBook Pro 16-inch 2023", "处理器型号": "M3 Pro", "运行速度": "3.5GHz"},
    "X12": {"产品类型": "Mac mini 2024", "处理器型号": "M4", "运行速度": "3.7GHz"}
}

步骤2：编写包装函数扩展原模块 #

假设原模块的核心函数是get_device_info(serial)，我们写一个自定义函数优先查自己的数据库：

from mac_model_shelf import get_device_info as original_get_info
import shelve

def custom_mac_info(serial_number):
    # 1. 先检查自定义数据库
    with shelve.open("my_custom_mac_db") as db:
        # 按前缀匹配（注意：建议把长前缀放前面，避免短前缀先匹配）
        # 比如如果有"W881"和"W88"，先检查"W881"
        sorted_prefixes = sorted(db.keys(), key=len, reverse=True)
        for prefix in sorted_prefixes:
            if serial_number.startswith(prefix):
                return db[prefix]
    
    # 2. 自定义数据库没匹配到，调用原模块的逻辑
    original_result = original_get_info(serial_number)
    # 可选：如果原结果缺少字段，你可以在这里补充默认值
    # original_result["处理器型号"] = "未查询到详细信息"
    return original_result

# 测试使用
test_serial = "W88XXXXXXXXX"
print(custom_mac_info(test_serial))  # 会返回你自定义的详细信息
test_old_serial = "X09XXXXXXXXX"
print(custom_mac_info(test_old_serial))  # 会返回原模块的结果

步骤3：如果原模块是类结构，用子类继承 #

如果原模块是基于类实现的（比如MacModelShelf类），可以继承并重写方法：

from mac_model_shelf import MacModelShelf
import shelve

class CustomMacModelShelf(MacModelShelf):
    def get_info(self, serial_number):
        # 先查自定义数据库
        with shelve.open("my_custom_mac_db") as db:
            sorted_prefixes = sorted(db.keys(), key=len, reverse=True)
            for prefix in sorted_prefixes:
                if serial_number.startswith(prefix):
                    return db[prefix]
        # 调用父类的原方法
        return super().get_info(serial_number)

# 使用示例
shelf = CustomMacModelShelf()
print(shelf.get_info("W88XXXXXXXXX"))

注意事项 #

• 确保自定义数据库的前缀没有冲突，比如不要同时存在"W88"和"W881"这样的前缀（除非按长度从长到短检查，上面的代码已经处理了）。
• 如果原模块有缓存机制，记得在自定义逻辑里同步处理缓存，避免旧数据干扰。
• 测试时覆盖两种场景：匹配自定义前缀的序列号、不匹配的旧款Mac序列号，确保都能正常返回。

内容的提问来源于stack exchange，提问作者William Lombard