Django執(zhí)行指定腳本的幾種方法
前言
用Django寫項目,遇到臨時需要處理數(shù)據(jù)庫的問題時,會考慮直接讓腳本在項目環(huán)境中運行,這樣就可以直接用定義好的模型和方法了。
這里提供幾種方法,通過Django執(zhí)行指定的腳本,讓腳本直接在Django項目的環(huán)境中來運行。
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這個包內(nèi)置的runscript來實現(xiàn),這個方法可參考文檔:https://django-extensions-zh.readthedocs.io/zh_CN/latest/runscript.html
總結
到此這篇關于Django執(zhí)行指定腳本的幾種方法的文章就介紹到這了,更多相關Django執(zhí)行指定腳本內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
基于python實現(xiàn)垂直爬蟲系統(tǒng)的方法詳解
這篇文章主要為大家詳細介紹了python實現(xiàn)垂直爬蟲系統(tǒng)的方法,文中示例代碼介紹的非常詳細,具有一定的參考價值,感興趣的小伙伴們可以參考一下,希望能夠給你帶來幫助2022-03-03Python?pygame派生精靈和精靈組創(chuàng)建敵機
這篇文章主要為大家介紹了Python?pygame派生精靈和精靈組創(chuàng)建敵機示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-08-08