Run middleware migrations on config upload

(cherry picked from commit 319293f)

src/middlewared/middlewared/plugins/migration.py

@@ -1,6 +1,8 @@
 import asyncio
 import os
 
+import sqlite3
+
 from middlewared.service import Service
 import middlewared.sqlalchemy as sa
 from middlewared.utils.plugins import load_modules
@@ -45,3 +47,15 @@ async def run(self):
                 await self.middleware.call("datastore.insert", "system.migration", {"name": name}, {"ha_sync": False})
 
             await self.middleware.call("keyvalue.set", "run_migration", False, {"ha_sync": False})
+
+
+def on_config_upload(middleware, path):
+    conn = sqlite3.connect(path)
+    try:
+        conn.execute("REPLACE INTO system_keyvalue (key, value) VALUES ('run_migration', 'true')")
+    finally:
+        conn.close()
+
+
+async def setup(middleware):
+    middleware.register_hook('config.on_upload', on_config_upload, sync=True)