Django笔记四十之运行Django环境的python脚本

欧米尼亚

本文首发于公众号：Hunter后端
原文链接：Django笔记四十之运行Django环境的python脚本

这一篇笔记介绍如何在 Django 中运行脚本。
假设说我们要实现一个功能，需要获取 blog.models.Blog 这张表里的总数且使用 print() 输出。
如果代码逻辑很短，且是一次性执行的操作，我们可以在系统的根目录下使用 python3 manage.py shell 在 shell 中操作实现。

1. from blog.models import Blog
2. print(Blog.objects.all().count())

复制代码

但是如果这个需求是一个很长的逻辑，且需要可以重复执行，那么应该如何操作呢？
首先，我们需要可以用到 Django 里的 Blog model，然后使用 model 的方式来进行查询。
我们可以将其写入一个文件，每次都使用 shell 来调用。
写入文件，在 shell 调用

比如，我们可以在 blog 创建一个 scripts 文件夹，然后创建一个 blog_script.py 文件：

1. # blog/scripts/blog_script.py
3. from blog.models import Blog
5. def get_blog_count():
6.     print(Blog.objects.all().count())

复制代码

然后每次调用的时候就进入 python3 manage.py shell 操作：

1. from blog.scripts.blog_script import get_blog_count
3. get_blog_count()

复制代码

直接运行脚本

如果我们想创建一个文件，然后直接运行该脚本，那么就需要在脚本中引入当前 Django 系统环境的一些参数和变量，可以直接在系统根目录下创建一个 blog_script.py 文件：

1. # blog_script.py
3. import django, os
5. os.environ.setdefault("DJANGO_SETTINGS_MODULE", "hunter.settings")
6. django.setup()
8. from blog.models import Blog
10. if __name__ == "__main__":
11.     print(Blog.objects.count())

复制代码

然后我们在系统根目录下直接执行该脚本：

1. python3 blog_script.py

复制代码

这个文件目前只有放在系统的根目录下才可以运行，但是如果太多的脚本都放到系统根目录下也不太好，所以我们如果有这种脚本一般都可以放到对应的 application 的 scripts 文件夹下。
如果我们想要直接在 blog/scripts/ 下直接运行该脚本的话，则需要在我们的脚本中添加 Django 系统的路径，假设我们的 Django 系统名为 hunter，在服务器的路径为：/Users/xxxx/hunter/，我们可以如下操作：

1. # blog/scripts/
3. import django, os, sys
5. django_path = "/Users/xxxx/hunter/"
6. sys.path.extend([django_path])
7. os.environ.setdefault("DJANGO_SETTINGS_MODULE", "hunter.settings")
9. django.setup()
11. from blog.models import Blog
13. if __name__ == "__main__":
14.     print(Blog.objects.count())

复制代码

如果想获取更多后端相关文章，可扫码关注阅读：

来源:https://www.cnblogs.com/hunterxiong/p/17396431.html
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！