from vivopump import read_csv_fp, write_csv_fp, improve_course_title
import sys
data_in = read_csv_fp(sys.stdin)
var_names = data_in[
data_in.keys()[1]].keys() # create a list of var_names from the first row
print >> sys.stderr, "Columns in", var_names
data_out = {}
for row, data in data_in.items():
new_data = dict(data)
# Add these columns
new_data['remove'] = ''
new_data['uri'] = ''
new_data['title'] = improve_course_title(new_data['UF_COURSE_TITLE'])
new_data['ccn'] = new_data['UF_COURSE_CD']
# Delete all the upper case column names
for name in new_data.keys():
if name == name.upper():
del new_data[name]
data_out[row] = new_data
var_names = data_out[
data_out.keys()[1]].keys() # create a list of var_names from the first row
print >> sys.stderr, "Columns out", var_names
write_csv_fp(sys.stdout, data_out)
from vivopump import read_csv_fp, write_csv_fp, improve_course_title
import sys
data_in = read_csv_fp(sys.stdin)
var_names = data_in[data_in.keys()[1]].keys() # create a list of var_names from the first row
print >>sys.stderr, "Columns in", var_names
data_out = {}
for row, data in data_in.items():
new_data =dict(data)
# Add these columns
new_data['remove'] = ''
new_data['uri'] = ''
new_data['title'] = improve_course_title(new_data['UF_COURSE_TITLE'])
new_data['ccn'] = new_data['UF_COURSE_CD']
# Delete all the upper case column names
for name in new_data.keys():
if name == name.upper():
del new_data[name]
data_out[row] = new_data
var_names = data_out[data_out.keys()[1]].keys() # create a list of var_names from the first row
print >>sys.stderr, "Columns out", var_names
write_csv_fp(sys.stdout, data_out)
def test_substitution_at_end(self):
in_title = "HIST OF HLTHCARE"
out_title = improve_course_title(in_title)
print out_title
self.assertEqual("History of Healthcare", out_title)
def test_preserve_unicode(self):
in_title = u"SPEC TOP IN PRAC"
out_title = improve_course_title(in_title)
print out_title
self.assertEqual(u"Special Topics in Practice", out_title)
def test_simple_substitution(self):
in_title = "INTRO TO STAT"
out_title = improve_course_title(in_title)
print out_title
self.assertEqual("Introduction to Statistics", out_title)
