40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
|
from traceback import format_exc
|
||
|
|
|
||
|
|
from providers.controller import ProviderController
|
||
|
|
from lib.tvchannel import TvChannel
|
||
|
|
from lib.league import League
|
||
|
|
from lib.match import Match
|
||
|
|
from core.webutils import WebUtils
|
||
|
|
|
||
|
|
|
||
|
|
class UpdateTVSchedule(WebUtils):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__(module_='update_tvschedule')
|
||
|
|
leagues = list(League.get_leagues(db=self.mysqldb, origin=self.module, id_league=self.args.id_league))
|
||
|
|
self.tv_channels = [tvc for tvc in TvChannel.get_tvchannels(db=self.mysqldb) if tvc.names is not None]
|
||
|
|
self.nb_tasks = len(leagues)
|
||
|
|
self.start()
|
||
|
|
self.run(leagues, self.update_tvschedule)
|
||
|
|
self.end()
|
||
|
|
|
||
|
|
def update_tvschedule(self, league, data):
|
||
|
|
try:
|
||
|
|
league.matches = list(Match.get_matches(db=self.mysqldb, origin=self.module, id_league=league.id))
|
||
|
|
provider = ProviderController.get_provider(league.url)
|
||
|
|
provider.get_tvschedule(league, self.tv_channels, data)
|
||
|
|
for match in league.matches:
|
||
|
|
if match.tv_channels:
|
||
|
|
match.store_tvchannels(db=self.mysqldb)
|
||
|
|
self.quantity += 1
|
||
|
|
|
||
|
|
except BaseException as e:
|
||
|
|
self.logger.error('[-] scheduler {}: {} - {}\n{}'.format(league.url, type(e), e, format_exc()))
|
||
|
|
self.errors += 1
|
||
|
|
else:
|
||
|
|
self.logger.info('[+] scheduler {}: OK'.format(league.url))
|
||
|
|
self.tasks_done += 1
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
UpdateTVSchedule()
|