-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels.py
96 lines (87 loc) · 2.93 KB
/
models.py
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
88
89
90
91
92
93
94
95
96
from ufc_api import db, ma
#Event table
class Events(db.Model):
id = db.Column(db.Integer, primary_key=True)
event = db.Column(db.String(64),index=True)
location = db.Column(db.String(64))
date = db.Column(db.String(64))
def __repr__(self):
return '{}'.format(self.id)
#Bout Table
class Bouts(db.Model):
###Bout statistics
id = db.Column(db.Integer, primary_key=True)
event_id = db.Column(db.Integer, index=True)
weight_class = db.Column(db.String(32))
red_fighter = db.Column(db.String(64))
blue_fighter = db.Column(db.String(64))
winner = db.Column(db.String(64))
loser = db.Column(db.String(64))
result = db.Column(db.String(32))
end_round = db.Column(db.Integer)
time = db.Column(db.String(64))
method = db.Column(db.String(32))
###
###Striking Stats
###
b_KD = db.Column(db.Integer)
r_KD = db.Column(db.Integer)
b_SIGSTR = db.Column(db.String(12))
r_SIGSTR = db.Column(db.String(12))
b_SIGSTR_PRCT = db.Column(db.String(12))
r_SIGSTR_PRCT = db.Column(db.String(12))
b_TTLSTR = db.Column(db.String(12))
r_TTLSTR = db.Column(db.String(12))
b_TD= db.Column(db.Integer)
r_TD= db.Column(db.Integer)
b_TD_PRCT = db.Column(db.String(12))
r_TD_PRCT = db.Column(db.String(12))
b_SUB= db.Column(db.Integer)
r_SUB= db.Column(db.Integer)
b_PASS= db.Column(db.Integer)
r_PASS= db.Column(db.Integer)
class EventSchema(ma.SQLAlchemySchema):
class Meta:
model = Events
id = ma.auto_field()
event = ma.auto_field()
date = ma.auto_field()
location = ma.auto_field()
class BoutSchema(ma.SQLAlchemySchema):
class Meta:
model = Bouts
event_id = ma.auto_field()
weight_class = ma.auto_field()
red_fighter = ma.auto_field()
blue_fighter = ma.auto_field()
winner = ma.auto_field()
loser = ma.auto_field()
result = ma.auto_field()
end_round = ma.auto_field()
time = ma.auto_field()
method = ma.auto_field()
blue_stats = ma.Method("ser_bfighter")
red_stats = ma.Method("ser_rfighter")
### parse Bout fighter stats into a dict
def ser_bfighter(self,obj):
stats = dict()
stats["KD"] = obj.b_KD
stats["SIGSTR"] = obj.b_SIGSTR
stats["SIGSTR_PRCT"] = obj.b_SIGSTR_PRCT
stats["TTLSTR"] = obj.b_TTLSTR
stats["TD"] = obj.b_TD
stats["TD_PRCT"] = obj.b_TD_PRCT
stats["SUB"] = obj.b_SUB
stats["PASS"] = obj.b_PASS
return stats
def ser_rfighter(self,obj):
stats = dict()
stats["KD"] = obj.r_KD
stats["SIGSTR"] = obj.r_SIGSTR
stats["SIGSTR_PRCT"] = obj.r_SIGSTR_PRCT
stats["TTLSTR"] = obj.r_TTLSTR
stats["TD"] = obj.r_TD
stats["TD_PRCT"] = obj.r_TD_PRCT
stats["SUB"] = obj.r_SUB
stats["PASS"] = obj.r_PASS
return stats