· 8 years ago · Dec 29, 2017, 03:44 AM
1Id,PlayerName,TeamName
2641393,John Doe1,Team1
3663844,John Doe2,Team1
4607469,John Doe3,Team1
5
6import csv
7import sqlite3
8
9# connect to sqlite database
10conn = sqlite3.connect('players.db')
11c = conn.cursor()
12
13# table values
14table_spec = """CREATE TABLE IF NOT EXISTS players(
15 player_id INTEGER UNIQUE,
16 name_last TEXT,
17 name_first TEXT,
18 TeamName TEXT);"""
19
20
21# Create Table
22def create_table():
23 c.execute(table_spec)
24
25
26# Takes values and inserts them into the table
27def write_core_data(player_id, name_last, name_first):
28 c.execute('INSERT OR IGNORE
29 INTO players ("player_id","name_last","name_first")
30 VALUES (?,?,?)', (player_id,name_last, name_first))
31 conn.commit()
32
33
34def update_teams(player_id, TeamName):
35 c.execute('UPDATE players
36 SET TeamName=?
37 WHERE player_id=?',(TeamName, player_id))
38 conn.commit()
39
40
41# create_table()
42
43
44with open('players_master.csv') as csvfile:
45 readCSV = csv.reader(csvfile, delimiter=',')
46 for x in readCSV:
47 write_core_data(x[0], x[1], x[2])
48 print(x[0])
49
50with open('team_list.csv') as teamlist:
51 readCSV = csv.reader(teamlist, delimiter=',')
52 for x in readCSV:
53 update_teams(x[0], x[2])
54 print("Added team {} to {}".format(x[2], x[1]))
55
56c.close()
57conn.close()