From 75414f5f078cdcef75d3fd562e87832caf601231 Mon Sep 17 00:00:00 2001
From: SpootDev <spootdev@gmail.com>
Date: Thu, 31 Mar 2016 10:48:10 -0500
Subject: [PATCH] self cursor

---
 resources/lib/embydb_functions.py | 627 +++++++++++++++---------------
 1 file changed, 303 insertions(+), 324 deletions(-)

diff --git a/resources/lib/embydb_functions.py b/resources/lib/embydb_functions.py
index 518498f5..c1150272 100644
--- a/resources/lib/embydb_functions.py
+++ b/resources/lib/embydb_functions.py
@@ -1,325 +1,304 @@
-# -*- coding: utf-8 -*-
-
-#################################################################################################
-
-import utils
-import clientinfo
-
-#################################################################################################
-
-
-class Embydb_Functions():
-
-
-    def __init__(self, embycursor):
-
-        self.embycursor = embycursor
-
-        self.clientInfo = clientinfo.ClientInfo()
-        self.addonName = self.clientInfo.getAddonName()
-
-    def logMsg(self, msg, lvl=1):
-
-        className = self.__class__.__name__
-        utils.logMsg("%s %s" % (self.addonName, className), msg, lvl)
-
-    def getViews(self):
-
-        embycursor = self.embycursor
-        views = []
-
-        query = ' '.join((
-
-            "SELECT view_id",
-            "FROM view"
-        ))
-        embycursor.execute(query)
-        rows = embycursor.fetchall()
-        for row in rows:
-            views.append(row[0])
-        
-        return views
-
-    def getView_byId(self, viewid):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT view_name, media_type, kodi_tagid",
-            "FROM view",
-            "WHERE view_id = ?"
-        ))
-        embycursor.execute(query, (viewid,))
-        view = embycursor.fetchone()
-        
-        return view
-
-    def getView_byType(self, mediatype):
-
-        embycursor = self.embycursor
-        views = []
-
-        query = ' '.join((
-
-            "SELECT view_id, view_name",
-            "FROM view",
-            "WHERE media_type = ?"
-        ))
-        embycursor.execute(query, (mediatype,))
-        rows = embycursor.fetchall()
-        for row in rows:
-            views.append({
-
-                'id': row[0],
-                'name': row[1]
-            })
-
-        return views
-
-    def getView_byName(self, tagname):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT view_id",
-            "FROM view",
-            "WHERE view_name = ?"
-        ))
-        embycursor.execute(query, (tagname,))
-        try:
-            view = embycursor.fetchone()[0]
-        
-        except TypeError:
-            view = None
-
-        return view
-
-    def addView(self, embyid, name, mediatype, tagid):
-
-        query = (
-            '''
-            INSERT INTO view(
-                view_id, view_name, media_type, kodi_tagid)
-
-            VALUES (?, ?, ?, ?)
-            '''
-        )
-        self.embycursor.execute(query, (embyid, name, mediatype, tagid))
-
-    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):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT kodi_id, kodi_fileid, kodi_pathid, parent_id, media_type, emby_type",
-            "FROM emby",
-            "WHERE emby_id = ?"
-        ))
-        try:
-            embycursor.execute(query, (embyid,))
-            item = embycursor.fetchone()
-            return item
-        except: return None
-
-    def getItem_byWildId(self, embyid):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT kodi_id, media_type",
-            "FROM emby",
-            "WHERE emby_id LIKE ?"
-        ))
-        embycursor.execute(query, (embyid+"%",))
-        items = embycursor.fetchall()
-
-        return items
-
-    def getItem_byView(self, mediafolderid):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT kodi_id",
-            "FROM emby",
-            "WHERE media_folder = ?"
-        ))
-        embycursor.execute(query, (mediafolderid,))
-        items = embycursor.fetchall()
-
-        return items
-
-    def getItem_byKodiId(self, kodiid, mediatype):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT emby_id, parent_id",
-            "FROM emby",
-            "WHERE kodi_id = ?",
-            "AND media_type = ?"
-        ))
-        embycursor.execute(query, (kodiid, mediatype,))
-        item = embycursor.fetchone()
-
-        return item
-
-    def getItem_byParentId(self, parentid, mediatype):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT emby_id, kodi_id, kodi_fileid",
-            "FROM emby",
-            "WHERE parent_id = ?",
-            "AND media_type = ?"
-        ))
-        embycursor.execute(query, (parentid, mediatype,))
-        items = embycursor.fetchall()
-
-        return items
-
-    def getItemId_byParentId(self, parentid, mediatype):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT emby_id, kodi_id",
-            "FROM emby",
-            "WHERE parent_id = ?",
-            "AND media_type = ?"
-        ))
-        embycursor.execute(query, (parentid, mediatype,))
-        items = embycursor.fetchall()
-
-        return items
-
-    def getChecksum(self, mediatype):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT emby_id, checksum",
-            "FROM emby",
-            "WHERE emby_type = ?"
-        ))
-        embycursor.execute(query, (mediatype,))
-        items = embycursor.fetchall()
-
-        return items
-
-    def getMediaType_byId(self, embyid):
-
-        embycursor = self.embycursor
-
-        query = ' '.join((
-
-            "SELECT emby_type",
-            "FROM emby",
-            "WHERE emby_id = ?"
-        ))
-        embycursor.execute(query, (embyid,))
-        try:
-            itemtype = 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+"%",))
+# -*- coding: utf-8 -*-
+
+#################################################################################################
+
+import utils
+import clientinfo
+
+#################################################################################################
+
+
+class Embydb_Functions():
+
+
+    def __init__(self, embycursor):
+
+        self.embycursor = embycursor
+
+        self.clientInfo = clientinfo.ClientInfo()
+        self.addonName = self.clientInfo.getAddonName()
+
+    def logMsg(self, msg, lvl=1):
+
+        className = self.__class__.__name__
+        utils.logMsg("%s %s" % (self.addonName, className), msg, lvl)
+
+    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_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]
+            })
+
+        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):
+
+        query = (
+            '''
+            INSERT INTO view(
+                view_id, view_name, media_type, kodi_tagid)
+
+            VALUES (?, ?, ?, ?)
+            '''
+        )
+        self.embycursor.execute(query, (embyid, name, mediatype, tagid))
+
+    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+"%",))
+        items = self.embycursor.fetchall()
+
+        return items
+
+    def getItem_byView(self, mediafolderid):
+
+        query = ' '.join((
+
+            "SELECT kodi_id",
+            "FROM emby",
+            "WHERE media_folder = ?"
+        ))
+        self.embycursor.execute(query, (mediafolderid,))
+        items = self.embycursor.fetchall()
+
+        return items
+
+    def getItem_byKodiId(self, kodiid, mediatype):
+
+        query = ' '.join((
+
+            "SELECT emby_id, parent_id",
+            "FROM emby",
+            "WHERE kodi_id = ?",
+            "AND media_type = ?"
+        ))
+        self.embycursor.execute(query, (kodiid, mediatype,))
+        item = self.embycursor.fetchone()
+
+        return item
+
+    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,))
+        items = self.embycursor.fetchall()
+
+        return items
+
+    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,))
+        items = self.embycursor.fetchall()
+
+        return items
+
+    def getChecksum(self, mediatype):
+
+        query = ' '.join((
+
+            "SELECT emby_id, checksum",
+            "FROM emby",
+            "WHERE emby_type = ?"
+        ))
+        self.embycursor.execute(query, (mediatype,))
+        items = self.embycursor.fetchall()
+
+        return items
+
+    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+"%",))
         
\ No newline at end of file