forked from Cryz/battle_royale_sim
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from assets.events import EVENTS
|
|
import random as _random
|
|
|
|
class ArenaEventPicker():
|
|
def __init__(self, players):
|
|
self.players = players
|
|
self.event_list = EVENTS
|
|
self.already_picked_players = []
|
|
|
|
def pick_a_player():
|
|
the_player = _random.choice(self.players)
|
|
the_player_id = the_player.get('id')
|
|
|
|
self.players = list(filter(lambda x : x.get('id') != the_player_id)
|
|
|
|
def pick_event(the_player):
|
|
|
|
player_inventory = the_player.get('inventory') or []
|
|
status = the_player['health']
|
|
reputation = the_player['reputation']
|
|
|
|
elegible_events = []
|
|
for event in EVENTS:
|
|
requirements = event['requirements']
|
|
|
|
keys_to_check = ['item', 'weapon', 'status', 'reputation', 'number_of_players']
|
|
|
|
for check in keys_to_check:
|
|
if requirements.get(check) and check == 'item':
|
|
needed_items = requirements.get('check')
|
|
if needed_items in player_inventory:
|
|
elegible_events.append(event)
|
|
|
|
if requirements.get(check) and check == 'weapon':
|
|
needed_weapons = requirements.get('check')
|
|
if needed_items in player_inventory:
|
|
elegible_events.append(event)
|
|
|
|
if requirements.get(check) and check == 'status':
|
|
needed_health = requirements.get('check')
|
|
if '>' in requirements.get(check) and requirements.get(check) > in needed_health:
|
|
elegible_events.append(event)
|
|
if '<' in requirements.get(check) and requirements.get(check) < in needed_health:
|
|
elegible_events.append(event)
|
|
|
|
if requirements.get(check) and check == 'reputation':
|
|
needed_reputation = requirements.get('reputation')
|
|
if '>' in requirements.get(check) and requirements.get(check) > in needed_reputation:
|
|
elegible_events.append(event)
|
|
if '<' in requirements.get(check) and requirements.get(check) < in needed_reputation:
|
|
elegible_events.append(event)
|
|
|
|
the_event = _random.choice(elegible_events)
|
|
return the_event
|