1
|
|
|
""" |
2
|
|
|
Element callable. |
3
|
|
|
|
4
|
|
|
Call a function to validate data. |
5
|
|
|
|
6
|
|
|
.. code-block:: python |
7
|
|
|
|
8
|
|
|
from binascii import crc32 |
9
|
|
|
|
10
|
|
|
ExampleMessage = Message('VarTest', [('x', 'B'), ('y', 'B')]) |
11
|
|
|
|
12
|
|
|
CRCedMessage = Message('CRCedMessage', [ |
13
|
|
|
('data', ExampleMessage), # A data field that has the example message in it |
14
|
|
|
('crc', 'I', crc32, ['data']), # Crc the data, and give an error if we have something unexpected |
15
|
|
|
('crc', 'I', crc32, ['data'], False), # Crc the data, but don't give an error |
16
|
|
|
]) |
17
|
|
|
|
18
|
|
|
Following creating this message, you have two options: |
19
|
|
|
|
20
|
|
|
1. Specify a value. The function will be used to validate the value. |
21
|
|
|
|
22
|
|
|
.. code-block:: python |
23
|
|
|
|
24
|
|
|
def adder(x, y): |
25
|
|
|
return x + y |
26
|
|
|
|
27
|
|
|
AdderMessage = Message('AdderMessage', [ |
28
|
|
|
('item_a', 'H'), |
29
|
|
|
('item_b', 'B'), |
30
|
|
|
('function_data', 'I', adder, ['item_a', 'item_b']), |
31
|
|
|
]) |
32
|
|
|
|
33
|
|
|
test_data = { |
34
|
|
|
'item_a': 2, |
35
|
|
|
'item_b': 5, |
36
|
|
|
'function_data': 7, |
37
|
|
|
} |
38
|
|
|
|
39
|
|
|
made = AdderMessage.make(test_data) |
40
|
|
|
assert made.item_a == 2 |
41
|
|
|
assert made.item_b == 5 |
42
|
|
|
assert made.function_data == 7 |
43
|
|
|
|
44
|
|
|
# If you specify the wrong value, you'll get a ValueError |
45
|
|
|
test_data = { |
46
|
|
|
'item_a': 2, |
47
|
|
|
'item_b': 5, |
48
|
|
|
'function_data': 33, |
49
|
|
|
} |
50
|
|
|
|
51
|
|
|
try: |
52
|
|
|
made = AdderMessage.make(test_data) |
53
|
|
|
except ValueError: |
54
|
|
|
print('Told you so') |
55
|
|
|
|
56
|
|
|
# Unless you specify `False` in your original item, then |
57
|
|
|
# nobody will care. |
58
|
|
|
|
59
|
|
|
2. Use the function to generate a value. |
60
|
|
|
|
61
|
|
|
.. code-block:: python |
62
|
|
|
|
63
|
|
|
def adder(x, y): |
64
|
|
|
return x + y |
65
|
|
|
|
66
|
|
|
AdderMessage = Message('AdderMessage', [ |
67
|
|
|
('item_a', 'H'), |
68
|
|
|
('item_b', 'B'), |
69
|
|
|
('function_data', 'I', adder, ['item_a', 'item_b']), |
70
|
|
|
]) |
71
|
|
|
|
72
|
|
|
test_data = { |
73
|
|
|
'item_a': 2, |
74
|
|
|
'item_b': 5, |
75
|
|
|
} |
76
|
|
|
|
77
|
|
|
made = AdderMessage.make(test_data) |
78
|
|
|
assert made.item_a == 2 |
79
|
|
|
assert made.item_b == 5 |
80
|
|
|
assert made.function_data == 7 |
81
|
|
|
|
82
|
|
|
""" |
83
|
|
|
|
84
|
|
|
import copy |
85
|
|
|
import struct |
86
|
|
|
|
87
|
|
|
from typing import Optional |
|
|
|
|
88
|
|
|
|
89
|
|
|
from starstruct.element import register, Element |
90
|
|
|
from starstruct.modes import Mode |
91
|
|
|
|
92
|
|
|
|
93
|
|
|
@register |
|
|
|
|
94
|
|
|
class ElementCallable(Element): |
95
|
|
|
""" |
96
|
|
|
Initialize a StarStruct element object. |
97
|
|
|
|
98
|
|
|
:param field: The fields passed into the constructor of the element |
99
|
|
|
:param mode: The mode in which to pack the bytes |
100
|
|
|
:param alignment: Number of bytes to align to |
101
|
|
|
""" |
102
|
|
|
def __init__(self, field: list, mode: Optional[Mode]=Mode.Native, alignment: Optional[int]=1): |
103
|
|
|
# All of the type checks have already been performed by the class |
104
|
|
|
# factory |
105
|
|
|
self.name = field[0] |
106
|
|
|
|
107
|
|
|
# Callable elements use the normal struct packing method |
108
|
|
|
self.format = field[1] |
109
|
|
|
self._func_ref = field[2] |
110
|
|
|
self._func_args = field[3] |
111
|
|
|
|
112
|
|
|
if len(field) == 5: |
113
|
|
|
self._error_on_bad_result = field[4] |
114
|
|
|
else: |
115
|
|
|
self._error_on_bad_result = True |
116
|
|
|
|
117
|
|
|
self.update(mode, alignment) |
118
|
|
|
|
119
|
|
|
@property |
120
|
|
|
def _struct(self): |
121
|
|
|
return struct.Struct(self._mode.value + self.format) |
122
|
|
|
|
123
|
|
|
@staticmethod |
124
|
|
|
def valid(field: tuple) -> bool: |
125
|
|
|
""" |
126
|
|
|
See :py:func:`starstruct.element.Element.valid` |
127
|
|
|
|
128
|
|
|
:param field: The items to determine the structure of the element |
129
|
|
|
""" |
130
|
|
|
return len(field) >= 4 \ |
131
|
|
|
and isinstance(field[0], str) \ |
132
|
|
|
and isinstance(field[1], str) \ |
133
|
|
|
and callable(field[2]) \ |
134
|
|
|
and isinstance(field[3], list) |
135
|
|
|
|
136
|
|
|
def validate(self, msg): |
137
|
|
|
""" |
138
|
|
|
Ensure that the supplied message contains the required information for |
139
|
|
|
this element object to operate. |
140
|
|
|
|
141
|
|
|
All elements that are Variable must reference valid Length elements. |
142
|
|
|
""" |
143
|
|
|
# TODO: Validate the object |
144
|
|
|
self._elements = msg |
|
|
|
|
145
|
|
|
|
146
|
|
|
if not all(k in msg |
147
|
|
|
for k in [arg if isinstance(arg, str) else arg.decode('utf-8') |
148
|
|
|
for arg in self._func_args]): |
149
|
|
|
raise ValueError('Need all keys to be in the message') |
150
|
|
|
|
151
|
|
|
pass |
|
|
|
|
152
|
|
|
|
153
|
|
|
def update(self, mode=None, alignment=None): |
154
|
|
|
"""change the mode of the struct format""" |
155
|
|
|
if mode: |
156
|
|
|
self._mode = mode |
157
|
|
|
|
158
|
|
|
if alignment: |
159
|
|
|
self._alignment = alignment |
160
|
|
|
|
161
|
|
|
def pack(self, msg): |
162
|
|
|
"""Pack the provided values into the supplied buffer.""" |
163
|
|
|
return self._struct.pack(self.make(msg)) |
164
|
|
|
|
165
|
|
|
def unpack(self, msg, buf): |
166
|
|
|
"""Unpack data from the supplied buffer using the initialized format.""" |
167
|
|
|
ret = self._struct.unpack_from(buf) |
168
|
|
|
if isinstance(ret, (list, tuple)): |
169
|
|
|
# TODO: I don't know if there is a case where we want to keep |
170
|
|
|
# it as a list... but for now I'm just going to do this |
171
|
|
|
ret = ret[0] |
172
|
|
|
|
173
|
|
|
# Only check for errors if they haven't told us not to |
174
|
|
|
if self._error_on_bad_result: |
175
|
|
|
# Pretend we're getting a dictionary to make our item, |
176
|
|
|
# but it has no reference to `self`. This is so we check |
177
|
|
|
# for errors correctly. |
178
|
|
|
temp_dict = copy.deepcopy(msg._asdict()) |
179
|
|
|
temp_dict.pop(self.name) |
180
|
|
|
expected_value = self.make(temp_dict) |
181
|
|
|
|
182
|
|
|
# Check for an error |
183
|
|
|
if expected_value != ret: |
184
|
|
|
raise ValueError('Expected value was: {0}, but got: {1}'.format( |
185
|
|
|
expected_value, |
186
|
|
|
ret, |
187
|
|
|
)) |
188
|
|
|
|
189
|
|
|
return (ret, buf[self._struct.size:]) |
190
|
|
|
|
191
|
|
|
def make(self, msg): |
192
|
|
|
"""Return the expected "made" value""" |
193
|
|
|
# If we aren't going to error on a bad result |
194
|
|
|
# and our name is in the message, just send the value |
195
|
|
|
# No need to do extra work. |
196
|
|
|
if not self._error_on_bad_result \ |
197
|
|
|
and self.name in msg \ |
198
|
|
|
and msg[self.name] is not None: |
199
|
|
|
return msg[self.name] |
200
|
|
|
|
201
|
|
|
items = [] |
202
|
|
|
|
203
|
|
|
for reference in self._func_args: |
204
|
|
|
if isinstance(reference, str): |
205
|
|
|
index = reference |
206
|
|
|
attr = 'make' |
207
|
|
|
elif isinstance(reference, bytes): |
208
|
|
|
index = reference.decode('utf-8') |
209
|
|
|
attr = 'pack' |
210
|
|
|
else: |
211
|
|
|
raise ValueError('Needed str or bytes for the reference') |
212
|
|
|
|
213
|
|
|
items.append( |
214
|
|
|
getattr(self._elements[index], attr)(msg) |
215
|
|
|
) |
216
|
|
|
|
217
|
|
|
ret = self._func_ref(*items) |
|
|
|
|
218
|
|
|
|
219
|
|
|
if self.name in msg: |
220
|
|
|
if ret != msg[self.name]: |
221
|
|
|
raise ValueError('Excepted value: {0}, but got: {1}'.format(ret, msg[self.name])) |
222
|
|
|
|
223
|
|
|
return ret |
224
|
|
|
|
This can be caused by one of the following:
1. Missing Dependencies
This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands.
2. Missing __init__.py files
This error could also result from missing
__init__.py
files in your module folders. Make sure that you place one file in each sub-folder.