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