Added error handling for bad register types.

If a unknown register type is accessed via read_register or write_register an exception will be generated instead of failing silently.
This commit is contained in:
Erik Bjorge
2017-07-28 15:39:03 -07:00
parent 90dbc0ece0
commit e79f42afef
+6
View File
@@ -247,6 +247,8 @@ class DeviceNotFoundError(RuntimeError):
class RegisterNotFoundError(RuntimeError):
pass
class RegisterTypeNotFoundError(RuntimeError):
pass
class Chipset:
@@ -585,6 +587,8 @@ class Chipset:
reg_value = self.msgbus.msgbus_reg_read( int(reg['port'],16), int(reg['offset'],16) )
elif RegisterType.MM_MSGBUS == rtype:
reg_value = self.msgbus.mm_msgbus_reg_read(int(reg['port'],16), int(reg['offset'],16))
else:
raise RegisterTypeNotFoundError("Register type not found: {}".format(rtype))
return reg_value
@@ -621,6 +625,8 @@ class Chipset:
self.msgbus.msgbus_reg_write( int(reg['port'],16), int(reg['offset'],16), reg_value )
elif RegisterType.MM_MSGBUS == rtype:
self.msgbus.mm_msgbus_reg_write(int(reg['port'],16), int(reg['offset'],16), reg_value)
else:
raise RegisterTypeNotFoundError("Register type not found: {}".format(rtype))
def read_register_dict( self, reg_name):
reg_value = self.read_register(reg_name)