forked from ghazi-arman/dfs-lineup-generator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathget_top_scores.py
executable file
·30 lines (27 loc) · 1.04 KB
/
get_top_scores.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
import os
import sys
import csv
directory = '{}/outputs/{}'.format(sys.argv[1], sys.argv[2])
top_scores=[]
for filename in os.listdir(directory):
# if we want to limit output files by certain parameters (date, player_limit, overlap, etc.)
if os.path.isfile('{}/outputs/{}/{}'.format(sys.argv[1], sys.argv[2], filename)):
if (len(sys.argv) >= 5):
contains_queries = True
queries = sys.argv[4].split(",")
for query in queries:
if query not in filename:
contains_queries = False
if (len(sys.argv) <= 4 or contains_queries):
# iterate through each output file and take top X scores
with open('{}/outputs/{}/{}'.format(sys.argv[1], sys.argv[2], filename),'rt') as f:
data = csv.reader(f)
scores=[]
for row in data:
if row[0] == 'PG' or row[0] == 'C' or row[0] == 'QB':
continue
scores.append((float(row[10]), filename))
scores.sort(reverse = True)
top_scores += scores[:int(sys.argv[3])]
top_scores.sort(reverse = True)
print(top_scores)