Group db migrations (#3583)

Author: r4victor

src/dstack/_internal/server/alembic.ini

@@ -8,7 +8,7 @@ script_location = migrations
 # Uncomment the line below if you want the files to be prepended with date and time
 # see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
 # for all available tokens
-# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
+file_template = %%(year)d/%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d_%%(rev)s_%%(slug)s
 
 # sys.path path, will be prepended to sys.path if present.
 # defaults to the current working directory.
@@ -20,7 +20,7 @@ prepend_sys_path = .
 # installed by adding `alembic[tz]` to the pip requirements
 # string value is passed to dateutil.tz.gettz()
 # leave blank for localtime
-# timezone =
+timezone = utc
 
 # max length of characters to apply to the
 # "slug" field
@@ -50,7 +50,7 @@ path_separator = os
 # set to 'true' to search source files recursively
 # in each "version_locations" directory
 # new in Alembic version 1.10
-# recursive_version_locations = false
+recursive_version_locations = true
 
 # the output encoding used when revision files
 # are written from script.py.mako

src/dstack/_internal/server/db.py

@@ -125,5 +125,6 @@ async def sqlite_commit(session: AsyncSession):
 def _run_alembic_upgrade(connection):
     alembic_cfg = config.Config()
     alembic_cfg.set_main_option("script_location", settings.ALEMBIC_MIGRATIONS_LOCATION)
+    alembic_cfg.set_main_option("recursive_version_locations", "true")
     alembic_cfg.attributes["connection"] = connection
     command.upgrade(alembic_cfg, "head")

src/dstack/_internal/server/migrations/script.py.mako

@@ -6,9 +6,9 @@ Create Date: ${create_date}
 
 """
 
-from alembic import op
 import sqlalchemy as sa
 import sqlalchemy_utils
+from alembic import op
 
 import dstack._internal.server.models
 ${imports if imports else ""}