Matthias Bilger
61c37d6700
All checks were successful
continuous-integration/drone/push Build is passing
Some explanations added and options implemented
77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
import bs4
|
|
import itertools
|
|
import time
|
|
import requests
|
|
import configparser
|
|
import urllib.parse
|
|
import mechanize
|
|
import pushover
|
|
import datetime
|
|
import os
|
|
|
|
def split_notify(string):
|
|
values = string.split(':')
|
|
return (values[0], values[1:])
|
|
|
|
|
|
def main():
|
|
while True:
|
|
pushover.init(os.environ['PUSHOVER_KEY'])
|
|
if 'HEALTHCHECK_URL' in os.environ:
|
|
requests.get(f"{os.environ['HEALTHCHECK_URL']}/start")
|
|
allinfo = []
|
|
users = list(map(lambda x: x.split(':', 1), os.environ['BIB_USERS'].split(',')))
|
|
notify = dict(map(split_notify, os.environ['NOTIFY_USERS'].split(',')))
|
|
for user, pwd in users:
|
|
allinfo += check(user, pwd, notify.get(user, []))
|
|
if 'HEALTHCHECK_URL' in os.environ:
|
|
requests.post(f"{os.environ['HEALTHCHECK_URL']}", data='\n'.join(allinfo).encode('utf8'))
|
|
if 'RUN_FOREVER' in os.environ and os.environ['RUN_FOREVER'] == 'False':
|
|
break
|
|
|
|
now = datetime.datetime.utcnow
|
|
to = (now() + datetime.timedelta(days = 1)).replace(hour=os.environ.get('RUN_AT_HOUR', 6), minute=os.environ.get('RUN_AT_MINUTE', 0), second=0)
|
|
time.sleep((to-now()).seconds)
|
|
|
|
def check(username, password, notify_ids):
|
|
br = mechanize.Browser()
|
|
starturl = os.environ.get('LIBRARY_URL', 'https://ssl.muenchen.de/aDISWeb/app?service=direct/0/Home/$DirectLink&sp=SOPAC')
|
|
response = br.open(starturl)
|
|
br.follow_link(text_regex=r"Anmelden")
|
|
br.select_form('Form0')
|
|
br['$Textfield'] = username
|
|
br['$Textfield$0'] = password
|
|
response = br.submit()
|
|
br.follow_link(text_regex=r"Konto")
|
|
try:
|
|
response = br.follow_link(text_regex=r"Ausleihen? zeigen")
|
|
br.select_form('Form0')
|
|
response = br.submit(name='textButton$0', label='Alle verlängern')
|
|
lentlist = bs4.BeautifulSoup(response.read(), 'html.parser')
|
|
table = lentlist.select('table[class="rTable_table"]')[0]
|
|
allinfo = []
|
|
for entry in table.tbody.select('tr'):
|
|
info = list(map(lambda x: str(x.text).strip(), entry.select('td')))
|
|
date = datetime.datetime.strptime(info[1], '%d.%m.%Y')
|
|
delta = date - datetime.datetime.now()
|
|
allinfo.append(str(info))
|
|
if delta.days <= 10 or delta.days == 20 or delta.days == 15:
|
|
message = f'Bitte an {info[3]} denken\n'
|
|
if delta.days <= 7:
|
|
message += '<font color="#ff0000">'
|
|
message += f'Abgabe <b>{info[1]}</b>'
|
|
if delta.days <= 7:
|
|
message += '</font>'
|
|
message += f' - {username}'
|
|
for client in itertools.chain(notify_ids, os.environ.get('PUSHOVER_CLIENTS', '').split(',')):
|
|
try:
|
|
pushover.Client(client).send_message(message, title="Erinnerung", html=1)
|
|
except:
|
|
print("No client")
|
|
except (StopIteration, mechanize._mechanize.LinkNotFoundError) as e:
|
|
return []
|
|
return allinfo
|
|
|
|
if __name__ == "__main__":
|
|
main()
|