def test_import_idempotent(self):
for i in range(3):
import_guests(self.path)
self.assertEqual(3, Party.objects.count())
self.assertEqual(5, Guest.objects.count())
the_starks = Guest.objects.filter(party__name='The Starks')
self.assertEqual(3, the_starks.count())
def handle(self, *args, **kwargs):
try:
with open(kwargs['filename']):
pass
except:
print 'Failed to open {}'.format(kwargs['filename'])
else:
csv_import.import_guests(kwargs['filename'])
def handle(self, *args, **kwargs):
if kwargs.get('file'):
print("importing FILE {}.".format(kwargs['file']))
csv_import.import_guests(kwargs['file'])
elif kwargs.get('url'):
print("importing URL {}.".format(kwargs['url']))
csv_import.import_guests_url(kwargs['url'])
else:
print("Please provde a URL or FILE.")
exit(1)
def setUpClass(cls):
super(GuestImporterTest, cls).setUpClass()
cls.path = os.path.join(os.path.dirname(__file__), 'data', 'guests-test.csv')
import_guests(cls.path)
def test_email_default(self):
import_guests(self.path)
self.assertEqual(None, Guest.objects.get(first_name='Tyrion').email)
def test_email(self):
import_guests(self.path)
self.assertEqual('*****@*****.**',
Guest.objects.get(first_name='Ned').email)
def test_is_invited(self):
import_guests(self.path)
for party in Party.objects.all():
self.assertTrue(party.is_invited)
def handle(self, *args, **kwargs):
filename = kwargs['filename']
csv_import.import_guests(filename)
def handle(self, filename, *args, **kwargs):
csv_import.import_guests(filename)
def handle(self, *args, **kwargs):
csv_import.import_guests(kwargs['filename'])
def handle(self, filename, *args, **kwargs):
print filename
csv_import.import_guests(filename[0])
def handle(self, *args, **options):
csv_import.import_guests(options['csv'])