database: fix error when trying to upgrade the schema for a game that wasn't created yet

2023-03-03 19:23:14 -05:00 · 2023-03-03 19:23:14 -05:00 · 4f3d3d8395
commit 4f3d3d8395
parent 02e1838d95
1 changed files with 6 additions and 1 deletions
--- a/core/data/schema/base.py
+++ b/core/data/schema/base.py
@ -85,7 +85,12 @@ class BaseData():
        result = self.execute(sql)
        if result is None:
            return None
-        return result.fetchone()["version"]
+        
+        row = result.fetchone()
+        if row is None:
+            return None
+        
+        return row["version"]
    
    def set_schema_ver(self, ver: int, game: str = "CORE") -> Optional[int]:
        sql = insert(schema_ver).values(game = game, version = ver)