Source code for queuereader

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
.. module:: queuereader.py
   :platform: Unix, Windows
   :synopsis: Ulyxes - an open source project to drive total stations and
           publish observation results.
           GPL v2.0 license
           Copyright (C) 2010-2020 Zoltan Siki <siki.zoltan@epito.bme.hu>

.. moduleauthor:: Bence Turák <turak.bence@epito.bme.hu>
"""
import queue
from reader import Reader

class QueueReader(Reader):
    """ Class to read queue

        :param qu: queue (Queue), default None and queue is created
        :param name: name of reader (str), default None
        :param filt: list of keys to output (list), deafult None
    """

    def __init__(self, qu=None, name=None, filt=None):
        """ Constuctor
        """
        super().__init__(name, filt)

        if isinstance(qu, queue.Queue):
            self.q = qu
        elif queue is None:
            self.q = queue.Queue()
        else:
            raise TypeError('qu must be Queue type!')

    def GetQueue(self):
        """ method to get queue
            :returns: queue
        """
        return self.q

    def GetLine(self):
        """ Get next item from queue
            :returns: next item from queu or none if queue is empty
        """
        try:
            buf = self.q.get(False)
        except Exception:
            return None     # empty queu
        res = {}
        for key, val in buf.items():
            if self.filt is None or key in self.filt:
                res[key] = val
        return res

    def GetNext(self):
        """ Get next item from queue
            :returns: next
        """
        return self.GetLine()

if __name__ == "__main__":
    from queuewriter import QueueWriter
    from angle import Angle
    qu = queue.Queue()

    quWr = QueueWriter(qu)
    quRe = QueueReader(qu)
    print(quRe.GetNext())
    data = {'hz': Angle(0.12345), 'v': Angle(100.2365, 'GON'), 'dist': 123.6581}
    quWr.WriteData(data)
    print(quRe.GetNext())
    quRe.GetQueue()