87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import, division, unicode_literals
|
|
from .. import variables as v
|
|
|
|
|
|
class Playlists(object):
|
|
def playlist_ids(self):
|
|
"""
|
|
Returns an iterator of all Plex ids of playlists.
|
|
"""
|
|
self.cursor.execute('SELECT plex_id FROM playlists')
|
|
return (x[0] for x in self.cursor)
|
|
|
|
def kodi_playlist_paths(self):
|
|
"""
|
|
Returns an iterator of all Kodi playlist paths.
|
|
"""
|
|
self.cursor.execute('SELECT kodi_path FROM playlists')
|
|
return (x[0] for x in self.cursor)
|
|
|
|
def delete_playlist(self, playlist):
|
|
"""
|
|
Removes the entry for playlist [Playqueue_Object] from the Plex
|
|
playlists table.
|
|
Be sure to either set playlist.id or playlist.kodi_path
|
|
"""
|
|
if playlist.plex_id:
|
|
query = 'DELETE FROM playlists WHERE plex_id = ?'
|
|
var = playlist.plex_id
|
|
elif playlist.kodi_path:
|
|
query = 'DELETE FROM playlists WHERE kodi_path = ?'
|
|
var = playlist.kodi_path
|
|
else:
|
|
raise RuntimeError('Cannot delete playlist: %s' % playlist)
|
|
self.cursor.execute(query, (var, ))
|
|
|
|
def add_playlist(self, playlist):
|
|
"""
|
|
Inserts or modifies an existing entry in the Plex playlists table.
|
|
"""
|
|
query = '''
|
|
INSERT OR REPLACE INTO playlists(
|
|
plex_id,
|
|
plex_name,
|
|
plex_updatedat,
|
|
kodi_path,
|
|
kodi_type,
|
|
kodi_hash)
|
|
VALUES (?, ?, ?, ?, ?, ?)
|
|
'''
|
|
self.cursor.execute(
|
|
query,
|
|
(playlist.plex_id,
|
|
playlist.plex_name,
|
|
playlist.plex_updatedat,
|
|
playlist.kodi_path,
|
|
playlist.kodi_type,
|
|
playlist.kodi_hash))
|
|
|
|
def playlist(self, playlist, plex_id=None, path=None, kodi_hash=None):
|
|
"""
|
|
Returns a complete Playlist (empty one passed in via playlist)
|
|
for the entry with plex_id OR kodi_hash OR kodi_path.
|
|
Returns None if not found
|
|
"""
|
|
query = 'SELECT * FROM playlists WHERE %s = ? LIMIT 1'
|
|
if plex_id:
|
|
query = query % 'plex_id'
|
|
var = plex_id
|
|
elif kodi_hash:
|
|
query = query % 'kodi_hash'
|
|
var = kodi_hash
|
|
else:
|
|
query = query % 'kodi_path'
|
|
var = path
|
|
self.cursor.execute(query, (var, ))
|
|
answ = self.cursor.fetchone()
|
|
if not answ:
|
|
return
|
|
playlist.plex_id = answ[0]
|
|
playlist.plex_name = answ[1]
|
|
playlist.plex_updatedat = answ[2]
|
|
playlist.kodi_path = answ[3]
|
|
playlist.kodi_type = answ[4]
|
|
playlist.kodi_hash = answ[5]
|
|
return playlist
|