Django執(zhí)行指定腳本的方法有哪些

這篇文章主要介紹Django執(zhí)行指定腳本的方法有哪些，文中介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們一定要看完！

Django Shell

Django自帶的shell其實就加載了項目的各種環(huán)境變量，網(wǎng)上描述得也很多，直接這么跑就可以了：

python manage.py shell < script.py

簡單是非常簡單，不足的地方就是不好加參數(shù)進去，而且喜歡寫main的話也不好讓腳本跑起來。

自定義Command

也可以通過自定義一個簡單的command來實現(xiàn)需要的效果，以下是最近寫的一個示例，可支持執(zhí)行指定的腳本文件，同時還能加入更多的腳本執(zhí)行參數(shù)：

"""
@author:knktc
@contact:me@knktc.com
"""

import os
import sys
import argparse
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    help = 'Run script in current project environment'

    def add_arguments(self, parser):
        parser.add_argument('script', type=str, nargs=argparse.REMAINDER,
                            help='script file path and args')

    def handle(self, *args, **options):
        # get args
        args = options['script']
        script_path = args[0]

        # check file existence
        if not os.path.isfile(script_path):
            self.stderr.write(f'No such file: [{script_path}]')
            sys.exit(1)

        # set args
        sys_argv = [script_path] + args[1:]
        sys.argv = sys_argv

        # run
        with open(script_path, 'r') as f:
            exec(f.read(), {'__name__': '__main__'})

把這段代碼放到任意app的management/commands目錄下，命名為run_script.py即可，此后就可以一勞永逸地使用這個自定義的命令了。

此后，運行指定的腳本可參考以下的命令：

python manage.py run_script your_script arg1 arg2 -v arg3

Django extensions

此外還可以通過django-extensions這個包內置的runscript來實現(xiàn)，這個方法可參考文檔：https://django-extensions-zh.readthedocs.io/zh_CN/latest/runscript.html

以上是“Django執(zhí)行指定腳本的方法有哪些”這篇文章的所有內容，感謝各位的閱讀！希望分享的內容對大家有幫助，更多相關知識，歡迎關注億速云行業(yè)資訊頻道！