diff options
Diffstat (limited to 'nextcloud_shares')
-rwxr-xr-x | nextcloud_shares | 78 |
1 files changed, 37 insertions, 41 deletions
diff --git a/nextcloud_shares b/nextcloud_shares index 8e5173f..dc56fba 100755 --- a/nextcloud_shares +++ b/nextcloud_shares @@ -23,24 +23,31 @@ class NextcloudShares: config = { 'shares': [ 'graph_title Nextcloud Shares', - 'graph_args --base 1024 -l 0', + 'graph_args --base 1000 -l 0', 'graph_vlabel number of shares', 'graph_info graph showing the number of shares', 'graph_category nextcloud', 'num_fed_shares_received.label federated shares recieved', 'num_fed_shares_received.info current total of federated shares recieved', + 'num_fed_shares_received.min 0', 'num_fed_shares_sent.label federated shares sent', 'num_fed_shares_sent.info current total of federated shares sent', + 'num_fed_shares_sent.min 0', 'num_shares.label total number of shares', 'num_shares.info current over all total of shares', + 'num_shares.min 0', 'num_shares_groups.label group shares', 'num_shares_groups.info current total of group shares', + 'num_shares_groups.min 0', 'num_shares_link.label link shares', 'num_shares_link.info current total of shares through a link', + 'num_shares_link.min 0', 'num_shares_link_no_password.label link shares without a password', 'num_shares_link_no_password.info current total of shares through a link without a password protection', + 'num_shares_link_no_password.min 0', 'num_shares_user.label user shares', - 'num_shares_user.info current total of user shares' + 'num_shares_user.info current total of user shares', + 'num_shares_user.min 0' ] } @@ -50,41 +57,35 @@ class NextcloudShares: data ={ 'nextcloud_shares': [], } - shares = api_response['ocs']['data']['nextcloud']['shares'] - # use regex to remove permission stats from api response - reg = re.compile("num.*") - share_keys = shares.keys() - sharelist = list(filter(reg.match, share_keys)) + # shares + shares = api_response['ocs']['data']['nextcloud']['shares'] + data['nextcloud_shares'].append('multigraph nextcloud_shares') - for key in sharelist: - data['nextcloud_shares'].append(str(key) + ".value " + str(shares[key])) + # append for every key in shares the key and the value if the key starts with "num" + [data['nextcloud_shares'].append(str(key) + ".value " + str(shares[key])) + for key in shares if key.startswith('num')] 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 + # init request 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 status code is successful continue if r.status_code == 200: - s.close() - api_response = r.json() + result = self.get_data(r.json()) - result = self.get_data(api_response) - - for key in result.keys(): - print('\n'.join(result[key])) + # for key in results print every entry in dict + [print('\n'.join(result[key])) for key in result.keys()] elif r.status_code == 996: print('server error') @@ -96,26 +97,21 @@ class NextcloudShares: 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 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": + # for key in config().keys() print every entry in dict + [print('\n'.join(self.config()[key])) for key in self.config().keys()] + 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() - if __name__ == "__main__": NextcloudShares().main() - quit(0) |