diff --git a/SungrowClient/SungrowClient.py b/SungrowClient/SungrowClient.py index 867ebdb..e9dc975 100644 --- a/SungrowClient/SungrowClient.py +++ b/SungrowClient/SungrowClient.py @@ -312,6 +312,24 @@ def getRegisterUnit(self, check_register): return register.get('unit','') return '' + def getRegisterType(self, check_register): + for register in self.registers: + if check_register == register['name']: + return register.get('metric_type','gauge') + for register in self.registers_custom: + if check_register == register['name']: + return register.get('metric_type','gauge') + return 'gauge' + + def getRegisterDescription(self, check_register): + for register in self.registers: + if check_register == register['name']: + return register.get('description','') + for register in self.registers_custom: + if check_register == register['name']: + return register.get('description','') + return '' + def validateLatestScrape(self, check_register): for register, value in self.latest_scrape.items(): if check_register == register: