File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -207,17 +207,19 @@ def __read_values_name(
207207 elif column .type == FIELD_TYPE .YEAR :
208208 return self .packet .read_uint8 () + 1900
209209 elif column .type == FIELD_TYPE .ENUM :
210- return column .enum_values [self .packet .read_uint_by_size (column .size )]
210+ if column .enum_values :
211+ return column .enum_values [self .packet .read_uint_by_size (column .size )]
212+ self .packet .read_uint_by_size (column .size )
213+ return None
211214 elif column .type == FIELD_TYPE .SET :
212215 bit_mask = self .packet .read_uint_by_size (column .size )
213- return (
214- set (
216+ if column . set_values :
217+ return {
215218 val
216219 for idx , val in enumerate (column .set_values )
217- if bit_mask & 2 ** idx
218- )
219- or None
220- )
220+ if bit_mask & (1 << idx )
221+ } or None
222+ return None
221223 elif column .type == FIELD_TYPE .BIT :
222224 return self .__read_bit (column )
223225 elif column .type == FIELD_TYPE .GEOMETRY :
You can’t perform that action at this time.
0 commit comments