41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
import json
|
||
|
|
|
||
|
|
|
||
|
|
class TvChannelSource:
|
||
|
|
|
||
|
|
SOURCES = {
|
||
|
|
3: 'http://www.programme-television.org/chaines-tv'
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, id_country, url):
|
||
|
|
self.id_country = id_country
|
||
|
|
self.url = url
|
||
|
|
self.tv_channels = list()
|
||
|
|
self.error = None
|
||
|
|
|
||
|
|
|
||
|
|
class TvChannel:
|
||
|
|
def __init__(self, id_country, name, idt=0):
|
||
|
|
self.id = idt
|
||
|
|
self.id_country = id_country
|
||
|
|
self.name = name
|
||
|
|
self.names = None
|
||
|
|
|
||
|
|
def store(self, db):
|
||
|
|
db.exec(
|
||
|
|
'INSERT IGNORE INTO tv_channels (id_country, name) VALUES (:id_country, :name)',
|
||
|
|
{'id_country': self.id_country, 'name': self.name}
|
||
|
|
)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_sources():
|
||
|
|
for id_country, url in TvChannelSource.SOURCES.items():
|
||
|
|
yield TvChannelSource(id_country, url)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_tvchannels(db):
|
||
|
|
for row in db.query('SELECT id, id_country, name, names FROM tv_channels'):
|
||
|
|
tv_channel = TvChannel(id_country=row['id_country'], name=row['name'], idt=row['id'])
|
||
|
|
tv_channel.names = json.loads(row['names'])
|
||
|
|
yield tv_channel
|