可修改内容的优先级队列

• 以优先级入队，即入队前要求队列已排序，从而确定当前优先级所在位置。同优先级按先后次序入队。
• 可由管理员对队列内容进行修改，修改时应暂时锁住队列。
• 以优先级出队，同优先级按当前位置（即入队顺序）出队(若已排序，则可直接出队操作而不需再判断)。
• 采用数组存字典的形式，模拟队列 {"pri":0, "msg":"txt"}
• 功能
 a. 增
 可插入数据（单个或全部）
 b. 删
 可删除指定 优先级 的数据（单个或全部）
 c. 改
 可修改指定 优先级和内容 的数据（单个）
 d. 查
 可查询指定 优先级 的数据（单个或全部）

# coding:utf-8

'''
    • 以优先级入队，即入队前要求队列已排序，从而确定当前优先级所在位置。同优先级按先后次序入队。
    • 可由管理员对队列内容进行修改，修改时应暂时锁住队列。
    • 以优先级出队，同优先级按当前位置（即入队顺序）出队(若已排序，则可直接出队操作而不需再判断)。
    • 采用数组存字典的形式，模拟队列 {"pri":0, "msg":"txt"}
    • 功能
        a. 增
        可插入数据（单个或全部）
        b. 删
        可删除指定 优先级 的数据（单个或全部）
        c. 改
        可修改指定 优先级和内容 的数据（单个）
        d. 查
        可查询指定 优先级 的数据（单个或全部）
'''
class enhanceQueue:
    def __init__(self):
        self.queue = []
        self.length = 0
        self.unlock = True

    # 批量插入数据，其中 arr -> [(pri, 'msg'), (pri, 'msg'), (pri, 'msg')]
    def Create(self, arr):
        self.queue.clear()
        for item in arr:
            priority = item[0]
            value = item[1]
            self.Push(priority, value)
        return self.queue

    # 插入单个数据
    def Push(self, priority, value):
        content = {"pri": priority, "msg": value}
        if self.unlock:
            self.unlock = False
            if self.queue:
                for i in range(len(self.queue)):
                    if self.queue[i]['pri'] > priority:
                        self.queue.insert(i, content)
                        self.length += 1
                        break
                    elif i+1 == len(self.queue):
                        self.queue.append(content)
                        self.length += 1
                        break
                    elif self.queue[i]['pri'] <= priority:
                        continue
                self.unlock = True
            else:
                self.queue.append(content)
                self.length += 1
                self.unlock = True

    # 出队操作，以优先级出队，同优先级按当前位置（即入队顺序）出队
    def Pop(self):
        if self.queue and self.unlock:
            self.unlock = False
            self.length -= 1
            pop = self.queue.pop(0)
            self.unlock = True
            return pop
        else:
            return None

    # 删除指定 优先级 的数据（其中，isall表示单个或全部同优先级）
    def Delete(self, priority, isall=False):
        if self.queue and self.unlock:
            self.unlock = False
            temp = []
            for item in self.queue:
                if item['pri'] == priority:
                    temp.append(item)
                    if isall:
                        continue
                    else:
                        break
            for i in temp:
                self.queue.remove(i)
                self.length -= 1
            self.unlock = True

    # 查找指定 优先级 的数据（其中，isall表示单个或全部同优先级）
    def Search(self, priority, isall=False):
        temp = []
        if self.queue and self.unlock:
            self.unlock = False
            for item in self.queue:
                if item['pri'] == priority:
                    temp.append(item)
                    if isall:
                        continue
                    else:
                        break
            self.unlock = True
        return temp

    # 修改指定 优先级和内容 的数据
    def Modify(self, priority, oldvalue, newValue):
        if self.queue and self.unlock:
            self.unlock = False
            for item in self.queue:
                if item['pri'] == priority and item['msg'] == oldvalue:
                    item['msg'] = newValue
                    self.unlock = True
                    return True
            self.unlock = True
        return False

    # 打印当前队列，也可直接调用 obj.queue查看
    def Display(self):
        for i in self.queue:
            print(i)

    # 查询是否是锁定状态
    def IsLock(self):
        return self.unlock

    # 清空整个队列
    def ClearAll(self):
        if self.unlock:
            self.unlock = False
            self.queue.clear()
            self.unlock = True


def test():
    queue = enhanceQueue()
    print('*' * 40)
    print(">> 批量创建队列")
    print("输入的数据: [(3, 'a'), (1, 'b'), (5, 'c'), (2, 'c'), (7, 'c'), (5, 'c')]")
    queue.Create([(3, 'a'), (1, 'b'), (5, 'c'), (2, 'c'), (7, 'c'), (5, 'c')])
    print("输出的数据: ", queue.queue)

    print('*' * 40)
    print(">> 清空整个队列")
    queue.ClearAll()
    print("输出的数据: ", queue.queue)

    print('*' * 40)
    print(">> 单个输入数据")
    print("输入的数据: (3, 'a'), (1, 'b'), (5, 'c'), (2, 'c'), (7, 'c'), (5, 'c')")
    queue.Push(3, 'a')
    queue.Push(1, 'b')
    queue.Push(5, 'c')
    queue.Push(2, 'd')
    queue.Push(7, 'e')
    queue.Push(5, 'f')
    print("输出的数据: ", queue.queue)

    print('*' * 40)
    print(">> 出队一个数据")
    queue.Pop()
    print("剩余的数据: ", queue.queue)

    print('*' * 40)
    print(">> 搜索指定优先级的单个数据, 如5")
    print("剩余的数据: ", queue.Search(5))

    print('*' * 40)
    print(">> 搜索指定优先级的全部数据, 如5")
    print("剩余的数据: ", queue.Search(5, True))

    print('*' * 40)
    print(">> 修改指定 优先级和内容 的数据, 如5,c->5,o")
    queue.Modify(5, 'c', 'o')
    print("修改后的数据: ", queue.Search(5))

    print('*' * 40)
    print(">> 删除指定优先级的单个数据, 如2")
    queue.Delete(5)
    print("删除后的数据: ", queue.queue)

    print('*' * 40)
    print(">> 删除指定优先级的全部数据, 如5")
    queue.Delete(5, True)
    print("删除后的数据: ", queue.queue)

if __name__ == '__main__':
    test()




****************************************
>> 批量创建队列
输入的数据: [(3, 'a'), (1, 'b'), (5, 'c'), (2, 'c'), (7, 'c'), (5, 'c')]
输出的数据:  [{'pri': 1, 'msg': 'b'}, {'pri': 2, 'msg': 'c'}, {'pri': 3, 'msg': 'a'}, {'pri': 5, 'msg': 'c'}, {'pri': 5, 'msg': 'c'}, {'pri': 7, 'msg': 'c'}]
****************************************
>> 清空整个队列
输出的数据:  []
****************************************
>> 单个输入数据
输入的数据: (3, 'a'), (1, 'b'), (5, 'c'), (2, 'c'), (7, 'c'), (5, 'c')
输出的数据:  [{'pri': 1, 'msg': 'b'}, {'pri': 2, 'msg': 'd'}, {'pri': 3, 'msg': 'a'}, {'pri': 5, 'msg': 'c'}, {'pri': 5, 'msg': 'f'}, {'pri': 7, 'msg': 'e'}]
****************************************
>> 出队一个数据
剩余的数据:  [{'pri': 2, 'msg': 'd'}, {'pri': 3, 'msg': 'a'}, {'pri': 5, 'msg': 'c'}, {'pri': 5, 'msg': 'f'}, {'pri': 7, 'msg': 'e'}]
****************************************
>> 搜索指定优先级的单个数据, 如5
剩余的数据:  [{'pri': 5, 'msg': 'c'}]
****************************************
>> 搜索指定优先级的全部数据, 如5
剩余的数据:  [{'pri': 5, 'msg': 'c'}, {'pri': 5, 'msg': 'f'}]
****************************************
>> 修改指定 优先级和内容 的数据, 如5,c->5,o
修改后的数据:  [{'pri': 5, 'msg': 'o'}]
****************************************
>> 删除指定优先级的单个数据, 如2
删除后的数据:  [{'pri': 2, 'msg': 'd'}, {'pri': 3, 'msg': 'a'}, {'pri': 5, 'msg': 'f'}, {'pri': 7, 'msg': 'e'}]
****************************************
>> 删除指定优先级的全部数据, 如5
删除后的数据:  [{'pri': 2, 'msg': 'd'}, {'pri': 3, 'msg': 'a'}, {'pri': 7, 'msg': 'e'}]