voussoir/ycdl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use insert_filler for add_channel.

Browse source
This commit is contained in:
voussoir 2020-04-03 11:12:56 -07:00
parent 718399806a
commit 11a147fbcd
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
ycdl/ycdldb.py
View file

@ -124,14 +124,18 @@ class YCDLDB:
if name is None:
name = self.youtube.get_user_name(channel_id)
data = [None] * len(SQL_CHANNEL)
data[SQL_CHANNEL['id']] = channel_id
data[SQL_CHANNEL['name']] = name
if download_directory is not None:
assert_is_abspath(download_directory)
data[SQL_CHANNEL['directory']] = download_directory
self.cur.execute('INSERT INTO channels VALUES(?, ?, ?, ?)', data)
data = {
'id': channel_id,
'name': name,
'directory': download_directory,
}
(qmarks, bindings) = sqlhelpers.insert_filler(SQL_CHANNEL, data)
query = f'INSERT INTO channels VALUES({qmarks})'
self.cur.execute(query)
if get_videos:
self.refresh_channel(channel_id, commit=False)