mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2024-11-24 22:10:19 +00:00
409 lines
11 KiB
Python
409 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
|
|
import logging
|
|
from sqlite3 import OperationalError
|
|
|
|
import downloadutils
|
|
|
|
##################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
|
|
##################################################################################################
|
|
|
|
|
|
class Embydb_Functions():
|
|
|
|
|
|
def __init__(self, embycursor):
|
|
|
|
self.embycursor = embycursor
|
|
self.download = downloadutils.DownloadUtils().downloadUrl
|
|
|
|
|
|
def get_version(self, version=None):
|
|
|
|
if version is not None:
|
|
self.embycursor.execute("DELETE FROM version")
|
|
query = "INSERT INTO version(idVersion) VALUES (?)"
|
|
self.embycursor.execute(query, (version,))
|
|
else:
|
|
query = "SELECT idVersion FROM version"
|
|
self.embycursor.execute(query)
|
|
try:
|
|
version = self.embycursor.fetchone()[0]
|
|
except TypeError:
|
|
pass
|
|
|
|
return version
|
|
|
|
def getViews(self):
|
|
|
|
views = []
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT view_id",
|
|
"FROM view"
|
|
))
|
|
self.embycursor.execute(query)
|
|
rows = self.embycursor.fetchall()
|
|
for row in rows:
|
|
views.append(row[0])
|
|
|
|
return views
|
|
|
|
def getView_embyId(self, item_id):
|
|
# Returns ancestors using embyId
|
|
url = "{server}/emby/Items/%s/Ancestors?UserId={UserId}&format=json" % item_id
|
|
|
|
try:
|
|
view_list = self.download(url)
|
|
except Exception as error:
|
|
log.info("Error getting views: " + str(error))
|
|
view_list = []
|
|
|
|
if view_list is None:
|
|
view_list = []
|
|
|
|
for view in view_list:
|
|
|
|
if view['Type'] == "CollectionFolder":
|
|
# Found view
|
|
view_id = view['Id']
|
|
break
|
|
else: # No view found
|
|
return [None, None]
|
|
|
|
# Compare to view table in emby database
|
|
query = ' '.join((
|
|
|
|
"SELECT view_name",
|
|
"FROM view",
|
|
"WHERE view_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (view_id,))
|
|
try:
|
|
view_name = self.embycursor.fetchone()[0]
|
|
except TypeError:
|
|
view_name = None
|
|
|
|
return [view_name, view_id]
|
|
|
|
def getView_byId(self, viewid):
|
|
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT view_name, media_type, kodi_tagid",
|
|
"FROM view",
|
|
"WHERE view_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (viewid,))
|
|
view = self.embycursor.fetchone()
|
|
|
|
return view
|
|
|
|
def getView_byType(self, mediatype):
|
|
|
|
views = []
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT view_id, view_name",
|
|
"FROM view",
|
|
"WHERE media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (mediatype,))
|
|
rows = self.embycursor.fetchall()
|
|
for row in rows:
|
|
views.append({
|
|
|
|
'id': row[0],
|
|
'name': row[1],
|
|
'mediatype': mediatype
|
|
})
|
|
|
|
if mediatype in ('tvshows', 'movies'):
|
|
query = ' '.join((
|
|
"SELECT view_id, view_name",
|
|
"FROM view",
|
|
"WHERE media_type = ?"
|
|
))
|
|
|
|
self.embycursor.execute(query, ("mixed",))
|
|
rows = self.embycursor.fetchall()
|
|
for row in rows:
|
|
views.append({
|
|
|
|
'id': row[0],
|
|
'name': row[1],
|
|
'mediatype': "mixed"
|
|
})
|
|
|
|
return views
|
|
|
|
def getView_byName(self, tagname):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT view_id",
|
|
"FROM view",
|
|
"WHERE view_name = ?"
|
|
))
|
|
self.embycursor.execute(query, (tagname,))
|
|
try:
|
|
view = self.embycursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
view = None
|
|
|
|
return view
|
|
|
|
def addView(self, embyid, name, mediatype, tagid, group_series):
|
|
|
|
query = (
|
|
'''
|
|
INSERT INTO view(
|
|
view_id, view_name, media_type, kodi_tagid, group_series)
|
|
|
|
VALUES (?, ?, ?, ?, ?)
|
|
'''
|
|
)
|
|
self.embycursor.execute(query, (embyid, name, mediatype, tagid, group_series))
|
|
|
|
def get_view_grouped_series(self, view_id):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT group_series",
|
|
"FROM view",
|
|
"WHERE view_id = ?"
|
|
))
|
|
try:
|
|
self.embycursor.execute(query, (view_id,))
|
|
return self.embycursor.fetchone()
|
|
except: return False
|
|
|
|
def update_view_grouped_series(self, view_id, group_series):
|
|
|
|
query = ' '.join((
|
|
|
|
"UPDATE view",
|
|
"SET group_series = ?",
|
|
"WHERE view_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (group_series, view_id))
|
|
|
|
def updateView(self, name, tagid, mediafolderid):
|
|
|
|
query = ' '.join((
|
|
|
|
"UPDATE view",
|
|
"SET view_name = ?, kodi_tagid = ?",
|
|
"WHERE view_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (name, tagid, mediafolderid))
|
|
|
|
def removeView(self, viewid):
|
|
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM view",
|
|
"WHERE view_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (viewid,))
|
|
|
|
def getItem_byId(self, embyid):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT kodi_id, kodi_fileid, kodi_pathid, parent_id, media_type, emby_type",
|
|
"FROM emby",
|
|
"WHERE emby_id = ?"
|
|
))
|
|
try:
|
|
self.embycursor.execute(query, (embyid,))
|
|
item = self.embycursor.fetchone()
|
|
return item
|
|
except: return None
|
|
|
|
def getItem_byWildId(self, embyid):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT kodi_id, media_type",
|
|
"FROM emby",
|
|
"WHERE emby_id LIKE ?"
|
|
))
|
|
self.embycursor.execute(query, (embyid+"%",))
|
|
return self.embycursor.fetchall()
|
|
|
|
def getItem_byView(self, mediafolderid):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT kodi_id",
|
|
"FROM emby",
|
|
"WHERE media_folder = ?"
|
|
))
|
|
self.embycursor.execute(query, (mediafolderid,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def get_item_by_view(self, view_id):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id",
|
|
"FROM emby",
|
|
"WHERE media_folder = ?"
|
|
))
|
|
self.embycursor.execute(query, (view_id,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def getItem_byKodiId(self, kodiid, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id, parent_id, media_folder",
|
|
"FROM emby",
|
|
"WHERE kodi_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (kodiid, mediatype,))
|
|
return self.embycursor.fetchone()
|
|
|
|
def getItem_byParentId(self, parentid, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id, kodi_id, kodi_fileid",
|
|
"FROM emby",
|
|
"WHERE parent_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (parentid, mediatype,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def getItemId_byParentId(self, parentid, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id, kodi_id",
|
|
"FROM emby",
|
|
"WHERE parent_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (parentid, mediatype,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def get_checksum(self, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id, checksum",
|
|
"FROM emby",
|
|
"WHERE emby_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (mediatype,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def get_checksum_by_view(self, media_type, view_id):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_id, checksum",
|
|
"FROM emby",
|
|
"WHERE emby_type = ?",
|
|
"AND media_folder = ?"
|
|
))
|
|
self.embycursor.execute(query, (media_type, view_id,))
|
|
return self.embycursor.fetchall()
|
|
|
|
def getMediaType_byId(self, embyid):
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT emby_type",
|
|
"FROM emby",
|
|
"WHERE emby_id = ?"
|
|
))
|
|
self.embycursor.execute(query, (embyid,))
|
|
try:
|
|
itemtype = self.embycursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
itemtype = None
|
|
|
|
return itemtype
|
|
|
|
def sortby_mediaType(self, itemids, unsorted=True):
|
|
|
|
sorted_items = {}
|
|
|
|
for itemid in itemids:
|
|
|
|
mediatype = self.getMediaType_byId(itemid)
|
|
if mediatype:
|
|
sorted_items.setdefault(mediatype, []).append(itemid)
|
|
elif unsorted:
|
|
sorted_items.setdefault('Unsorted', []).append(itemid)
|
|
|
|
return sorted_items
|
|
|
|
def addReference(self, embyid, kodiid, embytype, mediatype, fileid=None, pathid=None,
|
|
parentid=None, checksum=None, mediafolderid=None):
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO emby(
|
|
emby_id, kodi_id, kodi_fileid, kodi_pathid, emby_type, media_type, parent_id,
|
|
checksum, media_folder)
|
|
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
'''
|
|
)
|
|
self.embycursor.execute(query, (embyid, kodiid, fileid, pathid, embytype, mediatype,
|
|
parentid, checksum, mediafolderid))
|
|
|
|
def updateReference(self, embyid, checksum):
|
|
|
|
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
|
|
self.embycursor.execute(query, (checksum, embyid))
|
|
|
|
def updateParentId(self, embyid, parent_kodiid):
|
|
|
|
query = "UPDATE emby SET parent_id = ? WHERE emby_id = ?"
|
|
self.embycursor.execute(query, (parent_kodiid, embyid))
|
|
|
|
def removeItems_byParentId(self, parent_kodiid, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM emby",
|
|
"WHERE parent_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (parent_kodiid, mediatype,))
|
|
|
|
def removeItem_byKodiId(self, kodiid, mediatype):
|
|
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM emby",
|
|
"WHERE kodi_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
self.embycursor.execute(query, (kodiid, mediatype,))
|
|
|
|
def removeItem(self, embyid):
|
|
|
|
query = "DELETE FROM emby WHERE emby_id = ?"
|
|
self.embycursor.execute(query, (embyid,))
|
|
|
|
def removeWildItem(self, embyid):
|
|
|
|
query = "DELETE FROM emby WHERE emby_id LIKE ?"
|
|
self.embycursor.execute(query, (embyid+"%",))
|
|
|