在生產系統中重置 Django 遷移

Question

自從幾年前開始使用該框架以來，我經歷了很多涉及 Django 遷移的帖子/文章/試錯，所以我決定發布一個自我回答的問題，說明完成清理的正確方法重置生產數據庫中的遷移，使您擁有與您離開時相同的數據庫結構，但重新開始初始遷移。

總的來說，問題是這樣的：

當您有一個更大的項目時，您開始為使用 Django 構建的系統積累大量遷移。 這通常不是問題，但是當您開始累積超過 50-100 個遷移文件（其中很多是相同字段的添加和刪除）時，最好有一個“清理”選項，因為它應該是眾所周知，如果您錯誤地更改遷移歷史記錄，您將留下一個或多或少凍結在以前的數據庫 state 中的系統，其中解決問題的唯一方法是手動基於 sql 的遷移更改。

Answer 1

我為這個問題提出的解決方案是分步驟的：

第 1 步創建遷移以刪除您想要的任何模型或字段並在本地運行它們，您的開發系統必須與所有其他開發人員系統以及生產系統同步......如果不是這種情況，您需要確保它是！

第2步

1. 刪除本地遷移文件（一個不錯的選擇是更改下面的命令，我目前有一個目錄結構，我的系統的應用程序位於一個名為/apps/的目錄中）

運行調用python manage.py delete_local_migration_files （如果你這樣命名的話）

import os

import django.apps
from django.conf import settings
from django.core.management.base import BaseCommand


def delete_migrations(app):
    print(f"Deleting {app}'s migration files")
    migrations_dir = os.path.join(settings.BASE_DIR, f'apps{os.path.sep}{app}{os.path.sep}migrations')

    if os.path.exists(migrations_dir):
        for the_file in os.listdir(migrations_dir):
            file_path = os.path.join(migrations_dir, the_file)
            try:
                if os.path.isfile(file_path):
                    os.unlink(file_path)
            except Exception as e:
                print(e)

            f = open(f"{os.path.join(migrations_dir, '__init__.py')}", "w")
            f.close()

    else:
        print('-' * 20, migrations_dir, 'does not exist')


class Command(BaseCommand):
    """
    Resets migrations and clears directories
    """
    help = 'reset migrations'

    def handle(self, *args, **options):
        set_of_apps = set()
        disregard = []

        # get all apps
        for model in django.apps.apps.get_models():
            if model._meta.app_label not in disregard:
                set_of_apps.add(model._meta.app_label)

        for app in set_of_apps:
            delete_migrations(app)

第 3 步

1. 從數據庫中刪除遷移（您可以使用下面的命令，它應該適用於任何使用 Postgres 的設置，但您必須根據需要更新連接字符串）

運行調用python manage.py delete_migrations_from_db （如果你這樣命名的話）

import os

import psycopg2
from django.conf import settings
from django.core.management import call_command
from django.core.management.base import BaseCommand
from django.db import connections


class Command(BaseCommand):
    help = 'Migrate on every database in settings.py'

    def handle(self, *args, **options):
        db_list = settings.DATABASES
        # del db_list['default']

        for db, _ in db_list.items():
            # we have the db name, now lets remove the migration tables in each

            try:
                host = os.environ['_HOST']
                user = os.environ['_USER']
                port = os.environ['_PORT']
                password = os.environ['_PASSWORD']

                conn_str = f"host={host} port={port} user={user} password={password}"

                conn = psycopg2.connect(conn_str)
                conn.autocommit = True
                with connections[db].cursor() as cursor:
                    delete_statement = 'DELETE from public.django_migrations'
                    cursor.execute(delete_statement)
                    print(f'Migration table cleared: {db}')
            except psycopg2.Error as ex:
                raise SystemExit(f'Error: {ex}')

        print('Done!')

步驟 4調用python manage.py makemigrations重新初始化初始遷移文件

步驟 5調用python manage.py migrate --database=[YourDB] --fake重新初始化初始遷移文件。 --fake參數允許在數據庫中恢復歷史記錄時不更改數據庫結構（如果您想要一個簡單的命令在所有數據庫中運行此遷移命令，您可以使用類似下面的代碼）

使用python manage.py migrate_all --fake調用（取決於命名）

from django.conf import settings
from django.core.management import call_command
from django.core.management.base import BaseCommand


class Command(BaseCommand):
    help = 'Migrate on every database in settings.py'


    def add_arguments(self, parser):
        parser.add_argument(
            '--fake',
            action='store_true',
            help='fake migrations',
        )

    def handle(self, *args, **options):
        db_list = settings.DATABASES


        for db, _ in db_list.items():
            self.stdout.write('Migrating database {}'.format(db))

            if options['fake']:
                call_command('migrate', '--fake', database=db)
            else:
                # no fake, call regularly
                call_command('migrate', database=db)


        self.stdout.write('Done!')