Error Trying to Save Token into SqliteTokenBackEnd

[pkstacey]

Hi, I've adapted some code from prior posts almost successfully but I'm unable to get the token to save. The error: Token could not be saved: dictionary update sequence element #0 has length 1; 2 is required. What am I doing wrong please?

Here is my subclass:

`import sqlite3
from O365 import utils

class SqliteTokenBackend(utils.BaseTokenBackend):

def __init__(self, db=None, table=None, column=None, user_id=None):
    super().__init__() 

    self.db = db
    self.user_id = user_id
    self.get_statement = "SELECT {} FROM {} WHERE user_id = ?".format(column, table)
    self.update_statement = "UPDATE {} SET {} = ? WHERE user_id = ?".format(table, column) #token, user
    self.insert_statement = "INSERT INTO {} VALUES (?,?)".format(table) #token, user
    self.userlookup_statement = "SELECT {} FROM {} WHERE user_id = ?".format(column, table)

def __repr__(self):
    return "{} for user: {}".format(self.__class__.__name__, self.user_id)

def load_token(self):
    token = None #self.token? need? seems unused. or needs init?
    con = sqlite3.connect(self.db) #why db not recog? call self...?
    cursor = con.cursor()
    cursor.execute(self.get_statement, (self.user_id, ))
    token_str = cursor.fetchone()
    
    if token_str:
        return self.token_constructor(self.serializer.loads(token_str))

def save_token(self):

    #lookup user
    con = sqlite3.connect(self.db) #why db not recog? call self...?
    cursor = con.cursor()
    cursor.execute(self.userlookup_statement, (self.user_id, ))
    user_str = cursor.fetchone()
    con.close()

    try:

        # if user found, update
        if user_str:
            con = sqlite3.connect(self.db) 
            cursor = con.cursor()
            cursor.execute(self.update_statement, (self.token_constructor(self.serializer.dumps(self.token)), self.user_id, ))
            self.db.commit()
            con.close()

        else:
            con = sqlite3.connect(self.db) 
            cursor = con.cursor()
            cursor.execute(self.insert_statement, (self.token_constructor(self.serializer.dumps(self.token)), self.user_id, ))
            self.db.commit()
            con.close()

        return True

    except Exception as e:

        print ('Token could not be saved: {}'.format(str(e)))
        
        return False

def delete_token(self):
    # optional
    pass

def check_token(self):
    # optional
    pass`

Thanks.

[pkstacey]

I have since fixed the code after sorting out conflicting errors. The code as follows inserts the token into the db just fine:

`
import sqlite3
from O365 import utils

class SqliteTokenBackend(utils.BaseTokenBackend):

def __init__(self, db=None, table=None, column=None, user_id=None):
    super().__init__()  # must call it

    # this can be done in multiple ways ....
    self.db = db
    self.user_id = user_id
    self.get_statement = "SELECT {} FROM {} WHERE user_id = ?".format(column, table)
    self.update_statement = "UPDATE {} SET {} = ? WHERE user_id = ?".format(table, column) #token, user
    self.insert_statement = "INSERT INTO {} VALUES (?,?)".format(table) #token, user
    self.userlookup_statement = "SELECT {} FROM {} WHERE user_id = ?".format(column, table)

def __repr__(self):
    return "{} for user: {}".format(self.__class__.__name__, self.user_id)

def load_token(self):
    token = None #self.token? need? seems unused. or needs init?
    con = sqlite3.connect(self.db) #why db not recog? call self...?
    cursor = con.cursor()
    cursor.execute(self.get_statement, (self.user_id, ))
    token_str = cursor.fetchone()
    
    if token_str:
        return self.token_constructor(self.serializer.loads(token_str))

def save_token(self):

    #lookup user
    con = sqlite3.connect(self.db) #why db not recog? call self...?
    cursor = con.cursor()
    cursor.execute(self.userlookup_statement, (self.user_id, ))
    user_str = cursor.fetchone()
    con.close()

    try:

        # if user found, update
        if user_str:
            con = sqlite3.connect(self.db) #why db not recog? call self...?
            cursor = con.cursor()
            cursor.execute(self.update_statement, (self.serializer.dumps(self.token), self.user_id, ))
            con.commit()
            con.close()

        # if user not found in database, insert:
        else:
            con = sqlite3.connect(self.db) #why db not recog? call self...?
            cursor = con.cursor()
            cursor.execute(self.insert_statement, (self.serializer.dumps(self.token), self.user_id, ))
            con.commit()
            con.close()

        return True

    except Exception as e:

        print ('Token could not be saved: {}'.format(str(e)))
        
        return False

def delete_token(self):
    # optional
    pass

def check_token(self):
    # optional
    pass

`