Head First Python--读入字符串文件、替换、排序、去重以及使用函数 --使用函数进一步简化代码

import os
import pickle
import nester

os.chdir('d:/Python_test/ch5')

def sanitize(time_string):
    if '-' in time_string:
        splitter = '-'
    elif ':' in time_string:
        splitter = ':'
    else:
        return time_string
    (mins, secs) = time_string.split(splitter)
    return(mins + '.' + secs)

def get_coach_data(filename):
    try:
        with open(filename) as data_file:
            data = data_file.readline()
        return(data.strip().split(','))
    except IOError as err:
        print('File error! ' + str(err))
        return(None)

James = get_coach_data('james.txt')
Sarah = get_coach_data('sarah.txt')
Julie = get_coach_data('julie.txt')
Mikey = get_coach_data('mikey.txt')

print(sorted(set([sanitize(each_t) for each_t in James]))[0:3])
print(sorted(set([sanitize(each_t) for each_t in Sarah]))[0:3])
print(sorted(set([sanitize(each_t) for each_t in Julie]))[0:3])
print(sorted(set([sanitize(each_t) for each_t in Mikey]))[0:3])