Update main.py

Stop saving paperdex entries for AI and Gauntlet teams
This commit is contained in:
Cal Corum 2023-05-02 13:56:46 -05:00
parent 90098eb60c
commit 5f150ebe5a

23
main.py
View File

@ -2299,6 +2299,11 @@ async def v1_cards_post(cards: CardModel, token: str = Depends(oauth2_scheme)):
new_cards = []
player_ids = []
inc_dex = True
this_team = Team.get_by_id(cards.cards[0].team_id)
if this_team.is_ai or 'Gauntlet' in this_team.abbrev:
inc_dex = False
# new_dex = []
# now = int(datetime.timestamp(datetime.now()) * 1000)
for x in cards.cards:
@ -2308,7 +2313,8 @@ async def v1_cards_post(cards: CardModel, token: str = Depends(oauth2_scheme)):
pack_id=x.pack_id,
value=x.value
)
Paperdex.get_or_create(team_id=x.team_id, player_id=x.player_id)
if inc_dex:
Paperdex.get_or_create(team_id=x.team_id, player_id=x.player_id)
player_ids.append(x.player_id)
new_cards.append(this_card)
@ -4411,6 +4417,21 @@ async def v1_paperdex_delete(paperdex_id, token: str = Depends(oauth2_scheme)):
raise HTTPException(status_code=500, detail=f'Paperdex {this_dex} was not deleted')
@app.post('/api/v1/paperdex/wipe-ai')
async def v1_paperdex_wipeai(token: str = Depends(oauth2_scheme)):
if not valid_token(token):
logging.warning(f'Bad Token: {token}')
db.close()
raise HTTPException(
status_code=401,
detail='Unauthorized'
)
g_teams = Team.select().where(Team.abbrev.contains('Gauntlet'))
count = Paperdex.delete().where(Paperdex.team << g_teams).execute()
return f'Deleted {count} records'
"""
GAMEREWARDS ENDPOINTS
"""