use correct argument for asyncmysql

Author: rishsriv

Diff for: defog/async_admin_methods.py

@@ -338,8 +338,10 @@ async def create_empty_tables(self, dev: bool = False):
 
         elif self.db_type == "mysql":
             import aiomysql
-
-            conn = await aiomysql.connect(**self.db_creds)
+            db_creds = self.db_creds.copy()
+            db_creds["db"] = db_creds["database"]
+            del db_creds["database"]
+            conn = await aiomysql.connect(**db_creds)
             async with conn.cursor() as cur:
                 for statement in ddl.split(";"):
                     await cur.execute(statement)

Diff for: defog/async_generate_schema.py

@@ -243,7 +243,10 @@ async def generate_mysql_schema(
     except:
         raise Exception("aiomysql not installed.")
 
-    conn = await aiomysql.connect(**self.db_creds)
+    db_creds = self.db_creds.copy()
+    db_creds["db"] = db_creds["database"]
+    del db_creds["database"]
+    conn = await aiomysql.connect(**db_creds)
     cur = await conn.cursor()
     schemas = {}
 

Diff for: defog/query.py

@@ -206,6 +206,9 @@ async def async_execute_query_once(db_type: str, db_creds, query: str):
             import aiomysql
         except:
             raise Exception("aiomysql not installed.")
+        db_creds = self.db_creds.copy()
+        db_creds["db"] = db_creds["database"]
+        del db_creds["database"]
         conn = await aiomysql.connect(**db_creds)
         cur = await conn.cursor()
         await cur.execute(query)