summaryrefslogtreecommitdiffstats
path: root/src/console/zmq/backend/cffi/message.py
blob: c35decb619723010b4056ff19d90dca053d48c39 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
"""Dummy Frame object"""

# Copyright (C) PyZMQ Developers
# Distributed under the terms of the Modified BSD License.

from ._cffi import ffi, C

import zmq
from zmq.utils.strtypes import unicode

try:
    view = memoryview
except NameError:
    view = buffer

_content = lambda x: x.tobytes() if type(x) == memoryview else x

class Frame(object):
    _data = None
    tracker = None
    closed = False
    more = False
    buffer = None


    def __init__(self, data, track=False):
        try:
            view(data)
        except TypeError:
            raise

        self._data = data

        if isinstance(data, unicode):
            raise TypeError("Unicode objects not allowed. Only: str/bytes, " +
                            "buffer interfaces.")

        self.more = False
        self.tracker = None
        self.closed = False
        if track:
            self.tracker = zmq.MessageTracker()

        self.buffer = view(self.bytes)

    @property
    def bytes(self):
        data = _content(self._data)
        return data

    def __len__(self):
        return len(self.bytes)

    def __eq__(self, other):
        return self.bytes == _content(other)

    def __str__(self):
        if str is unicode:
            return self.bytes.decode()
        else:
            return self.bytes

    @property
    def done(self):
        return True

Message = Frame

__all__ = ['Frame', 'Message']