football.py

import sys
import functools

def load_results(filename):
    with open(filename, 'r') as _f:
        raw = _f.read()

    lines = raw.split('\n')
    return [parse_result(line) for line in lines if line]

def parse_result(line):
    away_index = line.index(' - ') + 3
    result_index = line.index(' ', away_index) + 1
    goals = line[result_index:].split(':')
    return (line[:away_index - 3], line[away_index:result_index - 1], (int(goals[0]), int(goals[1])))

def calculate_points(results, team):
    return functools.reduce(lambda memo, result: memo + result_points(team, result[0], result[1], result[2]), results, 0)

def result_points(team, host, away, goals):
    if host == team and goals[0] > goals[1] or away == team and goals[0] < goals[1]:
        return 3
    elif goals[0] == goals[1] and (host == team or away == team):
        return 1
    else:
        return 0

if len(sys.argv) < 3:
    print('usage: football <stats-file> <team>')
else:
    results = load_results(sys.argv[1])
    print(calculate_points(results, sys.argv[2]))