Python RetrosheetLineParser Examples

Example #1

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_badj():
	rsp = RetrosheetLineParser()
	retro_list = ['badj','rodrw002','R']
	id_map = rsp.parse_badj(retro_list)
	print(id_map)
	assert(all([id_map['player_id'] == 'rodrw002',
		id_map['badj_note'] == 'R'
		]))

Example #2

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_padj():
	rsp = RetrosheetLineParser()
	retro_list = ['padj','harrg001','L']
	id_map = rsp.parse_padj(retro_list)
	print(id_map)
	assert(all([id_map['player_id'] == 'harrg001',
		id_map['padj_note'] == 'L'
		]))

Example #3

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_data():
	rsp = RetrosheetLineParser()
	retro_list = ['data','er','durbc001','3']
	id_map = rsp.parse_data(retro_list)
	print(id_map)
	assert(all([id_map['player_id'] == 'durbc001', 
		id_map['earned_runs'] == '3'
		]))

Example #4

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_sub():
	rsp = RetrosheetLineParser()
	retro_list = ['sub','happj001','"J.A. Happ"','0','7','1']
	id_map = rsp.parse_sub(retro_list)
	print(id_map)
	assert(all([id_map['player_id'] == 'happj001', 
		id_map['player_name'] == '"J.A. Happ"', 
		id_map['player_team'] == '0', 
		id_map['batting_order'] == '7',
		id_map['field_position'] == '1'
		]))

Example #5

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_id():
    rsp = RetrosheetLineParser()
    retro_list = ['id','DET198204150']
    id_map = rsp.parse_id(retro_list)
    print(id_map)
    assert(all([id_map['team'] == 'DET',
        id_map['year'] == '1982',
        id_map['month'] == '04',
        id_map['day'] == '15',
        id_map['game_num'] == '0'
        ]))

Example #6

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_start():
	rsp = RetrosheetLineParser()
	retro_list = ['start','utlec001','"Chase Utley"','0','3','4']
	id_map = rsp.parse_start(retro_list)
	print(id_map)
	assert(all([id_map['player_id'] == 'utlec001', 
		id_map['player_name'] == '"Chase Utley"', 
		id_map['player_team'] == '0', 
		id_map['batting_order'] == '3', 
		id_map['field_position'] == '4'
		]))

Example #7

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_play():
	rsp = RetrosheetLineParser()
	retro_list = ['play','1','1','spilr001','12','>B.SFX','FC6/G.2X3(65)']
	id_map = rsp.parse_play(retro_list)
	print(id_map)
	assert(all([id_map['inning'] == '1', 
		id_map['home_team'] == '1', 
		id_map['player_id'] == 'spilr001', 
		id_map['rts_pitch_count'] == '12',
		id_map['rts_pitch_seq'] == '>B.SFX',
		id_map['play_meta'] == 'FC6/G.2X3(65)'
		]))

Example #8

File: RetrosheetTableBuilder.py Project: fishpm/bball_prj_games

    def rts_line_parser(self, file_path, tmp_dir=''):
        """From a retrosheet file, parse it into a relational table"""

        rsp = RetrosheetLineParser()
        tmp_f = open(join(tmp_dir, 'temp.txt'), 'wb')
        tmp_output = csv.writer(tmp_f, quoting = csv.QUOTE_ALL)
        with open(self.file_path,'r') as f:
            for line in f:
                line_parts = line.rstrip().split(',')
         		parsed_line = rsp.type_parser(line_parts, a.parser_map)
         		if parsed_line:
         			parsed_line_list = list(itertools.chain.from_iterable([[k, parsed_line[k]] for k in parsed_line]))
         			tmp_output.writerow(parsed_line_list)

Example #9

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_com():
	rsp = RetrosheetLineParser()
	retro_list = ['com',"Rockies first baseman Todd Helton left the game due to an injured head"]
	id_map = rsp.parse_com(retro_list)
	print(id_map)
	assert(id_map['comment'] == "Rockies first baseman Todd Helton left the game due to an injured head")

Example #10

File: test_etl.py Project: fishpm/bball_prj_games

def test_parse_info():
    rsp = RetrosheetLineParser()
    retro_list = ['info','hometeam','COL']
    id_map = rsp.parse_info(retro_list)
    print(id_map)
    assert(id_map['hometeam'] == 'COL')