James McCool commited on
Commit
0ff4b2c
·
1 Parent(s): f31fe20

Increase database query limits in lineup initialization for DraftKings and FanDuel, enhancing data retrieval for player selections across various slate types.

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +36 -36
src/streamlit_app.py CHANGED
@@ -242,11 +242,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
242
 
243
  collection = db['DK_NFL_seed_frame']
244
  if prio_var == None:
245
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
246
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
247
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
248
  else:
249
- cursor = collection.find().sort(prio_var, -1).limit(10000)
250
  raw_display = pd.DataFrame(list(cursor))
251
 
252
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -262,11 +262,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
262
 
263
  collection = db['DK_NFL_Secondary_seed_frame']
264
  if prio_var == None:
265
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
266
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
267
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
268
  else:
269
- cursor = collection.find().sort(prio_var, -1).limit(10000)
270
  raw_display = pd.DataFrame(list(cursor))
271
 
272
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -282,11 +282,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
282
 
283
  collection = db['DK_NFL_Late_seed_frame']
284
  if prio_var == None:
285
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
286
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
287
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
288
  else:
289
- cursor = collection.find().sort(prio_var, -1).limit(10000)
290
  raw_display = pd.DataFrame(list(cursor))
291
 
292
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -297,11 +297,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
297
  if slate_var == 'Main':
298
  collection = db['DK_NFL_SD_seed_frame']
299
  if prio_var == None:
300
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
301
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
302
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
303
  else:
304
- cursor = collection.find().sort(prio_var, -1).limit(10000)
305
  raw_display = pd.DataFrame(list(cursor))
306
 
307
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -310,11 +310,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
310
  elif slate_var == 'Secondary':
311
  collection = db['DK_NFL_Secondary_SD_seed_frame']
312
  if prio_var == None:
313
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
314
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
315
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
316
  else:
317
- cursor = collection.find().sort(prio_var, -1).limit(10000)
318
  raw_display = pd.DataFrame(list(cursor))
319
 
320
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -323,11 +323,11 @@ def init_DK_lineups(type_var, slate_var, prio_var, prio_mix, db_translation):
323
  elif slate_var == 'Auxiliary':
324
  collection = db['DK_NFL_Auxiliary_SD_seed_frame']
325
  if prio_var == None:
326
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
327
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
328
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
329
  else:
330
- cursor = collection.find().sort(prio_var, -1).limit(10000)
331
  raw_display = pd.DataFrame(list(cursor))
332
 
333
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -355,11 +355,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
355
 
356
  collection = db['FD_NFL_seed_frame']
357
  if prio_var == None:
358
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
359
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
360
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
361
  else:
362
- cursor = collection.find().sort(prio_var, -1).limit(10000)
363
  raw_display = pd.DataFrame(list(cursor))
364
 
365
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -375,11 +375,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
375
 
376
  collection = db['FD_NFL_Secondary_seed_frame']
377
  if prio_var == None:
378
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
379
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
380
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
381
  else:
382
- cursor = collection.find().sort(prio_var, -1).limit(10000)
383
  raw_display = pd.DataFrame(list(cursor))
384
 
385
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -395,11 +395,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
395
 
396
  collection = db['FD_NFL_Late_seed_frame']
397
  if prio_var == None:
398
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
399
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
400
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
401
  else:
402
- cursor = collection.find().sort(prio_var, -1).limit(10000)
403
  raw_display = pd.DataFrame(list(cursor))
404
 
405
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -411,11 +411,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
411
  if slate_var == 'Main':
412
  collection = db['FD_NFL_SD_seed_frame']
413
  if prio_var == None:
414
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
415
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
416
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
417
  else:
418
- cursor = collection.find().sort(prio_var, -1).limit(10000)
419
  raw_display = pd.DataFrame(list(cursor))
420
 
421
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -424,11 +424,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
424
  elif slate_var == 'Secondary':
425
  collection = db['FD_NFL_Secondary_SD_seed_frame']
426
  if prio_var == None:
427
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
428
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
429
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
430
  else:
431
- cursor = collection.find().sort(prio_var, -1).limit(10000)
432
  raw_display = pd.DataFrame(list(cursor))
433
 
434
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
@@ -437,11 +437,11 @@ def init_FD_lineups(type_var, slate_var, prio_var, prio_mix):
437
  elif slate_var == 'Auxiliary':
438
  collection = db['FD_NFL_Auxiliary_SD_seed_frame']
439
  if prio_var == None:
440
- cursor1 = collection.find().limit(math.ceil(10000 * (prio_mix / 100)))
441
- cursor2 = collection.find().sort('Own', -1).limit(math.ceil(10000 * ((100 - prio_mix) / 100)))
442
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
443
  else:
444
- cursor = collection.find().sort(prio_var, -1).limit(10000)
445
  raw_display = pd.DataFrame(list(cursor))
446
 
447
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
242
 
243
  collection = db['DK_NFL_seed_frame']
244
  if prio_var == None:
245
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
246
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
247
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
248
  else:
249
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
250
  raw_display = pd.DataFrame(list(cursor))
251
 
252
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
262
 
263
  collection = db['DK_NFL_Secondary_seed_frame']
264
  if prio_var == None:
265
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
266
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
267
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
268
  else:
269
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
270
  raw_display = pd.DataFrame(list(cursor))
271
 
272
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
282
 
283
  collection = db['DK_NFL_Late_seed_frame']
284
  if prio_var == None:
285
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
286
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
287
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
288
  else:
289
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
290
  raw_display = pd.DataFrame(list(cursor))
291
 
292
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
297
  if slate_var == 'Main':
298
  collection = db['DK_NFL_SD_seed_frame']
299
  if prio_var == None:
300
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
301
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
302
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
303
  else:
304
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
305
  raw_display = pd.DataFrame(list(cursor))
306
 
307
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
310
  elif slate_var == 'Secondary':
311
  collection = db['DK_NFL_Secondary_SD_seed_frame']
312
  if prio_var == None:
313
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
314
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
315
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
316
  else:
317
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
318
  raw_display = pd.DataFrame(list(cursor))
319
 
320
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
323
  elif slate_var == 'Auxiliary':
324
  collection = db['DK_NFL_Auxiliary_SD_seed_frame']
325
  if prio_var == None:
326
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
327
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
328
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
329
  else:
330
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
331
  raw_display = pd.DataFrame(list(cursor))
332
 
333
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
355
 
356
  collection = db['FD_NFL_seed_frame']
357
  if prio_var == None:
358
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
359
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
360
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
361
  else:
362
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
363
  raw_display = pd.DataFrame(list(cursor))
364
 
365
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
375
 
376
  collection = db['FD_NFL_Secondary_seed_frame']
377
  if prio_var == None:
378
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
379
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
380
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
381
  else:
382
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
383
  raw_display = pd.DataFrame(list(cursor))
384
 
385
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
395
 
396
  collection = db['FD_NFL_Late_seed_frame']
397
  if prio_var == None:
398
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
399
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
400
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
401
  else:
402
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
403
  raw_display = pd.DataFrame(list(cursor))
404
 
405
  raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
411
  if slate_var == 'Main':
412
  collection = db['FD_NFL_SD_seed_frame']
413
  if prio_var == None:
414
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
415
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
416
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
417
  else:
418
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
419
  raw_display = pd.DataFrame(list(cursor))
420
 
421
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
424
  elif slate_var == 'Secondary':
425
  collection = db['FD_NFL_Secondary_SD_seed_frame']
426
  if prio_var == None:
427
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
428
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
429
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
430
  else:
431
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
432
  raw_display = pd.DataFrame(list(cursor))
433
 
434
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]
 
437
  elif slate_var == 'Auxiliary':
438
  collection = db['FD_NFL_Auxiliary_SD_seed_frame']
439
  if prio_var == None:
440
+ cursor1 = collection.find().limit(math.ceil(50000 * (prio_mix / 100)))
441
+ cursor2 = collection.find().sort('Own', -1).limit(math.ceil(50000 * ((100 - prio_mix) / 100)))
442
  raw_display = pd.concat([pd.DataFrame(list(cursor1)), pd.DataFrame(list(cursor2))])
443
  else:
444
+ cursor = collection.find().sort(prio_var, -1).limit(50000)
445
  raw_display = pd.DataFrame(list(cursor))
446
 
447
  raw_display = raw_display[['CPT', 'FLEX1', 'FLEX2', 'FLEX3', 'FLEX4', 'FLEX5', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]