def test_get_field_names_to_extract(self): # field names to extract should be the same # as the field names to serialise self.assertEqual( Episode._get_fieldnames_to_serialize(), Episode._get_fieldnames_to_extract() )
def episode_csv(episodes, user, file_name): """ Given an iterable of EPISODES, create a CSV file containing Episode details. """ with open(file_name, "w") as csv_file: fieldnames = Episode._get_fieldnames_to_serialize() fieldnames.remove('consistency_token') headers = list(fieldnames) headers.append("tagging") writer = csv.DictWriter(csv_file, fieldnames=headers) writer.writeheader() for episode in episodes: row = {h: unicode(getattr(episode, h)).encode('UTF-8') for h in fieldnames} row["tagging"] = ';'.join(episode.get_tag_names(user, historic=True)) writer.writerow(row)
def episode_csv(episodes, user, file_name): """ Given an iterable of EPISODES, create a CSV file containing Episode details. """ with open(file_name, "w") as csv_file: fieldnames = Episode._get_fieldnames_to_serialize() fieldnames.remove('consistency_token') headers = list(fieldnames) headers.append("tagging") writer = csv.DictWriter(csv_file, fieldnames=headers) writer.writeheader() for episode in episodes: row = { h: unicode(getattr(episode, h)).encode('UTF-8') for h in fieldnames } row["tagging"] = ';'.join( episode.get_tag_names(user, historic=True)) writer.writerow(row)
def test_get_field_names_to_extract(self): # field names to extract should be the same # as the field names to serialise self.assertEqual(Episode._get_fieldnames_to_serialize(), Episode._get_fieldnames_to_extract())