31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from django.core.management.base import BaseCommand, CommandError
|
|
from storage.models import Item
|
|
from io import StringIO
|
|
import csv
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Imports book library from specified wiki page dump'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('parent')
|
|
parser.add_argument('file')
|
|
|
|
def handle(self, *args, **options):
|
|
with open(options['file']) as fd:
|
|
sio = StringIO(fd.read())
|
|
reader = csv.reader(sio, delimiter='|')
|
|
parent = Item.objects.get(pk=options['parent'])
|
|
for line in reader:
|
|
line = list(map(str.strip, line))
|
|
item = Item(parent=parent)
|
|
item.name = line[2]
|
|
item.props['author'] = line[1]
|
|
item.props['owner'] = line[3]
|
|
item.props['can_borrow'] = line[4]
|
|
item.props['borrowed_by'] = line[5]
|
|
item.save()
|
|
|
|
self.stdout.write(self.style.NOTICE('Book added: %r') % item)
|
|
|
|
self.stdout.write(self.style.SUCCESS('Successfully imported data'))
|