def test_natl_match_action_insert(session, natl_match_lineup):
nation, match, lineups = natl_match_lineup
events = [
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
period=1,
period_secs=0),
type=enums.ActionType.start_period),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
period=1,
period_secs=2902),
type=enums.ActionType.end_period)
]
session.add_all(events)
session.commit()
event_from_db = session.query(mn.NationalMatchEvents)
assert event_from_db.count() == 2
start_event = session.query(
mce.MatchActions).filter_by(type=enums.ActionType.start_period).one()
assert start_event.event == events[0].event
assert start_event.lineup_id is None
end_event = session.query(
mce.MatchActions).filter_by(type=enums.ActionType.end_period).one()
assert end_event.event == events[1].event
assert end_event.lineup_id is None
non_touch_events = session.query(mce.MatchActions).join(mn.NationalMatchEvents)\
.filter(mn.NationalMatchEvents.x.is_(None), mn.NationalMatchEvents.y.is_(None))
assert non_touch_events.count() == 2
assert set([rec.type for rec in non_touch_events]) == {
enums.ActionType.start_period, enums.ActionType.end_period
}
def test_natl_missed_pass_event(session, natl_match_lineup, modifiers):
nation, match, lineups = natl_match_lineup
modifier_objs = [
mce.Modifiers(type=enums.ModifierType.from_string(record['modifier']),
category=enums.ModifierCategoryType.from_string(
record['category'])) for record in modifiers
]
session.add_all(modifier_objs)
session.commit()
events = [
mce.MatchActions(
event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=921,
x=75.3,
y=98.4),
type=enums.ActionType.ball_pass,
lineup_id=lineups[0].id,
x_end=82.2,
y_end=75.0,
),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1143,
x=67.3,
y=62.1),
type=enums.ActionType.ball_pass,
lineup_id=lineups[0].id,
x_end=82.2,
y_end=50.0,
is_success=False)
]
session.add_all(events)
session.commit()
pass_actions = session.query(mce.MatchActions).join(mn.NationalMatchEvents).\
filter(mce.MatchActions.type == enums.ActionType.ball_pass)
assert pass_actions.count() == 2
missed_passes = pass_actions.filter(
mce.MatchActions.is_success.in_([False]))
assert missed_passes.count() == 1
missed_pass_event = session.query(mn.NationalMatchEvents).get(
missed_passes[0].event_id)
assert missed_pass_event.period_secs == 1143
assert missed_pass_event.x == 67.3
assert missed_pass_event.y == 62.1
assert missed_pass_event.team_id == nation.id
assert missed_passes[0].lineup_id == lineups[0].id
def test_natl_match_action_insert(session, natl_match_lineup):
nation, match, lineups = natl_match_lineup
events = [
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
period=1,
period_secs=0),
type=enums.ActionType.start_period),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1143,
x=67.3,
y=62.1),
type=enums.ActionType.ball_pass,
lineup_id=lineups[0].id,
x_end=82.2,
y_end=50.0),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1150,
x=84.0,
y=48.8),
type=enums.ActionType.goal,
lineup_id=lineups[1].id,
x_end=100.0,
y_end=45.0,
z_end=10.0)
]
session.add_all(events)
session.commit()
pass_actions = session.query(mce.MatchActions).join(mn.NationalMatchEvents).\
filter(mce.MatchActions.type == enums.ActionType.ball_pass)
assert pass_actions.count() == 1
pass_event = session.query(mn.NationalMatchEvents).get(
pass_actions[0].event_id)
assert pass_event.period == 1
assert pass_event.period_secs == 1143
assert pass_event.x == 67.3
assert pass_event.y == 62.1
def test_natl_goal_event_view(session, natl_match_lineup):
nation, match, lineups = natl_match_lineup
events = [
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
period=1,
period_secs=0),
type=enums.ActionType.start_period),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1143,
x=67.3,
y=62.1),
type=enums.ActionType.ball_pass,
lineup_id=lineups[0].id,
x_end=82.2,
y_end=50.0),
mce.MatchActions(event=mn.NationalMatchEvents(match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1150,
x=84.0,
y=48.8),
type=enums.ActionType.goal,
lineup_id=lineups[1].id,
x_end=100.0,
y_end=45.0,
z_end=10.0)
]
session.add_all(events)
session.commit()
goal_events = session.query(mn.NationalGoals)
assert goal_events.count() == 1
assert goal_events[0].match_id == match.id
assert goal_events[0].lineup_id == lineups[1].id
assert goal_events[0].team_id == lineups[1].team_id
assert goal_events[0].period == 1
assert goal_events[0].period_secs == 1150
assert goal_events[0].x == 84.0
assert goal_events[0].y == 48.8
def actions(self, data_frame):
action_set = set()
action_records = []
lineup_dict = None
modifier_ids = []
local_ids = []
action_fields = [
'event_id', 'type', 'x_end', 'y_end', 'z_end', 'is_success',
'match_id', 'player_id', 'modifier_type'
]
for _, row in data_frame.iterrows():
action_set.add(
tuple([(field, row[field]) for field in action_fields
if field in row and row[field] is not None]))
logger.info("{} unique actions".format(len(action_set)))
for indx, elements in enumerate(action_set):
if indx and indx % 100 == 0:
logger.info("Processing {} actions".format(indx))
action_dict = dict(elements)
match_id = action_dict.pop('match_id')
player_id = action_dict.pop('player_id', None)
modifier_type = action_dict.pop('modifier_type', None)
if not lineup_dict:
records = self.session.query(
mcm.MatchLineups).filter_by(match_id=match_id).all()
lineup_dict = {rec.player_id: rec.id for rec in records}
if player_id:
action_dict['lineup_id'] = lineup_dict[player_id]
if modifier_type:
try:
modifier_id = self.get_id(
mce.Modifiers,
type=enums.ModifierType.from_string(modifier_type))
except ValueError as ex:
logger.info(elements)
raise ex
else:
modifier_id = None
# if not self.record_exists(mce.MatchActions, **action_dict):
action_dict.update(id=uuid.uuid4())
action_records.append(mce.MatchActions(**action_dict))
modifier_ids.append(modifier_id)
local_ids.append(action_dict['id'])
self.session.bulk_save_objects(action_records)
modifier_records = [
mce.MatchActionModifiers(action_id=local_id,
modifier_id=modifier_id)
for modifier_id, local_id in zip(modifier_ids, local_ids)
]
self.session.bulk_save_objects(modifier_records)
self.session.commit()
def test_natl_goal_modifier_insert(session, natl_match_lineup, modifiers):
nation, match, lineups = natl_match_lineup
modifier_objs = [
mce.Modifiers(type=enums.ModifierType.from_string(record['modifier']),
category=enums.ModifierCategoryType.from_string(
record['category'])) for record in modifiers
]
session.add_all(modifier_objs)
session.commit()
goal_action = mce.MatchActions(event=mn.NationalMatchEvents(
match_id=match.id,
team_id=nation.id,
period=1,
period_secs=1150,
x=84.0,
y=48.8),
type=enums.ActionType.goal,
lineup_id=lineups[1].id,
x_end=100.0,
y_end=45.0,
z_end=10.0)
session.add(goal_action)
session.commit()
goal_modifiers = [
mce.MatchActionModifiers(
action_id=goal_action.id,
modifier_id=session.query(mce.Modifiers.id).filter_by(
type=enums.ModifierType.head).scalar()),
mce.MatchActionModifiers(
action_id=goal_action.id,
modifier_id=session.query(mce.Modifiers.id).filter_by(
type=enums.ModifierType.center_penalty_area).scalar()),
mce.MatchActionModifiers(
action_id=goal_action.id,
modifier_id=session.query(mce.Modifiers.id).filter_by(
type=enums.ModifierType.lower_right).scalar())
]
session.add_all(goal_modifiers)
session.commit()
action_from_db = session.query(
mn.NationalGoals).filter_by(action_id=goal_action.id).one()
goal_body_part = session.query(mce.Modifiers).join(mce.MatchActionModifiers)\
.filter(mce.MatchActionModifiers.action_id == action_from_db.action_id,
mce.Modifiers.category == enums.ModifierCategoryType.bodypart)
assert goal_body_part.count() == 1
assert goal_body_part[0].type.value == "Head"
goal_sector = session.query(mce.Modifiers).join(mce.MatchActionModifiers)\
.filter(mce.MatchActionModifiers.action_id == action_from_db.action_id,
mce.Modifiers.category == enums.ModifierCategoryType.field_sector)
assert goal_sector.count() == 1
assert goal_sector[0].type.value == "Central Penalty Area"
goal_goal_region = session.query(mce.Modifiers).join(mce.MatchActionModifiers)\
.filter(mce.MatchActionModifiers.action_id == action_from_db.action_id,
mce.Modifiers.category == enums.ModifierCategoryType.goal_region)
assert goal_goal_region.count() == 1
assert goal_goal_region[0].type.value == "Lower Right"
