如何在Django/Python/PostgreSQL演示环境中用户测试后重置数据库？

一、Cron Job 定时重置 #

完全可以用cron job实现，但它更适合固定周期自动重置（比如每小时/每天一次），而非用户完成测试后立即触发。具体步骤：

1. 编写Django自定义命令reset_demo_db：
   在app的management/commands目录下创建reset_demo_db.py：

   from django.core.management.base import BaseCommand
   from django.core.management import call_command
   
   class Command(BaseCommand):
       help = '重置演示数据库到初始状态'
   
       def handle(self, *args, **options):
           # 强制清空数据库，无需交互确认
           call_command('flush', '--noinput')
           # 加载预先准备的演示初始数据fixture
           call_command('loaddata', 'demo_initial_data.json')
           self.stdout.write(self.style.SUCCESS('演示数据库已重置完成'))
   

2. 配置cron定时任务：
   编辑crontab文件（crontab -e），添加定时规则，比如每小时执行一次：

   0 * * * * /path/to/your/venv/bin/python /path/to/your/project/manage.py reset_demo_db >> /var/log/demo_reset.log 2>&1
   

二、用户触发式重置（更贴合“测试完成后”需求） #

如果需要用户完成测试后手动触发重置，以下方案更合适：

1. 自定义Django重置视图

添加一个仅授权用户可访问的视图，执行数据库重置逻辑：

from django.core.management import call_command
from django.http import HttpResponseRedirect
from django.contrib.auth.decorators import user_passes_test

# 定义权限校验：仅超级管理员或演示管理员组用户可操作
def can_reset_demo_db(user):
    return user.is_superuser or user.groups.filter(name='DemoAdmins').exists()

@user_passes_test(can_reset_demo_db)
def reset_demo_database(request):
    # 执行重置操作
    call_command('flush', '--noinput')
    call_command('loaddata', 'demo_initial_data.json')
    # 重置后跳转回演示首页
    return HttpResponseRedirect('/demo/')

在urls.py中配置路由，前端添加一个“重置演示数据”按钮，点击后请求该接口即可。

2. PostgreSQL快照恢复

利用PostgreSQL的导出/导入功能，预先保存干净的数据库快照，重置时直接恢复：

1. 导出干净的演示数据库快照：

   pg_dump -U your_db_user -d demo_database > demo_clean_snapshot.sql
   

2. 编写重置脚本reset_demo_db.sh：

   #!/bin/bash
   # 清空当前数据库public schema
   psql -U your_db_user -d demo_database -c "DROP SCHEMA public CASCADE; CREATE SCHEMA public;"
   # 导入干净快照
   psql -U your_db_user -d demo_database < demo_clean_snapshot.sql
   

   可以通过Django命令调用这个脚本，或者直接在视图中用subprocess执行。

3. Docker容器化重置

如果演示应用用Docker部署，可通过临时卷或容器重启实现重置：

• 在docker-compose.yml中为数据库配置临时卷，容器重启后数据自动清空并重新初始化：

  services:
    db:
      image: postgres:15
      environment:
        - POSTGRES_USER=demo_user
        - POSTGRES_PASSWORD=demo_pass
        - POSTGRES_DB=demo_database
      volumes:
        # 挂载初始化脚本，容器启动时自动执行
        - ./demo_init.sql:/docker-entrypoint-initdb.d/demo_init.sql
        # 使用临时卷，容器重启后数据丢失
        - tmp_postgres_data:/var/lib/postgresql/data
    web:
      build: .
      command: python manage.py runserver 0.0.0.0:8000
      volumes:
        - .:/code
      ports:
        - "8000:8000"
      depends_on:
        - db
  
  volumes:
    tmp_postgres_data:
  

  每次需要重置时，只需重启db容器：docker-compose restart db，数据库会自动加载初始数据。

内容的提问来源于stack exchange，提问作者FBA