rpi: уведомлять владельца о скачивании Чернобыля
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Enable owner Telegram notifications for Chernobyl imports; run as root."""
|
||||
import json
|
||||
import urllib.request
|
||||
|
||||
def env(path):
|
||||
return dict(line.strip().split('=', 1) for line in open(path)
|
||||
if '=' in line and not line.startswith('#'))
|
||||
|
||||
key = env('/etc/media.env')['SONARR_KEY']
|
||||
|
||||
def api(path, data=None, method=None):
|
||||
req = urllib.request.Request('http://127.0.0.1:8989/api/v3/' + path,
|
||||
data=None if data is None else json.dumps(data).encode(),
|
||||
headers={'X-Api-Key': key, 'Content-Type': 'application/json'},
|
||||
method=method or ('GET' if data is None else 'POST'))
|
||||
with urllib.request.urlopen(req, timeout=30) as response:
|
||||
return json.load(response)
|
||||
|
||||
series = next(s for s in api('series') if s['tvdbId'] == 360893)
|
||||
tags = api('tag')
|
||||
tag = next((t for t in tags if t['label'] == 'notify-owner-chernobyl'), None)
|
||||
if tag is None:
|
||||
tag = api('tag', {'label': 'notify-owner-chernobyl'})
|
||||
if tag['id'] not in series['tags']:
|
||||
series['tags'].append(tag['id'])
|
||||
api('series/' + str(series['id']), series, 'PUT')
|
||||
name = 'Telegram: Chernobyl owner'
|
||||
existing = next((n for n in api('notification') if n['name'] == name), None)
|
||||
notification = existing or next(n for n in api('notification/schema') if n['implementation'] == 'Telegram')
|
||||
for field in notification['fields']:
|
||||
values = {'botToken': env('/etc/astra-bot.env')['TG_TOKEN'], 'chatId': '607015',
|
||||
'sendSilently': False, 'includeAppNameInTitle': True}
|
||||
if field['name'] in values:
|
||||
field['value'] = values[field['name']]
|
||||
for event in list(notification):
|
||||
if event.startswith('on') and isinstance(notification[event], bool):
|
||||
notification[event] = False
|
||||
notification.update(name=name, tags=[tag['id']], onDownload=True)
|
||||
api('notification/' + str(existing['id']) if existing else 'notification', notification,
|
||||
'PUT' if existing else 'POST')
|
||||
check = next(n for n in api('notification') if n['name'] == name)
|
||||
assert check['onDownload'] and check['tags'] == [tag['id']]
|
||||
print('Configured: Telegram notifications for Chernobyl episode imports, owner chat 607015.')
|
||||
Reference in New Issue
Block a user