Make sqlite and mysql table and column detection a little bit more resilient for UX

django_migration_linter/sql_analyser/mysql.py

@@ -27,3 +27,13 @@ class MySqlAnalyser(BaseAnalyser):
             "mode": "one_liner",
         }
     ]
+
+    @staticmethod
+    def detect_column(sql):
+        if isinstance(sql, str):
+            regex_result = re.search("COLUMN [`\"'](.*?)[`\"']", sql, re.IGNORECASE)
+            if regex_result:
+                return regex_result.group(1)
+            regex_result = re.search("MODIFY [`\"'](.*?)[`\"']", sql, re.IGNORECASE)
+            if regex_result:
+                return regex_result.group(1)

django_migration_linter/sql_analyser/sqlite.py

@@ -67,3 +67,13 @@ class SqliteAnalyser(BaseAnalyser):
         },
         {"code": "ADD_UNIQUE", "fn": has_add_unique, "mode": "transaction"},
     ]
+
+    @staticmethod
+    def detect_table(sql):
+        if isinstance(sql, str):
+            regex_result = re.search("TABLE [`\"'](.*?)[`\"']", sql, re.IGNORECASE)
+            if regex_result:
+                return regex_result.group(1)
+            regex_result = re.search("ON [`\"'](.*?)[`\"']", sql, re.IGNORECASE)
+            if regex_result:
+                return regex_result.group(1)