Fixed handling of poster size and language if user did not set these values
This commit is contained in:
parent
5069551064
commit
94baaf9296
|
@ -1,6 +1,6 @@
|
|||
maubot: 0.1.0
|
||||
id: lomion.tmdb
|
||||
version: 1.0.0
|
||||
version: 1.0.1
|
||||
license: AGPL 3.0
|
||||
modules:
|
||||
- tmdb
|
||||
|
|
|
@ -45,7 +45,11 @@ class Database:
|
|||
def get_language(self, user_id):
|
||||
rows = self.db.execute(select([self.language.c.language])
|
||||
.where(self.language.c.user_id == user_id))
|
||||
return rows.fetchone()['language']
|
||||
row = rows.fetchone()
|
||||
if row:
|
||||
return row['language']
|
||||
else:
|
||||
return None
|
||||
|
||||
def set_poster_size(self, user_id, size):
|
||||
with self.db.begin() as tx:
|
||||
|
@ -55,4 +59,8 @@ class Database:
|
|||
def get_poster_size(self, user_id):
|
||||
rows = self.db.execute(select([self.tmdb_poster_size.c.size])
|
||||
.where(self.tmdb_poster_size.c.user_id == user_id))
|
||||
return rows.fetchone()['size']
|
||||
row = rows.fetchone()
|
||||
if row:
|
||||
return row['size']
|
||||
else:
|
||||
return None
|
Loading…
Reference in a new issue