1 from django
.core
.management
.base
import BaseCommand
, CommandError
2 from meta
.models
import DeckListModel
, TournamentModel
, RecordModel
3 from datetime
import datetime
7 class Command(BaseCommand
):
9 help = 'Import tournament data from CSV'
11 def handle(self
, *args
, **options
):
12 with
open('data/tournaments.csv') as f
:
13 for row
in csv
.DictReader(f
, delimiter
=','):
15 tournament
= TournamentModel
.objects
.get(name__iexact
=row
['name'])
16 except TournamentModel
.DoesNotExist
:
17 tournament
= TournamentModel(name
=row
['name'], date
=datetime
.strptime(row
['date'], '%Y-%m-%d'))
20 decklist
= DeckListModel(name
=row
['decklist_name'], url
=row
['decklist_url'])
23 record
= RecordModel(placement
=int(row
['placement']), tournament
=tournament
, decklist
=decklist
, verified
=True)