Fix migrate to new database error

CHANGELOG.md

@@ -2,13 +2,22 @@
 
 ## 0.3
 
+### 0.3.2
+
+- Fix migrate to new database error. (#62)
+
+### 0.3.1
+
+- Fix first version error.
+- Fix init error. (#61)
+
 ### 0.3.0
 
 - Refactoring migrate logic, and this version is not compatible with previous version.
 - Now there don't need `old_models.py` and it store in database.
 - Upgrade steps:
   1. Upgrade aerich version.
-  2. Drop aerich model in db and recreate with new struct.
+  2. Drop aerich table in database.
   3. Delete `migrations/{app}` folder and rerun `aerich init-db`.
   4. Update model and `aerich migrate` normally.
 

README.md

@@ -10,6 +10,8 @@
 Aerich is a database migrations tool for Tortoise-ORM, which like alembic for SQLAlchemy, or Django ORM with it\'s
 own migrations solution.
 
+**If you upgrade aerich from <= 0.2.5 to >= 0.3.0, see [changelog](https://github.com/tortoise/aerich/blob/dev/CHANGELOG.md) for upgrade steps.**
+
 ## Install
 
 Just install from pypi:
@@ -33,7 +35,7 @@ Options:
   -h, --help         Show this message and exit.
 
 Commands:
-  downgrade  Downgrade to previous version.
+  downgrade  Downgrade to specified version.
   heads      Show current available heads in migrate location.
   history    List all migrate items.
   init       Init config file and generate root migrate location.

aerich/__init__.py

@@ -1 +1 @@
-__version__ = "0.3.0"
+__version__ = "0.3.2"

aerich/cli.py

@@ -28,7 +28,9 @@ def coro(f):
         loop = asyncio.get_event_loop()
         ctx = args[0]
         loop.run_until_complete(f(*args, **kwargs))
-        Migrate.remove_old_model_file(ctx.obj["app"], ctx.obj["location"])
+        app = ctx.obj.get("app")
+        if app:
+            Migrate.remove_old_model_file(app, ctx.obj["location"])
 
     return wrapper
 

aerich/migrate.py

@@ -4,7 +4,7 @@ import re
 from datetime import datetime
 from importlib import import_module
 from io import StringIO
-from typing import Dict, List, Tuple, Type
+from typing import Dict, List, Optional, Tuple, Type
 
 import click
 from tortoise import (
@@ -15,6 +15,7 @@ from tortoise import (
     Model,
     Tortoise,
 )
+from tortoise.exceptions import OperationalError
 from tortoise.fields import Field
 
 from aerich.ddl import BaseDDL
@@ -53,8 +54,11 @@ class Migrate:
         )
 
     @classmethod
-    async def get_last_version(cls) -> Aerich:
-        return await Aerich.filter(app=cls.app).first()
+    async def get_last_version(cls) -> Optional[Aerich]:
+        try:
+            return await Aerich.filter(app=cls.app).first()
+        except OperationalError:
+            pass
 
     @classmethod
     def remove_old_model_file(cls, app: str, location: str):
@@ -67,16 +71,16 @@ class Migrate:
     async def init_with_old_models(cls, config: dict, app: str, location: str):
         await Tortoise.init(config=config)
         last_version = await cls.get_last_version()
-        content = last_version.content
-        with open(cls.get_old_model_file(app, location), "w") as f:
-            f.write(content)
-
-        migrate_config = cls._get_migrate_config(config, app, location)
         cls.app = app
-        cls.migrate_config = migrate_config
         cls.migrate_location = os.path.join(location, app)
+        if last_version:
+            content = last_version.content
+            with open(cls.get_old_model_file(app, location), "w") as f:
+                f.write(content)
 
-        await Tortoise.init(config=migrate_config)
+            migrate_config = cls._get_migrate_config(config, app, location)
+            cls.migrate_config = migrate_config
+            await Tortoise.init(config=migrate_config)
 
         connection = get_app_connection(config, app)
         cls.dialect = connection.schema_generator.DIALECT

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "aerich"
-version = "0.3.0"
+version = "0.3.2"
 description = "A database migrations tool for Tortoise ORM."
 authors = ["long2ice <long2ice@gmail.com>"]
 license = "Apache-2.0"