Skip to content

Instantly share code, notes, and snippets.

@Toshakins
Created September 18, 2020 17:47
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Toshakins/2e23761378f4f4c1748df080795f6d25 to your computer and use it in GitHub Desktop.
Save Toshakins/2e23761378f4f4c1748df080795f6d25 to your computer and use it in GitHub Desktop.
Run big migration with data from CSV file
import csv
from django.core import management
from django.core.management import BaseCommand
class Command(BaseCommand):
def add_arguments(self, parser):
parser.add_argument('csv_path', nargs='?', type=str)
def handle(self, *args, **options):
command_args = ["block_layout_switching_preparations", "--dry-run"]
with open(options['csv_path'], 'r') as f:
user_reader = csv.DictReader(f)
for row in user_reader:
command_args.append(f'--user-id={row["user_id"]}')
management.call_command(*command_args)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment