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'))