add migrate and cli

tests/backends/mysql/__init__.py

@@ -2,17 +2,17 @@ from asynctest import TestCase
 from tortoise import Tortoise
 
 from alice.backends.mysql import MysqlDDL
-from tests.models import Category
+from alice.migrate import Migrate
 
 TORTOISE_ORM = {
     'connections': {
-        'default': 'mysql://root:123456@127.0.0.1:3306/test'
+        'default': 'mysql://root:123456@127.0.0.1:3306/test',
     },
     'apps': {
         'models': {
             'models': ['tests.models'],
             'default_connection': 'default',
-        }
+        },
     }
 }
 
@@ -22,6 +22,7 @@ class DBTestCase(TestCase):
         await Tortoise.init(config=TORTOISE_ORM)
         self.client = Tortoise.get_connection('default')
         self.ddl = MysqlDDL(self.client)
+        self.migrate = Migrate(ddl=self.ddl)
 
     async def tearDown(self) -> None:
         await Tortoise.close_connections()

tests/backends/mysql/test_ddl.py

@@ -1,5 +1,5 @@
 from tests.backends.mysql import DBTestCase
-from tests.models import Category, User
+from tests.models import Category
 
 
 class TestDDL(DBTestCase):
@@ -47,7 +47,3 @@ class TestDDL(DBTestCase):
     def test_drop_fk(self):
         ret = self.ddl.drop_fk(Category, Category._meta.fields_map.get('user'))
         self.assertEqual(ret, "ALTER TABLE category DROP FOREIGN KEY fk_category_user_366ffa6f")
-
-    async def test_aa(self):
-        user = await User.get(username='test')
-        await user.save()

tests/backends/mysql/test_migrate.py

@@ -0,0 +1,7 @@
+from tests.backends.mysql import DBTestCase
+
+
+class TestMigrate(DBTestCase):
+    async def test_migrate(self):
+        self.migrate.diff_models_module('tests.models', 'tests.new_models')
+        print(self.migrate.operators)

tests/new_models.py

@@ -0,0 +1,69 @@
+import datetime
+from enum import IntEnum
+from tortoise import fields, Model
+
+
+class ProductType(IntEnum):
+    article = 1
+    page = 2
+
+
+class PermissionAction(IntEnum):
+    create = 1
+    delete = 2
+    update = 3
+    read = 4
+
+
+class Status(IntEnum):
+    on = 1
+    off = 0
+
+
+class User(Model):
+    username = fields.CharField(max_length=20, unique=True)
+    password = fields.CharField(max_length=200)
+    last_login = fields.DatetimeField(description='Last Login', default=datetime.datetime.now)
+    is_active = fields.BooleanField(default=True, description='Is Active')
+    is_superuser = fields.BooleanField(default=False, description='Is SuperUser')
+    avatar = fields.CharField(max_length=200, default='')
+    intro = fields.TextField(default='')
+
+
+    def __str__(self):
+        return f'{self.pk}#{self.username}'
+
+
+class Category(Model):
+    slug = fields.CharField(max_length=200)
+    name = fields.CharField(max_length=200)
+    user = fields.ForeignKeyField('new_models.User', description='User')
+    created_at = fields.DatetimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.pk}#{self.name}'
+
+
+class Product(Model):
+    categories = fields.ManyToManyField('new_models.Category')
+    name = fields.CharField(max_length=50)
+    view_num = fields.IntField(description='View Num')
+    sort = fields.IntField()
+    is_reviewed = fields.BooleanField(description='Is Reviewed')
+    type = fields.IntEnumField(ProductType, description='Product Type')
+    image = fields.CharField(max_length=200)
+    body = fields.TextField()
+    created_at = fields.DatetimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.pk}#{self.name}'
+
+
+class Config(Model):
+    label = fields.CharField(max_length=200)
+    key = fields.CharField(max_length=20)
+    value = fields.JSONField()
+    status: Status = fields.IntEnumField(Status, default=Status.on)
+
+    def __str__(self):
+        return f'{self.pk}#{self.label}'