Code Duplication    Length = 61-74 lines in 2 locations

opcua/ua/uaprotocol_auto.py 1 location

@@ 297-370 (lines=74) @@
294
295
296
class AttributeWriteMask(IntEnum):
297
    '''
298
    Define bits used to indicate which attributes are writable.
299
300
    :ivar None_:
301
    :vartype None_: 0
302
    :ivar AccessLevel:
303
    :vartype AccessLevel: 1
304
    :ivar ArrayDimensions:
305
    :vartype ArrayDimensions: 2
306
    :ivar BrowseName:
307
    :vartype BrowseName: 4
308
    :ivar ContainsNoLoops:
309
    :vartype ContainsNoLoops: 8
310
    :ivar DataType:
311
    :vartype DataType: 16
312
    :ivar Description:
313
    :vartype Description: 32
314
    :ivar DisplayName:
315
    :vartype DisplayName: 64
316
    :ivar EventNotifier:
317
    :vartype EventNotifier: 128
318
    :ivar Executable:
319
    :vartype Executable: 256
320
    :ivar Historizing:
321
    :vartype Historizing: 512
322
    :ivar InverseName:
323
    :vartype InverseName: 1024
324
    :ivar IsAbstract:
325
    :vartype IsAbstract: 2048
326
    :ivar MinimumSamplingInterval:
327
    :vartype MinimumSamplingInterval: 4096
328
    :ivar NodeClass:
329
    :vartype NodeClass: 8192
330
    :ivar NodeId:
331
    :vartype NodeId: 16384
332
    :ivar Symmetric:
333
    :vartype Symmetric: 32768
334
    :ivar UserAccessLevel:
335
    :vartype UserAccessLevel: 65536
336
    :ivar UserExecutable:
337
    :vartype UserExecutable: 131072
338
    :ivar UserWriteMask:
339
    :vartype UserWriteMask: 262144
340
    :ivar ValueRank:
341
    :vartype ValueRank: 524288
342
    :ivar WriteMask:
343
    :vartype WriteMask: 1048576
344
    :ivar ValueForVariableType:
345
    :vartype ValueForVariableType: 2097152
346
    '''
347
    None_ = 0
348
    AccessLevel = 1
349
    ArrayDimensions = 2
350
    BrowseName = 4
351
    ContainsNoLoops = 8
352
    DataType = 16
353
    Description = 32
354
    DisplayName = 64
355
    EventNotifier = 128
356
    Executable = 256
357
    Historizing = 512
358
    InverseName = 1024
359
    IsAbstract = 2048
360
    MinimumSamplingInterval = 4096
361
    NodeClass = 8192
362
    NodeId = 16384
363
    Symmetric = 32768
364
    UserAccessLevel = 65536
365
    UserExecutable = 131072
366
    UserWriteMask = 262144
367
    ValueRank = 524288
368
    WriteMask = 1048576
369
    ValueForVariableType = 2097152
370
371
372
class BrowseDirection(IntEnum):
373
    '''

opcua/ua/uatypes.py 1 location

@@ 805-865 (lines=61) @@
802
            return VariantType.String
803
        elif isinstance(val, bytes):
804
            return VariantType.ByteString
805
        elif isinstance(val, datetime):
806
            return VariantType.DateTime
807
        elif isinstance(val, uuid.UUID):
808
            return VariantType.Guid
809
        else:
810
            if isinstance(val, object):
811
                try:
812
                    return getattr(VariantType, val.__class__.__name__)
813
                except AttributeError:
814
                    return VariantType.ExtensionObject
815
            else:
816
                raise UaError("Could not guess UA type of {0} with type {1}, specify UA type".format(val, type(val)))
817
818
    def __str__(self):
819
        return "Variant(val:{0!s},type:{1})".format(self.Value, self.VariantType)
820
    __repr__ = __str__
821
822
    def to_binary(self):
823
        b = []
824
        encoding = self.VariantType.value & 0b111111
825
        if self.is_array or type(self.Value) in (list, tuple):
826
            self.is_array = True
827
            encoding = uabin.set_bit(encoding, 7)
828
            if self.Dimensions is not None:
829
                encoding = uabin.set_bit(encoding, 6)
830
            b.append(uabin.Primitives.UInt8.pack(encoding))
831
            b.append(uabin.pack_uatype_array(self.VariantType, flatten(self.Value)))
832
            if self.Dimensions is not None:
833
                b.append(uabin.pack_uatype_array(VariantType.Int32, self.Dimensions))
834
        else:
835
            b.append(uabin.Primitives.UInt8.pack(encoding))
836
            b.append(uabin.pack_uatype(self.VariantType, self.Value))
837
838
        return b"".join(b)
839
840
    @staticmethod
841
    def from_binary(data):
842
        dimensions = None
843
        array = False
844
        encoding = ord(data.read(1))
845
        int_type = encoding & 0b00111111
846
        vtype = datatype_to_varianttype(int_type)
847
        if uabin.test_bit(encoding, 7):
848
            value = uabin.unpack_uatype_array(vtype, data)
849
            array = True
850
        else:
851
            value = uabin.unpack_uatype(vtype, data)
852
        if uabin.test_bit(encoding, 6):
853
            dimensions = uabin.unpack_uatype_array(VariantType.Int32, data)
854
            value = reshape(value, dimensions)
855
        return Variant(value, vtype, dimensions, is_array=array)
856
857
858
def reshape(flat, dims):
859
    subdims = dims[1:]
860
    subsize = 1
861
    for i in subdims:
862
        if i == 0:
863
            i = 1
864
        subsize *= i
865
    while dims[0] * subsize > len(flat):
866
        flat.append([])
867
    if not subdims or subdims == [0]:
868
        return flat