def add_player_to_queue(self, player, role, channel_id):
if role not in player.ratings:
logging.info("Creating a new PlayerRating for <{}> <{}>".format(
player.discord_string, role))
new_rating = PlayerRating(player, role)
self.bot.players_session.add(new_rating)
self.bot.players_session.commit()
# This step is required so our player object has access to the rating
player = self.bot.players_session.merge(player)
# Actually adding the player to the queue
# TODO A player can reset his ready_check status that way since it’s not blocking, see if it makes sense
session = get_session()
# noinspection PyTypeChecker
queue_player = QueuePlayer(channel_id=channel_id,
player_id=player.discord_id,
role=role,
ready_check=None)
session.merge(queue_player)
session.commit()
session.close()
logging.info(
f"Player <{player.discord_string}> has been added to the <{role}><{channel_id}> queue"
)
def add_player_to_queue(self, player, role, channel_id):
if role not in player.ratings:
logging.info('Creating a new PlayerRating for <{}> <{}>'.format(player.discord_string, role))
new_rating = PlayerRating(player, role)
self.bot.players_session.add(new_rating)
self.bot.players_session.commit()
# This step is required so our player object has access to the rating
player = self.bot.players_session.merge(player)
# Actually adding the player to the queue
self.channel_queues[channel_id][role].add(player)
logging.info(f'Player <{player.discord_string}> has been added to the <{role}><{channel_id}> queue')
async def test_player(caplog):
"""
Tests the addition of the Bot user as a player itself.
Also tests PlayerRating and makes sure the on_cascade flags are applied properly.
"""
caplog.set_level(logging.INFO)
from inhouse_bot.common_utils import discord_token
from inhouse_bot.sqlite.player_rating import PlayerRating
from inhouse_bot.sqlite.sqlite_utils import get_session
from inhouse_bot.sqlite.player import Player
client = discord.Client()
await client.login(discord_token)
test_user = await client.fetch_user(707853630681907253)
await client.close()
session = get_session()
player = Player(test_user)
player = session.merge(player)
session.commit()
assert player == session.query(Player).filter(
Player.discord_id == test_user.id).one()
player_rating = PlayerRating(player, 'mid')
session.add(player_rating)
session.commit()
assert player.ratings
session.delete(player)
session.commit()
assert session.query(Player).filter(
Player.discord_id == test_user.id).one_or_none() is None
assert session.query(PlayerRating).filter(
PlayerRating.player_id == test_user.id).one_or_none() is None
async def test_game(caplog):
"""
Tests the addition of a game.
"""
# TODO Make this test cleaner
from inhouse_bot.inhouse_bot import InhouseBot
from inhouse_bot.sqlite.sqlite_utils import get_session
from inhouse_bot.sqlite.sqlite_utils import roles_list
from inhouse_bot.sqlite.game import Game
from inhouse_bot.sqlite.player import Player
from inhouse_bot.sqlite.player_rating import PlayerRating
session = get_session()
# We create our bot object, a mock server, a mock channel, 10 mock members, and return our cog for testing
bot = InhouseBot()
dpytest.configure(bot, 1, 1, 10)
players = {}
for member_id in range(0, 10):
role = roles_list[member_id % 5]
player = Player(dpytest.get_config().members[member_id])
rating = PlayerRating(player, role)
session.add(player)
session.add(rating)
session.commit()
players["blue" if member_id % 2 else "red", role] = player
game = Game(players)
session.add(game)
# Printing ahead of time
try:
print(game)
except AttributeError:
assert True
session.commit()
print(game)