def test_assets_rows(self): rows = [item for item in get_assets_rows()] self.assertListEqual( rows, [ [ 'id', 'niw', 'barcode', 'sn', 'model__category__name', 'model__manufacturer__name', 'model__name', 'user__username', 'user__first_name', 'user__last_name', 'owner__username', 'owner__first_name', 'owner__last_name', 'status', 'service_name__name', 'property_of', 'warehouse__name', 'invoice_date', 'invoice_no', 'region__name', ], [ self.asset.id, 'niw=666', 'br-666', '1111-1111-1111-1111', 'Subcategory', 'Manufacturer1', 'Model1', 'user', 'Elmer', 'Stevens', 'owner', 'Eric', 'Brown', 1, None, None, 'Warehouse', datetime.date(2014, 4, 28), 'invoice-6666', 'Default region', ], ] )
def handle(self, *args, **options): only_licences = options['only_licences'] only_assets = options['only_assets'] filter_type = options['filter_type'] only_assigned_licences = options['only_assigned_licences'] if not any((only_licences, only_assets)): self.stdout.write( 'Arguments required, type --help for more informations\n', ) output = cStringIO.StringIO() writer = csv.writer(output) if only_assets and not only_licences: for row in get_assets_rows(filter_type): writer.writerow([smart_str(item) for item in row]) self.stdout.write(output.getvalue()) elif only_licences and not only_assets: for row in get_licences_rows(filter_type, only_assigned_licences): writer.writerow([smart_str(item) for item in row]) self.stdout.write(output.getvalue())
def test_assets_rows(self): rows = [item for item in get_assets_rows()] self.assertEqual(rows, [ [ 'id', 'niw', 'barcode', 'sn', 'model__category__name', 'model__manufacturer__name', 'model__name', 'user__username', 'user__first_name', 'user__last_name', 'owner__username', 'owner__first_name', 'owner__last_name', 'status', 'service_name__name', 'property_of', 'warehouse__name', ], [ 1, 'niw=666', 'br-666', '1111-1111-1111-1111', 'Subcategory', 'Manufacturer1', 'Model1', 'user', 'Elmer', 'Stevens', 'owner', 'Eric', 'Brown', 1, None, None, 'Warehouse', ], ])
def test_assets_rows(self): rows = [item for item in get_assets_rows()] self.assertEqual( rows, [ [ 'id', 'niw', 'barcode', 'sn', 'model__category__name', 'model__manufacturer__name', 'model__name', 'user__username', 'user__first_name', 'user__last_name', 'owner__username', 'owner__first_name', 'owner__last_name', 'status', 'service_name__name', 'property_of', 'warehouse__name', ], [ 1, 'niw=666', 'br-666', '1111-1111-1111-1111', 'Subcategory', 'Manufacturer1', 'Model1', 'user', 'Elmer', 'Stevens', 'owner', 'Eric', 'Brown', 1, None, None, 'Warehouse', ], ] )
def get_result(self, *args, **kwargs): filter_type = kwargs.get('mode') return list(get_assets_rows(filter_type=filter_type))