使用Python PriorityQueue与Enum时出现TypeError报错求助
问题原因
Python标准库的PriorityQueue基于堆实现,插入元素时会自动按优先级排序,这要求队列中的元素支持<比较操作。你用的普通Enum成员仅支持相等和身份比较,当队列中存在多个不同的Enum实例时,堆排序触发比较就会抛出TypeError。
解决方案
不需要弃用queue模块,也不用全改成整数,两种方案任选:
方案1:改用IntEnum
IntEnum继承自int,成员本质是整数类型,支持所有数值比较操作,直接替换原Enum即可:
import queue as q from enum import IntEnum Commands = IntEnum('Commands', 'power_on power_off etc') queue = q.PriorityQueue(50) def power_on(): queue.put(Commands.power_on)
方案2:存储(优先级数值,Enum实例)元组
如果不想修改Enum类型,存入队列时把元素包装成元组,第一个元素是优先级整数,PriorityQueue会优先比较元组的第一个值,避免直接比较Enum实例:
import queue as q from enum import Enum Commands = Enum('Commands', 'power_on power_off etc') queue = q.PriorityQueue(50) def power_on(): # 第一个值为优先级,可根据需求给不同命令设置不同数值 queue.put((1, Commands.power_on))
补充说明
两种方案都适配PyQt5的线程场景,PriorityQueue本身是线程安全的,不会引入新的线程问题。
内容的提问来源于stack exchange,提问作者B3at
相关产品推荐
相关产品推荐

