From 188be2dfc16c11efb3b60595b8e260f1c36c7a32 Mon Sep 17 00:00:00 2001
From: Kevin Trocolli <pitok236@gmail.com>
Date: Sat, 18 Mar 2023 02:12:58 -0400
Subject: [PATCH] database: add autoupgrade command

---
 core/data/database.py    | 74 +++++++++++++++++++++++++++++++---------
 core/data/schema/base.py |  9 +++++
 dbutils.py               |  3 ++
 3 files changed, 69 insertions(+), 17 deletions(-)

diff --git a/core/data/database.py b/core/data/database.py
index 0ce039e..07fe79e 100644
--- a/core/data/database.py
+++ b/core/data/database.py
@@ -145,25 +145,49 @@ class Data:
             )
             return
 
-        if not os.path.exists(
-            f"core/data/schema/versions/{game.upper()}_{version}_{action}.sql"
-        ):
-            self.logger.error(
-                f"Could not find {action} script {game.upper()}_{version}_{action}.sql in core/data/schema/versions folder"
-            )
-            return
+        if action == "upgrade":
+            for x in range(old_ver, version):
+                if not os.path.exists(
+                    f"core/data/schema/versions/{game.upper()}_{x + 1}_{action}.sql"
+                ):
+                    self.logger.error(
+                        f"Could not find {action} script {game.upper()}_{x + 1}_{action}.sql in core/data/schema/versions folder"
+                    )
+                    return
 
-        with open(
-            f"core/data/schema/versions/{game.upper()}_{version}_{action}.sql",
-            "r",
-            encoding="utf-8",
-        ) as f:
-            sql = f.read()
+                with open(
+                    f"core/data/schema/versions/{game.upper()}_{x + 1}_{action}.sql",
+                    "r",
+                    encoding="utf-8",
+                ) as f:
+                    sql = f.read()
 
-        result = self.base.execute(sql)
-        if result is None:
-            self.logger.error("Error execuing sql script!")
-            return None
+                result = self.base.execute(sql)
+                if result is None:
+                    self.logger.error("Error execuing sql script!")
+                    return None
+        
+        else:
+            for x in range(old_ver, version, -1):
+                if not os.path.exists(
+                    f"core/data/schema/versions/{game.upper()}_{x - 1}_{action}.sql"
+                ):
+                    self.logger.error(
+                        f"Could not find {action} script {game.upper()}_{x - 1}_{action}.sql in core/data/schema/versions folder"
+                    )
+                    return
+
+                with open(
+                    f"core/data/schema/versions/{game.upper()}_{x - 1}_{action}.sql",
+                    "r",
+                    encoding="utf-8",
+                ) as f:
+                    sql = f.read()
+
+                result = self.base.execute(sql)
+                if result is None:
+                    self.logger.error("Error execuing sql script!")
+                    return None
 
         result = self.base.set_schema_ver(version, game)
         if result is None:
@@ -237,3 +261,19 @@ class Data:
             if not cards:
                 self.logger.info(f"Delete hanging user {user['id']}")
                 self.user.delete_user(user["id"])
+
+    def autoupgrade(self) -> None:
+        all_games = self.base.get_all_schema_vers()
+        if all_games is None:
+            self.logger.warn("Failed to get schema versions")
+        
+        for x in all_games:
+            game = x["game"].upper()
+            update_ver = 1
+            for y in range(2, 100):
+                if os.path.exists(f"core/data/schema/versions/{game}_{y}_upgrade.sql"):
+                    update_ver = y
+                else:
+                    break
+        
+            self.migrate_database(game, update_ver, "upgrade")
\ No newline at end of file
diff --git a/core/data/schema/base.py b/core/data/schema/base.py
index 9899f29..f77a9aa 100644
--- a/core/data/schema/base.py
+++ b/core/data/schema/base.py
@@ -2,6 +2,7 @@ import json
 import logging
 from random import randrange
 from typing import Any, Optional, Dict, List
+from sqlalchemy.engine import Row
 from sqlalchemy.engine.cursor import CursorResult
 from sqlalchemy.engine.base import Connection
 from sqlalchemy.sql import text, func, select
@@ -80,6 +81,14 @@ class BaseData:
         Generate a random 5-7 digit id
         """
         return randrange(10000, 9999999)
+    
+    def get_all_schema_vers(self) -> Optional[List[Row]]:
+        sql = select(schema_ver)
+
+        result = self.execute(sql)
+        if result is None:
+            return None
+        return result.fetchall()
 
     def get_schema_ver(self, game: str) -> Optional[int]:
         sql = select(schema_ver).where(schema_ver.c.game == game)
diff --git a/dbutils.py b/dbutils.py
index 65b1f8e..176c67e 100644
--- a/dbutils.py
+++ b/dbutils.py
@@ -64,6 +64,9 @@ if __name__ == "__main__":
         else:
             data.migrate_database(args.game, int(args.version), args.action)
 
+    elif args.action == "autoupgrade":
+        data.autoupgrade()
+
     elif args.action == "create-owner":
         data.create_owner(args.email)