Skip to content
Snippets Groups Projects
Commit 8910a83f authored by aarif's avatar aarif
Browse files

updated the csv writer for python 3

re arranged the imports to fix quality errors
parent 955bf054
No related branches found
No related tags found
No related merge requests found
......@@ -8,9 +8,12 @@ from __future__ import absolute_import
import os
import os.path
import uuid
import csv
import unicodecsv
import six
from datetime import datetime
import unicodecsv
from bok_choy.promise import EmptyPromise
from pytz import UTC, utc
......@@ -322,7 +325,7 @@ class CohortConfigurationTest(EventsTestMixin, UniqueCourseTest, CohortTestMixin
"""
filename = self.instructor_dashboard_page.get_asset_path(filename)
with open(filename, 'w+') as csv_file:
writer = unicodecsv.writer(csv_file)
writer = csv.writer(csv_file) if six.PY3 else unicodecsv.writer(csv_file)
for line in csv_text_as_lists:
writer.writerow(line)
self.addCleanup(os.remove, filename)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment