printer: Fix NoneType issue

Fixes #143
This commit is contained in:
Jordan 2021-05-08 22:27:55 -04:00
parent ce8c8e44fc
commit f8ca339de5

View File

@ -221,11 +221,11 @@ class Printer:
def get_stat(self, stat, substat = None):
if stat not in self.data:
return None
return {}
if substat != None:
if substat in self.data[stat]:
return self.data[stat][substat]
return None
return {}
return self.data[stat]
def get_state(self):