文章目录
- 0.思路引导
- 1.脚本目录结构
- 2.使用 Faker 快速生成测试数据
- 3.批量生成测试数据
- 4.执行脚本
- 5.效果展示
0.思路引导
1)为了防止博客首页展示的文章过多以及提升加载速度,可以对文章列表进行分页展示。
2)不过这需要比较多的文章才能达到分页效果,但本地开发时一般都只有几篇测试文章,如果一篇篇手工添加将会非常麻烦。
3)解决方案是我们可以写一个脚本,自动生成任意数量的测试数据。脚本写好后,只需运行脚本就可以往数据库填充大量测试数据。
1.脚本目录结构
在项目根目录下新建scripts 文件夹,并新建__init__.py,fake.py,md.sample三个文件,目录结构如下:
HelloDjango-blog-tutorial\blog\blogproject\...scripts\__init__.pyfake.pymd.sample
注意:
1)其中 fake.py 是生成测试数据的脚本;
2)md.sample 是一个纯文本文件,内容是生成用于测试 Markdown 的文本。
2.使用 Faker 快速生成测试数据
博客文章包含丰富的内容元素,例如标题、正文、分类、标签。如果手工输入这些相关元素的文本会非常耗时,我们将借助一个 Python 的第三方库 Faker (造假工厂)来快速生成这些测试用的文本内容。
首先安装 Faker:
$ pipenv install Faker
Faker 通过不同的 Provider 来提供各种不同类型的假数据,我们将在下面的脚本中讲解它的部分用法,完整的用法可以参考其官方文档。
3.批量生成测试数据
文件位置:HelloDjango-blog-tutorial \scropts \fake.py
1)首先,导入一些会用到的模块,然后通过脚本所在文件找到项目根目录,将根目录添加到 Python 的模块搜索路径中,这样在运行脚本时 Python 才能够找到相应的模块并执行;
import os
import pathlib
import random
import sys
from datetime import timedeltaimport django
import faker
from django.utils import timezone# 将项目根目录添加到 Python 的模块搜索路径中
back = os.path.dirname
BASE_DIR = back(back(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
2)设置 DJANGO_SETTINGS_MODULE 环境变量,这将指定 django 启动时使用的配置文件,然后运行 django.setup() 启动 django。这是关键步骤,只有在 django 启动后,我们才能使用 django 的 ORM 系统。django 启动后,就可以导入各个模型,以便创建数据。
if __name__ == '__main__':os.environ.setdefault("DJANGO_SETTINGS_MODULE", "blogproject.settings.local")django.setup()from blog.models import Category, Post, Tagfrom comments.models import Commentfrom django.contrib.auth.models import User
3)清除旧数据,因此每次运行脚本,都会清除原有数据,然后重新生成。
print('clean database')Post.objects.all().delete()Category.objects.all().delete()Tag.objects.all().delete()Comment.objects.all().delete()User.objects.all().delete()
4)使用 django 的 ORM API 生成博客用户、分类、标签以及一篇 Markdown 测试文章。
print('create a blog user')user = User.objects.create_superuser('admin', 'admin@hellogithub.com', 'admin')category_list = ['Python学习笔记', '开源项目', '工具资源', '程序员生活感悟', 'test category']tag_list = ['django', 'Python', 'Pipenv', 'Docker', 'Nginx', 'Elasticsearch', 'Gunicorn', 'Supervisor', 'test tag']a_year_ago = timezone.now() - timedelta(days=365)print('create categories and tags')for cate in category_list:Category.objects.create(name=cate)for tag in tag_list:Tag.objects.create(name=tag)print('create a markdown sample post')Post.objects.create(title='Markdown 与代码高亮测试',body=pathlib.Path(BASE_DIR).joinpath('scripts', 'md.sample').read_text(encoding='utf-8'),category=Category.objects.create(name='Markdown测试'),author=user,)
5)生成 100 篇英文博客文章。博客文章通常内容比较长,因此我们使用了之前提及的 Faker 库来自动生成文本内容
print('create some faked posts published within the past year')fake = faker.Faker() # Englishfor _ in range(100):tags = Tag.objects.order_by('?')tag1 = tags.first()tag2 = tags.last()cate = Category.objects.order_by('?').first()created_time = fake.date_time_between(start_date='-1y', end_date="now",tzinfo=timezone.get_current_timezone())post = Post.objects.create(title=fake.sentence().rstrip('.'),body='\n\n'.join(fake.paragraphs(10)),created_time=created_time,category=cate,author=user,)post.tags.add(tag1, tag2)post.save()
注意:
-
fake = faker.Faker(),要使用 Faker 自动生成数据,首先实例化一个 Faker 对象,然后我们可以在脚本中使用这个实例的一些方法生成需要的数据。Faker默认生成英文数据,但也支持国际化。至于如何生成中文数据在下一段脚本中会看到。
-
order_by(’?’) 将返回随机排序的结果,脚本中这块代码的作用是达到随机选择标签(Tag) 和分类(Category) 的效果。
-
然后就是 2 个 Faker 的 API 了:
fake.date_time_between 这个方法将返回 2个指定日期间的随机日期。三个参数分别是起始日期,终止日期和时区。我们在这里设置起始日期为 1年前(-1y),终止日期为当下(now),时区为 get_current_timezone 返回的时区,这个函数是django.utils.timezone 模块的辅助函数,它会根据 django 设置文件中 TIME_ZONE的值返回对应的时区对象。 -
‘\n\n’.join(fake.paragraphs(10)) fake.paragraphs(10) 用于生成 10 个段落文本,以列表形式返回,列表的每个元素即为一个段落。要注意使用 2 个换行符连起来是为了符合 Markdown 语法,Markdown中只有 2 个换行符分隔的文本才会被解析为段落。
6)构造 Faker 实例,传入了一个语言代码 zh_CN,这将生成中文的虚拟数据,而不是默认的英文。
fake = faker.Faker('zh_CN')for _ in range(100): # Chinesetags = Tag.objects.order_by('?')tag1 = tags.first()tag2 = tags.last()cate = Category.objects.order_by('?').first()created_time = fake.date_time_between(start_date='-1y', end_date="now",tzinfo=timezone.get_current_timezone())post = Post.objects.create(title=fake.sentence().rstrip('.'),body='\n\n'.join(fake.paragraphs(10)),created_time=created_time,category=cate,author=user,)post.tags.add(tag1, tag2)post.save()
7)给前 20 篇文章(Post) 生成评论数据。要注意的是评论的发布时间必须位于被评论文章的发布时间和当前时间之间,这就是 delta_in_days = ‘-’ + str((timezone.now() - post_created_time).days) + ‘d’ 这句代码的作用。
print('create some comments')for post in Post.objects.all()[:20]:post_created_time = post.created_timedelta_in_days = '-' + str((timezone.now() - post_created_time).days) + 'd'for _ in range(random.randrange(3, 15)):Comment.objects.create(name=fake.name(),email=fake.email(),url=fake.uri(),text=fake.paragraph(),created_time=fake.date_time_between(start_date=delta_in_days, end_date="now", tzinfo=timezone.get_current_timezone()),post=post,)print('done!')
4.执行脚本
在项目根目录执行下面的命令运行整个脚本:
$ pipenv run python -m scripts.fake
看到如下的输出说明脚本执行成功了。
clean database
create a blog user
create categories and tags
create a markdown sample post
create some faked posts published within the past year
create some comments
done!
5.效果展示
此时,我们有了 200 多篇测试文章,用来测试分页效果就十分简单了,接下来我们将实现功能完整的分页效果。