Fix loading of SqliteSession

This commit is contained in:
Lonami Exo 2023-09-12 20:54:49 +02:00
parent 03a85dfd1a
commit f2b8a91fa9

View File

@ -113,7 +113,9 @@ class SqliteSession(Storage):
date=state[2], date=state[2],
seq=state[3], seq=state[3],
channels=[ChannelState(id=id, pts=pts) for id, pts in channelstate], channels=[ChannelState(id=id, pts=pts) for id, pts in channelstate],
), )
if state
else None,
) )
@staticmethod @staticmethod
@ -163,10 +165,11 @@ class SqliteSession(Storage):
) )
if c.fetchone(): if c.fetchone():
c.execute("select version from version") c.execute("select version from version")
res = c.fetchone()[0] tup = c.fetchone()
assert isinstance(res, int) if tup and isinstance(tup[0], int):
return res return tup[0]
else: SqliteSession._reset(c)
SqliteSession._create_tables(c) SqliteSession._create_tables(c)
c.execute("insert into version values (?)", (CURRENT_VERSION,)) c.execute("insert into version values (?)", (CURRENT_VERSION,))
return CURRENT_VERSION return CURRENT_VERSION