快捷導(dǎo)航

Django執(zhí)行指定腳本的幾種方法

更新時(shí)間：2022年02月08日 12:40:47 作者：knktc

這篇文章主要給大家介紹了關(guān)于Django執(zhí)行指定腳本的幾種方法,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用Django具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

前言

用Django寫項(xiàng)目，遇到臨時(shí)需要處理數(shù)據(jù)庫(kù)的問(wèn)題時(shí)，會(huì)考慮直接讓腳本在項(xiàng)目環(huán)境中運(yùn)行，這樣就可以直接用定義好的模型和方法了。

這里提供幾種方法，通過(guò)Django執(zhí)行指定的腳本，讓腳本直接在Django項(xiàng)目的環(huán)境中來(lái)運(yùn)行。

Django Shell

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

python manage.py shell < script.py

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

自定義Command

也可以通過(guò)自定義一個(gè)簡(jiǎn)單的command來(lái)實(shí)現(xiàn)需要的效果，以下是最近寫的一個(gè)示例，可支持執(zhí)行指定的腳本文件，同時(shí)還能加入更多的腳本執(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即可，此后就可以一勞永逸地使用這個(gè)自定義的命令了。

此后，運(yùn)行指定的腳本可參考以下的命令：

python manage.py run_script your_script arg1 arg2 -v arg3

Django extensions

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