深入理解 SortedList 中自定义对象的高效搜索

from typing import List
from sortedcontainers import SortedList

class Supplier:
    def __init__(self, name: str, id: int = 0, sap_id: int = 0):
        self.Name = name
        self.Id = id
        self.SapId = sap_id

    def __repr__(self):
        return f"Supplier(Name='{self.Name}', Id={self.Id})"

class Data:
    def __init__(self):
        # 初始化时指定key，按Supplier的Name属性（小写）排序
        self.suppliers = SortedList(key=lambda x: x.Name.lower())

# 示例数据
data_store = Data()
data_store.suppliers.add(Supplier("Apple", 101, 2001))
data_store.suppliers.add(Supplier("Banana", 102, 2002))
data_store.suppliers.add(Supplier("Cherry", 103, 2003))
print(data_store.suppliers)
# 输出: SortedList([Supplier(Name='Apple', Id=101), Supplier(Name='Banana', Id=102), Supplier(Name='Cherry', Id=103)])

# 假设在Data类中有一个查找方法
# def find_supplier(self, name: str):
#     index = self.suppliers.bisect_left(name.lower()) # 尝试直接传入字符串
#     # ... 后续检查

# 在Data类中
def find_supplier_with_temp_object(self, name: str):
    temporary_supplier = Supplier(name) # 创建一个临时对象
    index = self.suppliers.bisect_left(temporary_supplier)

    if index != len(self.suppliers) and self.suppliers[index].Name.lower() == name.lower():
        return self.suppliers[index]

    return None

# print(data_store.find_supplier_with_temp_object("Apple"))

from typing import List
from sortedcontainers import SortedList

class Supplier:
    def __init__(self, name: str, id: int = 0, sap_id: int = 0):
        self.Name = name
        self.Id = id
        self.SapId = sap_id

    def __repr__(self):
        return f"Supplier(Name='{self.Name}', Id={self.Id})"

    # 重载小于操作符
    def __lt__(self, other):
        if isinstance(other, str):
            # 如果另一个操作数是字符串，则与自己的Name属性进行比较
            return self.Name.lower() < other.lower()
        elif isinstance(other, Supplier):
            # 如果另一个操作数是Supplier对象，则与另一个Supplier的Name属性进行比较
            return self.Name.lower() < other.Name.lower()
        # 处理其他类型或抛出错误，这里简化为默认False
        return NotImplemented # 或者 raise TypeError(f"Cannot compare Supplier with {type(other)}")

    # 重载等于操作符 (推荐实现，确保精确匹配)
    def __eq__(self, other):
        if isinstance(other, str):
            return self.Name.lower() == other.lower()
        elif isinstance(other, Supplier):
            return self.Name.lower() == other.Name.lower()
        return NotImplemented

    # 如果实现了__eq__，通常也建议实现__hash__，除非明确不希望对象可哈希
    # def __hash__(self):
    #     return hash(self.Name.lower())

class Data:
    def __init__(self):
        # 此时SortedList不再需要key函数，因为它存储的对象本身就可比较了
        self.suppliers = SortedList()

    def add_supplier(self, supplier: Supplier):
        self.suppliers.add(supplier)

    def find_supplier(self, name: str):
        # 直接传入字符串进行二分查找
        index = self.suppliers.bisect_left(name)

        # 检查找到的索引是否有效，并且对应元素的名称是否与搜索名称匹配
        if index != len(self.suppliers) and self.suppliers[index].Name.lower() == name.lower():
            return self.suppliers[index]

        return None

# 示例用法
data_store = Data()
data_store.add_supplier(Supplier("Banana", 102, 2002))
data_store.add_supplier(Supplier("Apple", 101, 2001))
data_store.add_supplier(Supplier("Cherry", 103, 2003))

print("排序后的供应商列表:", data_store.suppliers)
# 预期输出: SortedList([Supplier(Name='Apple', Id=101), Supplier(Name='Banana', Id=102), Supplier(Name='Cherry', Id=103)])

found_supplier = data_store.find_supplier("Apple")
print("查找 'Apple':", found_supplier)
# 预期输出: 查找 'Apple': Supplier(Name='Apple', Id=101)

not_found_supplier = data_store.find_supplier("Grape")
print("查找 'Grape':", not_found_supplier)
# 预期输出: 查找 'Grape': None

found_supplier_case_insensitive = data_store.find_supplier("apple")
print("查找 'apple' (不区分大小写):", found_supplier_case_insensitive)
# 预期输出: 查找 'apple' (不区分大小写): Supplier(Name='Apple', Id=101)