Python difflib.SequenceMatcher删除用户逻辑返回错误结果求助

场景与问题 #

使用SequenceMatcher对比API返回的目标用户名列表（a）和LDAP组当前用户名列表（b），实现用户添加/移除功能：

• 添加功能正常工作
• 移除功能返回结果错误，误将user1纳入删除列表，实际仅需删除b中存在但a中没有的用户

测试数据 #

• 目标列表a：['<strong>user1</strong>', 'user2']
• 当前LDAP列表b：['usera', 'userb', 'userc', 'userd', '<strong>user1</strong>', 'userx', 'usery']

原代码 #

from difflib import SequenceMatcher

def compare_two_lists(a, b, action):
    users_to_add = []
    users_to_delete = []
    if action == 'add':
        s = SequenceMatcher(None, a, b)
        for tag, i1, i2, j1, j2 in s.get_opcodes():
            if tag in ('insert', 'replace'):
                users_to_add.append(b[j1:j2])
        print('Add:')
        print(users_to_add)
    if action == 'delete':
        s = SequenceMatcher(None, b, a)
        print('a:')
        print(a)
        print('b:')
        print(b)
        for tag, i1, i2, j1, j2 in s.get_opcodes():
            if tag == 'delete':
                users_to_delete.append(a[i1:i2])
        print('Delete: ')
        print(users_to_delete)

当前错误输出 #

Delete: [['user1', 'user2']]

错误原因 #

1. 索引对应列表混淆：当创建SequenceMatcher(None, b, a)时，opcodes中的i1/i2是原序列b的索引，j1/j2是目标序列a的索引。原代码错误地从a中取i1/i2切片，完全偏离逻辑。
2. 未处理replace场景：若b中存在被替换的元素，这部分也需要纳入删除列表。
3. 列表嵌套问题：使用append会将切片得到的列表作为单个元素加入，导致结果嵌套，应使用extend扁平化元素。

修正后的代码 #

from difflib import SequenceMatcher

def compare_two_lists(a, b, action):
    users_to_add = []
    users_to_delete = []
    if action == 'add':
        # 保留原正常工作的添加逻辑
        s = SequenceMatcher(None, a, b)
        for tag, i1, i2, j1, j2 in s.get_opcodes():
            if tag in ('insert', 'replace'):
                users_to_add.extend(b[j1:j2])
        print('Add:')
        print(users_to_add)
    if action == 'delete':
        s = SequenceMatcher(None, b, a)
        print('a (目标列表):')
        print(a)
        print('b (当前LDAP列表):')
        print(b)
        for tag, i1, i2, j1, j2 in s.get_opcodes():
            if tag == 'delete':
                # 取b中目标列表a不存在的片段
                users_to_delete.extend(b[i1:i2])
            elif tag == 'replace':
                # 取b中被替换的片段，需移除
                users_to_delete.extend(b[i1:i2])
        print('Delete: ')
        print(users_to_delete)

# 测试执行
a = ['<strong>user1</strong>', 'user2']
b = ['usera', 'userb', 'userc', 'userd', '<strong>user1</strong>', 'userx', 'usery']
compare_two_lists(a, b, 'delete')

修正后输出 #

Delete:
['usera', 'userb', 'userc', 'userd', 'userx', 'usery']

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