diff options
Diffstat (limited to 'nextcloud_dbsize.py')
| -rwxr-xr-x | nextcloud_dbsize.py | 138 |
1 files changed, 69 insertions, 69 deletions
diff --git a/nextcloud_dbsize.py b/nextcloud_dbsize.py index f6cc4e1..5e18244 100755 --- a/nextcloud_dbsize.py +++ b/nextcloud_dbsize.py @@ -18,75 +18,75 @@ import os class NextcloudDB: - def __init__(self): - self.config = [ - # dbsize - 'graph_title Nextcloud Database Size', - 'graph_args --base 1024 -l 0', - 'graph_vlabel size in byte', - 'graph_info graph showing the database size in byte', - 'graph_category nextcloud', - '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' - ] - self.result = list() - - def parse_data(self, api_response): - dbsize = api_response['ocs']['data']['server']['database']['size'] - self.result.append('db_size.value %s' % dbsize) - - def run(self): - # init request session with specific header and credentials - with requests.Session() as s: - # read credentials from env - s.auth = (os.environ.get('username'), os.environ.get('password')) - - # update header for json - s.headers.update({'Accept': 'application/json'}) - - # request the data - r = s.get(os.environ.get('url')) - - # if status code is successful continue - if r.status_code == 200: - self.parse_data(r.json()) - - # output results to stdout - for el in self.result: - print(el, file=sys.stdout) - - 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): - # check if any argument is given - if sys.argv.__len__() >= 2: - # check if first argument is config or autoconf if not fetch data - if sys.argv[1] == "config": - # output config list to stdout - for el in self.config: - print(el, file=sys.stdout) - - # if DIRTYCONFIG true also return the corresponding values - if os.environ.get('MUNIN_CAP_DIRTYCONFIG') == '1': - self.run() - - elif sys.argv[1] == 'autoconf': - if None in [os.environ.get('username'), os.environ.get('password')]: - print('env variables are missing') - else: - print('yes') - else: - self.run() + def __init__(self): + self.config = [ + # dbsize + 'graph_title Nextcloud Database Size', + 'graph_args --base 1024 -l 0', + 'graph_vlabel size in byte', + 'graph_info graph showing the database size in byte', + 'graph_category nextcloud', + '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' + ] + self.result = list() + + def parse_data(self, api_response): + dbsize = api_response['ocs']['data']['server']['database']['size'] + self.result.append('db_size.value %s' % dbsize) + + def run(self): + # init request session with specific header and credentials + with requests.Session() as s: + # read credentials from env + s.auth = (os.environ.get('username'), os.environ.get('password')) + + # update header for json + s.headers.update({'Accept': 'application/json'}) + + # request the data + r = s.get(os.environ.get('url')) + + # if status code is successful continue + if r.status_code == 200: + self.parse_data(r.json()) + + # output results to stdout + for el in self.result: + print(el, file=sys.stdout) + + 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): + # check if any argument is given + if sys.argv.__len__() >= 2: + # check if first argument is config or autoconf if not fetch data + if sys.argv[1] == "config": + # output config list to stdout + for el in self.config: + print(el, file=sys.stdout) + + # if DIRTYCONFIG true also return the corresponding values + if os.environ.get('MUNIN_CAP_DIRTYCONFIG') == '1': + self.run() + + elif sys.argv[1] == 'autoconf': + if None in [os.environ.get('username'), os.environ.get('password')]: + print('env variables are missing') + else: + print('yes') + else: + self.run() if __name__ == "__main__": - NextcloudDB().main() + NextcloudDB().main() |