import sqlite3 as sql

tbl_artist_info_ddl = str('CREATE TABLE artist_info ('
                            'artist_id        TEXT NOT NULL PRIMARY KEY,'
                            'artist_info      TEXT,'
                            'last_update      TEXT);')

class SQLiteDatabase(object):
    def __init__(self, db_filename):
        print("Init %s"%db_filename) 
        self.db_filename = db_filename
        self.conn = None
    
        self.connect()

    def connect(self):
        try:
            #xbmc.log("Trying connection to the database %s"%self.db_filename, xbmc.LOGINFO)
            print("Trying connection to the database %s"%self.db_filename)
            self.conn = sql.connect(self.db_filename)
            cursor = self.conn.cursor()
            cursor.execute(str('SELECT SQLITE_VERSION()'))
            #xbmc.log("Connection %s was successful %s"%(self.db_filename, cursor.fetchone()[0]), xbmc.LOGINFO)
            print("Connection %s was successful %s"%(self.db_filename, cursor.fetchone()[0]))
            cursor.row_factory = lambda cursor, row: row[0]
            cursor.execute(str('SELECT name FROM sqlite_master WHERE type=\'table\' ''AND name NOT LIKE \'sqlite_%\''))
            list_tables = cursor.fetchall()
            if not list_tables:
                # If no tables exist create a new one
                #xbmc.log("Creating Subsonic local DB", xbmc.LOGINFO)
                print("Creating Subsonic local DB")
                cursor.execute(tbl_artist_info_ddl)
        except sql.Error as e:
            #xbmc.log("SQLite error %s"%e.args[0], xbmc.LOGINFO)
            print("SQLite error %s"%e.args[0])

    def get_cursor(self):
        return self.conn.cursor()

    def run_query(self, query, params=None, cursor=None):
        print("Processing query %s params %s"%(str(query),str(params)))      
        try:
            if cursor is None:
                cursor = self.get_cursor()
            if params is not None:
                cursor.execute(query, params)
            else:
                cursor.execute(query)
            return cursor
        except sql.Error as e:
            print("SQLite error %s"%e.args[0])
        except ValueError:
            print("Error query %s"%str(query))  
            print("Error query type %s"%type(query))
            print("Error params %s"%str(params))  
            print("Error params type %s"%type(params))

    def update_artist(self, artist_id, artist_info, update_time):
        success = False
        query = 'INSERT or REPLACE INTO artist_info VALUES (?, ?, ?)'
        params = (str(artist_id), str(artist_info), str(update_time)) 
        cursor = self.run_query(query, params)
        try:
            self.conn.commit()
            success = True
        except Exception as e:
            print("Exception %s"%e)                
            pass
        return success

    def get_artist_info(self, artist_id):   
        query = 'SELECT * FROM artist_info WHERE artist_id = ?'
        params = [str(artist_id)]    
        cursor = self.run_query(query, params)
        return cursor.fetchall()

    def close(self):
        if self.conn:
            self.conn.close()