Rafa/rafa.py at master · Anylytics/Rafa · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
from flask import Flask, render_template, abort, request, jsonify, g
import sqlite3

app = Flask(__name__)
DATABASE = 'rafadb'

def get_db():
    db = getattr(g, '_database', None)
    if db is None:
        db = g._database = connect_db()
    return db

@app.teardown_appcontext
def close_connection(exception):
    db = getattr(g, '_database', None)
    if db is not None:
        db.close()

def connect_db():
    return sqlite3.connect(DATABASE)


@app.route('/')
def home_page():
    return render_template('index.html')

def search_db(opponent):
	c = get_db().cursor()
	for row in c.execute('SELECT w_ace, w_df, w_svpt, w_1stIn, w_1stWon, w_2ndWon, l_ace, l_df, l_svpt, l_1stIn, l_1stWon, l_2ndWon, surface FROM matches where ( loser_name == ? )', [opponent]):
		o = list(row)
		o.append('r_win')
		yield o
	for row in c.execute('SELECT l_ace, l_df, l_svpt, l_1stIn, l_1stWon, l_2ndWon, w_ace, w_df, w_svpt, w_1stIn, w_1stWon, w_2ndWon, surface FROM matches where ( winner_name == ? )', [opponent]):
		o = list(row)
		o.append('o_win')
		yield o


@app.route('/versus/<opponent>/<surface>')
def find_crimes(opponent, surface):
	c = get_db().cursor()
	features = {}
	features['r_aces'] = 0
	features['r_df'] = 0
	features['r_svpt'] = 0
	features['r_1stIn'] = 0
	features['r_1stWon'] = 0
	features['r_2ndWon'] = 0
	features['r_win'] = {'Overall': 0, 'Clay' : 0, 'Carpet': 0, 'Hard': 0, 'Grass': 0}
	features['o_aces'] = 0
	features['o_df'] = 0
	features['o_svpt'] = 0
	features['o_1stIn'] = 0
	features['o_1stWon'] = 0
	features['o_2ndWon'] = 0
	features['o_win'] = {'Overall': 0, 'Clay' : 0, 'Carpet': 0, 'Hard': 0, 'Grass': 0}


	for row in search_db(opponent):
		#print row
		if ",".join(row).find(',,') == -1:
			if surface == 'Overall' or row[12] == surface:
				features['r_aces'] = features['r_aces'] + int(row[0])
				features['r_df'] = features['r_df'] + int(row[1])
				features['r_svpt'] = features['r_svpt'] + int(row[2])
				features['r_1stIn'] = features['r_1stIn'] + int(row[3])
				features['r_1stWon'] = features['r_1stWon'] + int(row[4])
				features['r_2ndWon'] = features['r_2ndWon'] + int(row[5])

				features['o_aces'] = features['o_aces'] + int(row[6])
				features['o_df'] = features['o_df'] + int(row[7])
				features['o_svpt'] = features['o_svpt'] + int(row[8])
				features['o_1stIn'] = features['o_1stIn'] + int(row[9])
				features['o_1stWon'] = features['o_1stWon'] + int(row[10])
				features['o_2ndWon'] = features['o_2ndWon'] + int(row[11])
		features[row[13]]['Overall'] = features[row[13]]['Overall'] + 1
		features[row[13]][row[12]] = features[row[13]][row[12]] + 1


	return jsonify(features)


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=3000, debug='True')