python - 如何编写 django shell 操作脚本？

Question

我想创建一个可以执行以下操作的脚本(.sh 或 python，不重要):

heroku pg:reset DATABASE_URL
heroku run python manage.py migrate
heroku run python manage.py shell
> from myapp.scenarios import *; reset_demo_data(); exit()

第 1 到 3 行是 UNIX 命令，但第 4 行是要在打开的 Django shell 中执行的 python。

我尝试使用 | 和 > 将 python 代码“注入(inject)”到命令中，但没有任何效果。我想这很容易做到，但我不知道怎么做......

谢谢。

Answer 1

我想最好的选择是写一个 custom management command .

您的脚本可能如下所示:

heroku pg:reset DATABASE_URL
heroku run python manage.py migrate
heroku run python manage.py shell
heroku run python manage.py reset_demo

当管理命令是这样的:

from django.core.management.base import BaseCommand
from myapp.scenarios import *

class Command(BaseCommand):
    def handle(self, *args, **options):
        reset_demo_data()