80 lines
3.5 KiB
Python
80 lines
3.5 KiB
Python
|
from traceback import format_exc
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from lib.notification import Notification
|
||
|
|
from core.webutils import WebUtils
|
||
|
|
from lib.user import User
|
||
|
|
|
||
|
|
|
||
|
|
class UpdateNotifications(WebUtils):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__(module_='update_notifications')
|
||
|
|
league_users = list(User.get_users_league_bets(db=self.mysqldb))
|
||
|
|
europe_users = list(User.get_users_europe_bets(db=self.mysqldb))
|
||
|
|
self.nb_tasks = len(league_users) + len(europe_users)
|
||
|
|
self.start()
|
||
|
|
self.update_league_notifications(league_users)
|
||
|
|
self.update_europe_notifications(europe_users)
|
||
|
|
self.end()
|
||
|
|
|
||
|
|
def update_league_notifications(self, users):
|
||
|
|
self.logger.info('[*] Update users league notifications')
|
||
|
|
for user in users:
|
||
|
|
if user.notifications is not None and 'league' is user.notifications:
|
||
|
|
try:
|
||
|
|
for mday, matches in user.league_bets_matches.items():
|
||
|
|
if matches:
|
||
|
|
for method, hours in user.notifications['league'].items():
|
||
|
|
for hour in hours:
|
||
|
|
date_diff = matches[0].start_date - timedelta(hours=hour)
|
||
|
|
notif = Notification(
|
||
|
|
id_user=user.id,
|
||
|
|
id_match=matches[0].id,
|
||
|
|
date=date_diff,
|
||
|
|
mday_or_round=str(mday),
|
||
|
|
type_='league',
|
||
|
|
method=method
|
||
|
|
)
|
||
|
|
notif.save(db=self.mysqldb)
|
||
|
|
|
||
|
|
except BaseException as e:
|
||
|
|
self.logger.error('[-] user {}: {} - {}\n{}'.format(user.id, type(e), e, format_exc()))
|
||
|
|
self.errors += 1
|
||
|
|
else:
|
||
|
|
self.logger.info('[+] user {}: OK'.format(user.id))
|
||
|
|
self.tasks_done += 1
|
||
|
|
self.quantity += 1
|
||
|
|
|
||
|
|
def update_europe_notifications(self, users):
|
||
|
|
self.logger.info('[*] Update users europe notifications')
|
||
|
|
for user in users:
|
||
|
|
if user.notifications is not None and 'europe' is user.notifications:
|
||
|
|
try:
|
||
|
|
matches = user.europe_bets_matches
|
||
|
|
if matches:
|
||
|
|
for method, hours in user.notifications['europe'].items():
|
||
|
|
for hour in hours:
|
||
|
|
date_diff = matches[0].start_date - timedelta(hours=hour)
|
||
|
|
notif = Notification(
|
||
|
|
id_user=user.id,
|
||
|
|
id_match=matches[0].id,
|
||
|
|
date=date_diff,
|
||
|
|
mday_or_round=str(user.europe_round),
|
||
|
|
type_='europe',
|
||
|
|
method=method
|
||
|
|
)
|
||
|
|
notif.save(db=self.mysqldb)
|
||
|
|
|
||
|
|
except BaseException as e:
|
||
|
|
self.logger.error('[-] user {}: {} - {}\n{}'.format(user.id, type(e), e, format_exc()))
|
||
|
|
self.errors += 1
|
||
|
|
else:
|
||
|
|
self.logger.info('[+] user {}: OK'.format(user.id))
|
||
|
|
self.tasks_done += 1
|
||
|
|
self.quantity += 1
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
UpdateNotifications()
|