diff options
Diffstat (limited to 'nextcloud_files.py')
| -rwxr-xr-x | nextcloud_files.py | 138 |
1 files changed, 69 insertions, 69 deletions
diff --git a/nextcloud_files.py b/nextcloud_files.py index f56860a..8c6aeee 100755 --- a/nextcloud_files.py +++ b/nextcloud_files.py @@ -18,75 +18,75 @@ import os class NextcloudStorage: - def __init__(self): - self.config = [ - # filecount - 'graph_title Nextcloud Files', - 'graph_args --base 1000 -l 0', - 'graph_printf %.0lf', - 'graph_vlabel number of files', - 'graph_info graph showing the number of files', - 'graph_category nextcloud', - 'num_files.label number of files', - 'num_files.info current number of files in the repository', - 'num_files.min 0' - ] - self.result = list() - - def parse_data(self, api_response): - num_files = api_response['ocs']['data']['nextcloud']['storage']['num_files'] - self.result.append('num_files.value %s' % num_files) - - 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 = [ + # filecount + 'graph_title Nextcloud Files', + 'graph_args --base 1000 -l 0', + 'graph_printf %.0lf', + 'graph_vlabel number of files', + 'graph_info graph showing the number of files', + 'graph_category nextcloud', + 'num_files.label number of files', + 'num_files.info current number of files in the repository', + 'num_files.min 0' + ] + self.result = list() + + def parse_data(self, api_response): + num_files = api_response['ocs']['data']['nextcloud']['storage']['num_files'] + self.result.append('num_files.value %s' % num_files) + + 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__": - NextcloudStorage().main() + NextcloudStorage().main() |