def connect_match_data_season_1617(): df = pd.read_csv("./data/season_files/season1617.csv").drop(["Unnamed: 0"], axis=1) n_rels = 0 n_nodes = 0 for _, row in df.iterrows(): # NODES # Hometeam ht = Team.nodes.get(name=row["Home/Neutral"]) # Visitor vt = Team.nodes.get(name=row["Visitor/Neutral"]) # HomeScore hs = Score(score=row["PTS.1"]).save() # VisitorScore vs = Score(score=row["PTS"]).save() # Arena a = ht.arena.get() # Season s = Season.get_or_create({"name": "2016/2017"})[0] # Date d = Date.get_or_create({ "datetime": datetime.strptime(row["Date"] + " " + row["Start (ET)"], "%Y-%m-%d %I:%M %p") })[0] # Game g = Game(game_name=row["game_name"], game_type=row["game_type"], ot=row["OT"]).save() n_nodes += 1 # RELATIONSHIPS # Team -> Score ht.scored.connect(hs) vt.scored.connect(vs) # Score -> Game hs.in_game.connect(g) vs.in_game.connect(g) # Game -> Season g.season.connect(s) # Game -> Date g.date.connect(d) # Game -> Arena g.arena.connect(a) n_rels += 7 print("created (approx:) {} nodes and {} relationships".format( n_nodes, n_rels))