diff options
Diffstat (limited to 'nextcloud_dbsize')
| -rwxr-xr-x | nextcloud_dbsize | 69 |
1 files changed, 31 insertions, 38 deletions
diff --git a/nextcloud_dbsize b/nextcloud_dbsize index fc4f035..9a15774 100755 --- a/nextcloud_dbsize +++ b/nextcloud_dbsize @@ -29,6 +29,7 @@ class NextcloudDB: 'db_size.label database size in byte', 'db_size.info users connected in the last 5 minutes', 'db_size.draw AREA', + 'db_size.min 0' ] } @@ -45,55 +46,47 @@ class NextcloudDB: return data def run(self): - # read the configuration from munin environment - URL = os.environ['url'] - auth = (os.environ['username'], os.environ['password']) - # init requests session with specific header and credentials with requests.Session() as s: - s.auth = auth + # read credentials from env + s.auth = (os.environ.get('username'), os.environ.get('password')) + + # update header for json s.headers.update({'Accept': 'application/json'}) - s.stream = False - # request data from api - r = s.get(URL) + # request the data + r = s.get(os.environ.get('url')) - # if status code is successful close connection and continue - if r.status_code == 200: - s.close() - api_response = r.json() + # if status code is successful continue + if r.status_code == 200: + api_response = r.json() + result = self.get_data(api_response) - result = self.get_data(api_response) + for key in result.keys(): + print('\n'.join(result[key])) - for key in result.keys(): - print('\n'.join(result[key])) - - elif r.status_code == 996: - print('server error') - elif r.status_code == 997: - print('not authorized') - elif r.status_code == 998: - print('not found') - else: - print('unknown error') + elif r.status_code == 996: + print('server error') + elif r.status_code == 997: + print('not authorized') + elif r.status_code == 998: + print('not found') + else: + print('unknown error') def main(self): - if (sys.argv.__len__() == 2) and (sys.argv[1] == "config"): - for key in self.config().keys(): - print('\n'.join(self.config()[key])) - try: - if os.environ['MUNIN_CAP_DIRTYCONFIG'] == '1': + # check if first argument is config or autoconf if not fetch data + if sys.argv.__len__() >= 2: + if sys.argv[1] == "config": + for key in self.config().keys(): + print('\n'.join(self.config()[key])) + if os.environ.get('MUNIN_CAP_DIRTYCONFIG') == '1': self.run() - except KeyError: - pass - - elif (sys.argv.__len__() == 2) and (sys.argv[1] == 'autoconf'): - # check host if env variables are set - try: - if None not in {os.environ['url'], os.environ['username'], os.environ['password']}: + elif sys.argv[1] == 'autoconf': + if None in [os.environ.get('username'), os.environ.get('password')]: + print('env variables are missing') + else: print('yes') - except KeyError: - print('no env configuration options are missing') else: self.run() |