load دیتا در migration

گاهی پیش میاد که یک سری جدول توی دیتابیس باید همیشه یک دیتای از پیش تعریف شده توش ذخیره بشه. در واقع وقتی جدول رو میسازیم و migrate میکنیم بهتره که بلافاصله دیتای خودمون رو توش ذخیره کنیم. برای این کار توی فایل migration میتونیم operations رو تعریف کنیم و یک تابع رو توش صدا بزنیم مثال زیر همین کار رو کرده و توی اون تابع جدول رو پر میکنه:

from django.db import migrations
import csv
import os
from django.conf import settings

def load_data_from_csv(apps, schema_editor):
    
    NameService = apps.get_model('tanks', 'NameService')  
    csv_file_path = os.path.join(settings.BASE_DIR, 'csv_storage/servicename.csv') 
	
    with open(csv_file_path, newline='', encoding='utf-8') as csvfile:
        reader = csv.DictReader(csvfile)
        data_to_insert = [
            NameService(
                name=row['name'],
                ename=row['ename'],
                pname=row['pname'],
                comment=row['comment']
            )
            for row in reader
        ]

        
        NameService.objects.bulk_create(data_to_insert)

class Migration(migrations.Migration):

    dependencies = [
        ('tanks', '0001_initial'),  
    ]

    operations = [
        migrations.RunPython(load_data_from_csv),
    ]

from django.db import migrations

import csv

import os

from django.conf import settings

def load_data_from_csv(apps, schema_editor):

NameService = apps.get_model('tanks', 'NameService')

csv_file_path = os.path.join(settings.BASE_DIR, 'csv_storage/servicename.csv')

with open(csv_file_path, newline='', encoding='utf-8') as csvfile:

reader = csv.DictReader(csvfile)

data_to_insert = [

NameService(

name=row['name'],

ename=row['ename'],

pname=row['pname'],

comment=row['comment']

)

for row in reader

]

NameService.objects.bulk_create(data_to_insert)

class Migration(migrations.Migration):

dependencies = [

('tanks', '0001_initial'),

]

operations = [

migrations.RunPython(load_data_from_csv),

]

دیدگاهتان را بنویسید لغو پاسخ