108 lines
3.1 KiB
Python
Executable File
108 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
'''
|
|
=head1 NAME
|
|
|
|
Munin Plugin to grab signature count for change.org petition given its ID
|
|
You need a valid API key for this petition
|
|
|
|
=head2 CONFIGURATION
|
|
|
|
Example:
|
|
|
|
[changeorg_signature_count]
|
|
env.APIkey xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
|
|
env.petitions 1727001 1727053
|
|
|
|
=head1 AUTHOR
|
|
|
|
(c) 2017 Raphaël Droz <raphael.droz+floss@gmail.com>
|
|
|
|
=head1 LICENSE
|
|
|
|
General Public Licence v3 or later
|
|
|
|
SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
=head1 MAGIC MARKERS
|
|
|
|
#%# family=auto
|
|
#%# capabilities=autoconf
|
|
|
|
=cut
|
|
'''
|
|
|
|
from sys import argv, exit
|
|
import sys
|
|
import codecs
|
|
from os import environ, path, umask
|
|
import re
|
|
import urllib.request
|
|
import json
|
|
|
|
if len(argv) > 1 and argv[1] == 'autoconf':
|
|
ok = True
|
|
if not environ.get('APIkey') or not re.match('[a-fA-F0-9]{64}$', environ.get('APIkey')):
|
|
print("no (env.APIkey not defined or bad format)")
|
|
ok = False
|
|
for i in environ.get("petitions"):
|
|
if not re.match("[0-9]+$", i):
|
|
print("no ($i isn't a valid petition ID")
|
|
ok = False
|
|
if ok:
|
|
print("yes")
|
|
exit(0)
|
|
|
|
|
|
petition_titles = {}
|
|
write_cache = False
|
|
|
|
if environ.get('MUNIN_PLUGSTATE'):
|
|
petition_cache_names = path.join(environ['MUNIN_PLUGSTATE'], path.basename(argv[0]) + '-petition_names')
|
|
try:
|
|
with open(petition_cache_names, 'r') as f:
|
|
petition_titles = json.load(f)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
for i in environ.get('petitions').split():
|
|
if i in petition_titles:
|
|
continue
|
|
# NB: user-agent's tweak is needed to avoid a 403
|
|
req = urllib.request.Request("https://api.change.org/v1/petitions/{}?api_key={}&fields=title".format(i, environ.get('APIkey')),
|
|
data=None,
|
|
headers={ 'User-Agent': 'curl/7.38.0' })
|
|
response = urllib.request.urlopen(req).read().decode('utf-8')
|
|
petition_titles[i] = json.loads(response)
|
|
write_cache = True
|
|
|
|
if environ.get('MUNIN_PLUGSTATE') and write_cache:
|
|
umask(0o077)
|
|
with open(petition_cache_names, 'w') as outfile:
|
|
json.dump(petition_titles, outfile)
|
|
|
|
# equivalent of passing PYTHONIOENCODING=utf-8 to munin
|
|
sys.stdout = codecs.getwriter("utf-8")(sys.stdout.detach())
|
|
|
|
if len(argv) > 1 and argv[1] == 'config':
|
|
print('''graph_title change.org signature count
|
|
graph_args --base 1000 -l 0
|
|
graph_vlabel Signatures
|
|
graph_category other
|
|
graph_info change.org signature count
|
|
graph_period minute
|
|
''')
|
|
|
|
for i in petition_titles:
|
|
print('''signcount_{pid}.label {title}
|
|
signcount_{pid}.info Total signatures for {title}
|
|
signcount_{pid}.draw LINE3
|
|
signcount_{pid}.min 0'''.format(pid=i, title=petition_titles[i]["title"]))
|
|
exit(0)
|
|
|
|
for i in environ.get("petitions").split():
|
|
req = urllib.request.Request("https://api.change.org/v1/petitions/{}?api_key={}&fields=signature_count".format(i, environ.get('APIkey')),
|
|
data=None,
|
|
headers={ 'User-Agent': 'curl/7.38.0' })
|
|
response = urllib.request.urlopen(req).read().decode('utf-8')
|
|
print("signcount_%s.value %s" % (i, json.loads(response)["signature_count"]))
|