BLAHAJ
/
libsigrok
10
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

python: Add a ConfigKey class.

This commit is contained in:
Martin Ling 2013-04-24 17:47:40 +01:00 committed by Uwe Hermann
parent 05cfe1147a
commit f245b7669e
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
bindings/python/sigrok/core/classes.py
View File

@ -24,7 +24,7 @@ from . import lowlevel
import itertools import itertools
__all__ = ['Error', 'Context', 'Driver', 'Device', 'Session', 'Packet', 'Log', __all__ = ['Error', 'Context', 'Driver', 'Device', 'Session', 'Packet', 'Log',
'LogLevel', 'PacketType', 'Quantity', 'Unit', 'QuantityFlag'] 'LogLevel', 'PacketType', 'Quantity', 'Unit', 'QuantityFlag', 'ConfigKey']
class Error(Exception): class Error(Exception):
@ -35,12 +35,6 @@ def check(result):
if result != SR_OK: if result != SR_OK:
raise Error(result) raise Error(result)
def config_key(name):
if not name.lower() == name:
raise AttributeError
key_name = "SR_CONF_" + name.upper()
return getattr(lowlevel, key_name)
def gvariant_to_python(value): def gvariant_to_python(value):
type_string = g_variant_get_type_string(value) type_string = g_variant_get_type_string(value)
if type_string == 't': if type_string == 't':
@ -145,10 +139,11 @@ class Device(object):
self.struct = struct self.struct = struct
def __getattr__(self, name): def __getattr__(self, name):
key = config_key(name) key = getattr(ConfigKey, name.upper())
data = new_gvariant_ptr_ptr() data = new_gvariant_ptr_ptr()
try: try:
check(sr_config_get(self.driver.struct, key, data, self.struct)) check(sr_config_get(self.driver.struct,
key.id, data, self.struct))
except Error as error: except Error as error:
if error.errno == SR_ERR_NA: if error.errno == SR_ERR_NA:
raise NotImplementedError( raise NotImplementedError(
@ -160,11 +155,12 @@ class Device(object):
def __setattr__(self, name, value): def __setattr__(self, name, value):
try: try:
key = config_key(name) key = getattr(ConfigKey, name.upper())
except AttributeError: except AttributeError:
super(Device, self).__setattr__(name, value) super(Device, self).__setattr__(name, value)
return return
check(sr_config_set(self.struct, key, python_to_gvariant(value))) check(sr_config_set(self.struct,
key.id, python_to_gvariant(value)))
@property @property
def vendor(self): def vendor(self):
@ -327,13 +323,17 @@ class QuantityFlag(EnumValue):
mask = new_mask mask = new_mask
return result return result
class ConfigKey(EnumValue):
pass
for symbol_name in dir(lowlevel): for symbol_name in dir(lowlevel):
for prefix, cls in [ for prefix, cls in [
('SR_LOG_', LogLevel), ('SR_LOG_', LogLevel),
('SR_DF_', PacketType), ('SR_DF_', PacketType),
('SR_MQ_', Quantity), ('SR_MQ_', Quantity),
('SR_UNIT_', Unit), ('SR_UNIT_', Unit),
('SR_MQFLAG_', QuantityFlag)]: ('SR_MQFLAG_', QuantityFlag),
('SR_CONF_', ConfigKey)]:
if symbol_name.startswith(prefix): if symbol_name.startswith(prefix):
name = symbol_name[len(prefix):] name = symbol_name[len(prefix):]
value = getattr(lowlevel, symbol_name) value = getattr(lowlevel, symbol_name)