feat: support environment set

Author: long2ice

.idea/.gitignore

@@ -0,0 +1,7 @@
+# ChangeLog
+
+## 0.1
+
+### 0.1.0
+
+- First release.

CHANGELOG.md

@@ -50,7 +50,19 @@ TORTOISE_ORM = {
 Then you can start an interactive shell for TortoiseORM.
 
 ```shell
-tortoise-cli -c examples.TORTOISE_ORM shell
+tortoise-cli -c settings.TORTOISE_ORM shell
+```
+
+Or you can set config by set environment variable.
+
+```shell
+export TORTOISE_ORM=settings.TORTOISE_ORM
+```
+
+Then just run:
+
+```shell
+tortoise-cli shell
 ```
 
 ## License

README.md

@@ -1,5 +1,7 @@
 import asyncio
+import os
 from functools import wraps
+from typing import Optional
 
 import click
 from ptpython.repl import embed
@@ -27,11 +29,14 @@ def wrapper(*args, **kwargs):
     "-c",
     "--config",
     help="TortoiseORM config dictionary path, like settings.TORTOISE_ORM",
-    required=True,
 )
 @click.pass_context
 @coro
-async def cli(ctx: click.Context, config: str):
+async def cli(ctx: click.Context, config: Optional[str]):
+    if not config:
+        config = os.getenv("TORTOISE_ORM")
+    if not config:
+        raise click.UsageError("You must specify TORTOISE_ORM in option or env", ctx=ctx)
     tortoise_config = utils.get_tortoise_config(ctx, config)
     await Tortoise.init(config=tortoise_config)
     await Tortoise.generate_schemas(safe=True)