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