class TablePanelTest(unittest.TestCase):
def setUp(self):
self._app = wx.PySimpleApp()
self.panel = TablePanel(parent=None)
def test_add_card_creates_card_and_adds_it(self):
card = DurakCard('6H')
self.assertFalse(card in self.panel._cards)
with patch('durak.gui.widgets.Card') as Card_mock:
self.panel._add_card(card, (1, 2))
Card_mock.assert_called_once_with(
parent=self.panel, card=card, pos=(1, 2)
)
self.assertTrue(card in self.panel._cards)
def test_move(self):
card = DurakCard('6H')
with patch.object(self.panel, '_add_card') as add_card_mock:
self.panel.move(card)
add_card_mock.assert_called_once_with(
card,
(0 + self.panel.PADDING_LEFT, self.panel.UPPER_CARD_Y_OFFSET)
)
def test_respond(self):
card_mock = Mock()
card_mock.GetPosition.return_value = wx.Point(10, 10)
self.panel._cards[DurakCard('6H')] = card_mock
card = DurakCard('7H')
with patch.object(self.panel, '_add_card') as add_card_mock:
self.panel.respond(card)
add_card_mock.assert_called_once_with(
card,
wx.Point(10, 10) + wx.Point(
self.panel.UPPER_CARD_X_OFFSET,
- self.panel.UPPER_CARD_Y_OFFSET
)
)
def test_give_more(self):
card_mock = Mock()
card_mock.GetPosition.return_value = (10, 10)
self.panel._cards[DurakCard('6H')] = card_mock
card = DurakCard('7H')
self.assertEqual(self.panel._given_more, [])
with patch.object(self.panel, '_add_card') as add_card_mock:
self.panel.give_more(card)
add_card_mock.assert_called_once_with(
card,
(
10 + self.panel.GIVEN_MORE_OFFSET,
self.panel.UPPER_CARD_Y_OFFSET
)
)
self.assertEqual(self.panel._given_more, [DurakCard('7H')])
def test_remove_all_destroys_all_the_cards(self):
cards = [DurakCard('6H'), DurakCard('7H'), DurakCard('8H')]
card_values = [Mock(), Mock(), Mock()]
self.panel._cards = dict(zip(cards, card_values))
self.panel._give_more_mode = True
self.panel.remove_all()
self.assertTrue(all(card.Destroy.called for card in card_values))
self.assertEqual(self.panel._cards, {})
self.assertEqual(self.panel._given_more, [])
def test_cards_property(self):
cards = [DurakCard('6H'), DurakCard('7H'), DurakCard('8H')]
card_values = [Mock(), Mock(), Mock()]
self.panel._cards = OrderedDict(zip(cards, card_values))
self.assertEqual(self.panel.cards, cards)
def test_given_more_property(self):
cards = [DurakCard('6H'), DurakCard('7H')]
self.panel._given_more = cards
self.assertEqual(self.panel.given_more, cards)
def test_is_odd_property(self):
self.panel._cards[DurakCard('6H')] = Mock()
self.assertTrue(self.panel._is_odd)
self.panel._cards[DurakCard('7H')] = Mock()
self.assertFalse(self.panel._is_odd)
def test_pop_if_no_given_more(self):
card_mock = Mock()
self.panel._cards[DurakCard('6H')] = card_mock
self.assertEqual(self.panel.pop(), DurakCard('6H'))
self.assertDictEqual(self.panel._cards, {})
card_mock.Destroy.assert_called()
def test_pop_with_given_more(self):
card_mock = Mock()
self.panel._cards[DurakCard('6H')] = Mock()
self.panel._cards[DurakCard('7H')] = card_mock
self.panel._given_more = [DurakCard('7H')]
self.assertEqual(self.panel.pop(), DurakCard('7H'))
self.assertItemsEqual(self.panel._cards.keys(), [DurakCard('6H')])
card_mock.Destroy.assert_called()
class ViewLogFrame(wx.Frame):
WIDTH = 800
HEIGHT = 500
LAST_DIR_SETTING = 'last_log_dir'
def __init__(self):
super(ViewLogFrame, self).__init__(parent=None,
title='Durak log viewer',
size=(self.WIDTH, self.HEIGHT))
self._log_viever = None
self._create_layout()
def _create_layout(self):
self.Bind(wx.EVT_CLOSE, self._on_close)
self._panel = wx.Panel(parent=self)
self._top_player_sizer = LabeledCardSizer(
wx.HORIZONTAL,
parent=self._panel,
)
self._bottom_player_sizer = LabeledCardSizer(
wx.HORIZONTAL,
parent=self._panel,
)
self._table_sizer = wx.FlexGridSizer(cols=2, rows=1)
self._table_sizer.AddGrowableCol(0)
self._table = TablePanel(parent=self._panel)
self._table_sizer.Add(self._table, proportion=1)
self._deck = DeckPanel(self._panel, size=(100, 130))
self._table_sizer.Add(self._deck)
self._control_sizer = wx.BoxSizer(wx.HORIZONTAL)
self._to_game_start_button = wx.Button(parent=self._panel, label=u'<<')
self._to_game_start_button.Bind(wx.EVT_BUTTON,
self._menu_on_to_game_start)
self._prev_button = wx.Button(parent=self._panel, label=u'<')
self._prev_button.Bind(wx.EVT_BUTTON, self._menu_on_prev)
self._next_button = wx.Button(parent=self._panel, label=u'>')
self._next_button.Bind(wx.EVT_BUTTON, self._menu_on_next)
self._to_game_end_button = wx.Button(parent=self._panel, label=u'>>')
self._to_game_end_button.Bind(wx.EVT_BUTTON, self._menu_on_to_game_end)
self._control_sizer.AddMany([
self._to_game_start_button, self._prev_button, self._next_button,
self._to_game_end_button
])
self._main_sizer = wx.FlexGridSizer(cols=1, rows=4)
self._main_sizer.AddGrowableCol(0)
self._main_sizer.AddGrowableRow(1)
self._main_sizer.Add(self._top_player_sizer,
flag=wx.EXPAND,
proportion=1)
self._main_sizer.Add(self._table_sizer,
flag=wx.ALIGN_CENTER_VERTICAL,
proportion=1)
self._main_sizer.Add(self._control_sizer, flag=wx.EXPAND, proportion=1)
self._main_sizer.Add(self._bottom_player_sizer,
flag=wx.EXPAND,
proportion=1)
self._panel.SetSizer(self._main_sizer)
self.CreateStatusBar()
self._create_menu()
self._panel.Layout()
self.Refresh()
self.Layout()
self.PLAYER_SIZER_MAP = {
LogViewer.PLAYER1: self._bottom_player_sizer,
LogViewer.PLAYER2: self._top_player_sizer,
}
self._menu_on_open()
def _create_menu(self):
self.menu = wx.MenuBar()
menu = wx.Menu() # Файл
item = menu.Append(wx.ID_OPEN, u'Открыть файл лога...',
u'Открыть файл лога')
self.Bind(wx.EVT_MENU, self._menu_on_open, item)
item = menu.Append(wx.ID_ANY, u'Список игр...\tCTRL+L',
u'Просмотреть список игр в открытом файле')
self.Bind(wx.EVT_MENU, self._menu_on_select_game, item)
item = menu.Append(wx.ID_EXIT, u'Выход', u'Закрыть программу')
self.Bind(wx.EVT_MENU, self._on_close, item)
self.menu.Append(menu, u'Файл')
menu = wx.Menu() # Текущая игра
self._to_game_start_menu_item = menu.Append(wx.ID_ANY,
u'В начало <<\tHOME',
u'Перейти в начало игры')
self.Bind(wx.EVT_MENU, self._menu_on_to_game_start,
self._to_game_start_menu_item)
self._prev_menu_item = menu.Append(wx.ID_ANY, u'Назад <\tCTRL+LEFT',
u'Вернуться на ход назад')
self.Bind(wx.EVT_MENU, self._menu_on_prev, self._prev_menu_item)
self._next_menu_item = menu.Append(wx.ID_ANY, u'Вперед >\tCTRL+RIGHT',
u'Перейти на ход вперед')
self.Bind(wx.EVT_MENU, self._menu_on_next, self._next_menu_item)
self._to_game_end_menu_item = menu.Append(wx.ID_ANY,
u'В конец >>\tEND',
u'Перейти в конец игры')
self.Bind(wx.EVT_MENU, self._menu_on_to_game_end,
self._to_game_end_menu_item)
self.menu.Append(menu, u'Текущая игра')
self.SetMenuBar(self.menu)
def _on_close(self, event):
if event.GetEventType() in wx.EVT_CLOSE.evtType:
event.Skip()
else:
self.Close()
def _menu_on_open(self, event=None):
last_directory = get_setting(self.LAST_DIR_SETTING,
os.path.expanduser('~'))
dialog = wx.FileDialog(self, u'Выберите файл лога', last_directory, '',
'*', wx.OPEN)
if dialog.ShowModal() != wx.ID_OK:
self.Close()
filename = dialog.GetPath()
dialog.Destroy()
self._log_viever = LogViewer(filename)
set_setting(self.LAST_DIR_SETTING, os.path.dirname(filename))
self._menu_on_select_game()
def _menu_on_select_game(self, event=None):
choices = [
'%d. %s' % (i, game['title'])
for i, game in enumerate(self._log_viever.iterindex(), start=1)
]
dialog = wx.SingleChoiceDialog(
self,
u'Выберите игру',
u'Список игр',
choices,
wx.CHOICEDLG_STYLE,
)
dialog.SetSize((500, 400))
dialog.Refresh()
if dialog.ShowModal() == wx.ID_OK:
self._log_viever.load_game(dialog.GetSelection())
self._bottom_player_sizer.set_label(self._log_viever.player1_name)
self._top_player_sizer.set_label(self._log_viever.player2_name)
self._menu_on_to_game_start()
dialog.Destroy()
def _menu_on_to_game_start(self, event=None):
self._log_viever.to_game_start()
if not self._log_viever.has_next:
return
self._deck.set_opened_trump(self._log_viever.opened_trump)
for card_sizer in (self._bottom_player_sizer, self._top_player_sizer):
card_sizer.trump = self._log_viever.opened_trump
card_sizer.remove_all()
log_event = self._log_viever.get_next()
self._new_move(log_event)
self._panel.Layout()
self.Layout()
self.Refresh()
self._set_enabled_controls()
def _menu_on_to_game_end(self, event=None):
self._log_viever.to_game_end()
if not self._log_viever.has_prev:
return
log_event = self._log_viever.get_prev()
self._new_move(log_event, with_table=True)
self._set_enabled_controls()
def _menu_on_next(self, event):
if not self._log_viever.has_next:
return
log_event = self._log_viever.get_next()
if log_event['event_type'] == self._log_viever.NEW_MOVE:
self._new_move(log_event)
return
card_sizer = self.PLAYER_SIZER_MAP[log_event['to_move']]
table_method = getattr(self._table, log_event['event_type'])
card = log_event['card']
table_method(card)
card_sizer.remove_card(card, do_layout=True)
self._set_enabled_controls()
def _menu_on_prev(self, event):
if not self._log_viever.has_prev:
return
log_event = self._log_viever.get_prev()
if log_event['event_type'] == self._log_viever.NEW_MOVE:
self._new_move(log_event, with_table=True)
return
card_sizer = self.PLAYER_SIZER_MAP[log_event['to_move']]
card = log_event['card']
self._table.pop()
card_sizer.add_card(card, do_layout=True)
self._set_enabled_controls()
def _new_move(self, log_event, with_table=False):
assert log_event['event_type'] == self._log_viever.NEW_MOVE
self._table.remove_all()
self._deck.set_card_count(log_event['deck_count'])
if with_table:
for index, card in enumerate(log_event['moves_and_responds']):
if index % 2:
self._table.respond(card)
else:
self._table.move(card)
for card in log_event['given_more']:
self._table.give_more(card)
card_sizers = (self._bottom_player_sizer, self._top_player_sizer)
card_sets = (log_event['player1_cards'], log_event['player2_cards'])
for card_sizer, card_set in zip(card_sizers, card_sets):
if with_table:
card_sizer.remove_all()
cards_to_add = card_set - set(log_event['moves_and_responds'] +
log_event['given_more'])
else:
cards_to_add = card_set - card_sizer.cards
for card in cards_to_add:
card_sizer.add_card(card)
card_sizer.Layout()
if log_event['to_move'] == self._log_viever.PLAYER1:
self.SetStatusText(u'Атакует игрок снизу')
else:
self.SetStatusText(u'Атакует игрок сверху')
self._set_enabled_controls()
def _set_enabled_controls(self):
all_controls = {
self._to_game_start_button,
self._to_game_end_button,
self._next_button,
self._prev_button,
self._to_game_end_menu_item,
self._to_game_start_menu_item,
self._prev_menu_item,
self._next_menu_item,
}
[control.Enable(False) for control in all_controls]
if not self._log_viever or not self._log_viever.has_game_loaded:
return
if self._log_viever.has_prev:
self._to_game_start_button.Enable()
self._prev_button.Enable()
self._to_game_start_menu_item.Enable()
self._prev_menu_item.Enable()
if self._log_viever.has_next:
self._to_game_end_button.Enable()
self._next_button.Enable()
self._to_game_end_menu_item.Enable()
self._next_menu_item.Enable()
class ViewLogFrame(wx.Frame):
WIDTH = 800
HEIGHT = 500
LAST_DIR_SETTING = 'last_log_dir'
def __init__(self):
super(ViewLogFrame, self).__init__(
parent=None,
title='Durak log viewer',
size=(self.WIDTH, self.HEIGHT)
)
self._log_viever = None
self._create_layout()
def _create_layout(self):
self.Bind(wx.EVT_CLOSE, self._on_close)
self._panel = wx.Panel(parent=self)
self._top_player_sizer = LabeledCardSizer(
wx.HORIZONTAL,
parent=self._panel,
)
self._bottom_player_sizer = LabeledCardSizer(
wx.HORIZONTAL,
parent=self._panel,
)
self._table_sizer = wx.FlexGridSizer(cols=2, rows=1)
self._table_sizer.AddGrowableCol(0)
self._table = TablePanel(parent=self._panel)
self._table_sizer.Add(self._table, proportion=1)
self._deck = DeckPanel(self._panel, size=(100, 130))
self._table_sizer.Add(self._deck)
self._control_sizer = wx.BoxSizer(wx.HORIZONTAL)
self._to_game_start_button = wx.Button(parent=self._panel, label=u'<<')
self._to_game_start_button.Bind(
wx.EVT_BUTTON, self._menu_on_to_game_start
)
self._prev_button = wx.Button(parent=self._panel, label=u'<')
self._prev_button.Bind(wx.EVT_BUTTON, self._menu_on_prev)
self._next_button = wx.Button(parent=self._panel, label=u'>')
self._next_button.Bind(wx.EVT_BUTTON, self._menu_on_next)
self._to_game_end_button = wx.Button(parent=self._panel, label=u'>>')
self._to_game_end_button.Bind(wx.EVT_BUTTON, self._menu_on_to_game_end)
self._control_sizer.AddMany([
self._to_game_start_button,
self._prev_button,
self._next_button,
self._to_game_end_button
])
self._main_sizer = wx.FlexGridSizer(cols=1, rows=4)
self._main_sizer.AddGrowableCol(0)
self._main_sizer.AddGrowableRow(1)
self._main_sizer.Add(
self._top_player_sizer, flag=wx.EXPAND, proportion=1
)
self._main_sizer.Add(
self._table_sizer, flag=wx.ALIGN_CENTER_VERTICAL, proportion=1
)
self._main_sizer.Add(self._control_sizer, flag=wx.EXPAND, proportion=1)
self._main_sizer.Add(
self._bottom_player_sizer, flag=wx.EXPAND, proportion=1
)
self._panel.SetSizer(self._main_sizer)
self.CreateStatusBar()
self._create_menu()
self._panel.Layout()
self.Refresh()
self.Layout()
self.PLAYER_SIZER_MAP = {
LogViewer.PLAYER1: self._bottom_player_sizer,
LogViewer.PLAYER2: self._top_player_sizer,
}
self._menu_on_open()
def _create_menu(self):
self.menu = wx.MenuBar()
menu = wx.Menu() # Файл
item = menu.Append(
wx.ID_OPEN, u'Открыть файл лога...', u'Открыть файл лога'
)
self.Bind(wx.EVT_MENU, self._menu_on_open, item)
item = menu.Append(
wx.ID_ANY,
u'Список игр...\tCTRL+L',
u'Просмотреть список игр в открытом файле'
)
self.Bind(wx.EVT_MENU, self._menu_on_select_game, item)
item = menu.Append(wx.ID_EXIT, u'Выход', u'Закрыть программу')
self.Bind(wx.EVT_MENU, self._on_close, item)
self.menu.Append(menu, u'Файл')
menu = wx.Menu() # Текущая игра
self._to_game_start_menu_item = menu.Append(
wx.ID_ANY, u'В начало <<\tHOME', u'Перейти в начало игры'
)
self.Bind(
wx.EVT_MENU,
self._menu_on_to_game_start,
self._to_game_start_menu_item
)
self._prev_menu_item = menu.Append(
wx.ID_ANY, u'Назад <\tCTRL+LEFT', u'Вернуться на ход назад'
)
self.Bind(wx.EVT_MENU, self._menu_on_prev, self._prev_menu_item)
self._next_menu_item = menu.Append(
wx.ID_ANY, u'Вперед >\tCTRL+RIGHT', u'Перейти на ход вперед'
)
self.Bind(wx.EVT_MENU, self._menu_on_next, self._next_menu_item)
self._to_game_end_menu_item = menu.Append(
wx.ID_ANY, u'В конец >>\tEND', u'Перейти в конец игры'
)
self.Bind(
wx.EVT_MENU,
self._menu_on_to_game_end,
self._to_game_end_menu_item
)
self.menu.Append(menu, u'Текущая игра')
self.SetMenuBar(self.menu)
def _on_close(self, event):
if event.GetEventType() in wx.EVT_CLOSE.evtType:
event.Skip()
else:
self.Close()
def _menu_on_open(self, event=None):
last_directory = get_setting(
self.LAST_DIR_SETTING, os.path.expanduser('~')
)
dialog = wx.FileDialog(
self, u'Выберите файл лога', last_directory, '', '*', wx.OPEN
)
if dialog.ShowModal() != wx.ID_OK:
self.Close()
filename = dialog.GetPath()
dialog.Destroy()
self._log_viever = LogViewer(filename)
set_setting(self.LAST_DIR_SETTING, os.path.dirname(filename))
self._menu_on_select_game()
def _menu_on_select_game(self, event=None):
choices = [
'%d. %s' % (i, game['title']) for i, game
in enumerate(self._log_viever.iterindex(), start=1)
]
dialog = wx.SingleChoiceDialog(
self,
u'Выберите игру',
u'Список игр',
choices,
wx.CHOICEDLG_STYLE,
)
dialog.SetSize((500, 400))
dialog.Refresh()
if dialog.ShowModal() == wx.ID_OK:
self._log_viever.load_game(dialog.GetSelection())
self._bottom_player_sizer.set_label(self._log_viever.player1_name)
self._top_player_sizer.set_label(self._log_viever.player2_name)
self._menu_on_to_game_start()
dialog.Destroy()
def _menu_on_to_game_start(self, event=None):
self._log_viever.to_game_start()
if not self._log_viever.has_next:
return
self._deck.set_opened_trump(self._log_viever.opened_trump)
for card_sizer in (self._bottom_player_sizer, self._top_player_sizer):
card_sizer.trump = self._log_viever.opened_trump
card_sizer.remove_all()
log_event = self._log_viever.get_next()
self._new_move(log_event)
self._panel.Layout()
self.Layout()
self.Refresh()
self._set_enabled_controls()
def _menu_on_to_game_end(self, event=None):
self._log_viever.to_game_end()
if not self._log_viever.has_prev:
return
log_event = self._log_viever.get_prev()
self._new_move(log_event, with_table=True)
self._set_enabled_controls()
def _menu_on_next(self, event):
if not self._log_viever.has_next:
return
log_event = self._log_viever.get_next()
if log_event['event_type'] == self._log_viever.NEW_MOVE:
self._new_move(log_event)
return
card_sizer = self.PLAYER_SIZER_MAP[log_event['to_move']]
table_method = getattr(self._table, log_event['event_type'])
card = log_event['card']
table_method(card)
card_sizer.remove_card(card, do_layout=True)
self._set_enabled_controls()
def _menu_on_prev(self, event):
if not self._log_viever.has_prev:
return
log_event = self._log_viever.get_prev()
if log_event['event_type'] == self._log_viever.NEW_MOVE:
self._new_move(log_event, with_table=True)
return
card_sizer = self.PLAYER_SIZER_MAP[log_event['to_move']]
card = log_event['card']
self._table.pop()
card_sizer.add_card(card, do_layout=True)
self._set_enabled_controls()
def _new_move(self, log_event, with_table=False):
assert log_event['event_type'] == self._log_viever.NEW_MOVE
self._table.remove_all()
self._deck.set_card_count(log_event['deck_count'])
if with_table:
for index, card in enumerate(log_event['moves_and_responds']):
if index % 2:
self._table.respond(card)
else:
self._table.move(card)
for card in log_event['given_more']:
self._table.give_more(card)
card_sizers = (self._bottom_player_sizer, self._top_player_sizer)
card_sets = (log_event['player1_cards'], log_event['player2_cards'])
for card_sizer, card_set in zip(card_sizers, card_sets):
if with_table:
card_sizer.remove_all()
cards_to_add = card_set - set(
log_event['moves_and_responds'] + log_event['given_more']
)
else:
cards_to_add = card_set - card_sizer.cards
for card in cards_to_add:
card_sizer.add_card(card)
card_sizer.Layout()
if log_event['to_move'] == self._log_viever.PLAYER1:
self.SetStatusText(u'Атакует игрок снизу')
else:
self.SetStatusText(u'Атакует игрок сверху')
self._set_enabled_controls()
def _set_enabled_controls(self):
all_controls = {
self._to_game_start_button,
self._to_game_end_button,
self._next_button,
self._prev_button,
self._to_game_end_menu_item,
self._to_game_start_menu_item,
self._prev_menu_item,
self._next_menu_item,
}
[control.Enable(False) for control in all_controls]
if not self._log_viever or not self._log_viever.has_game_loaded:
return
if self._log_viever.has_prev:
self._to_game_start_button.Enable()
self._prev_button.Enable()
self._to_game_start_menu_item.Enable()
self._prev_menu_item.Enable()
if self._log_viever.has_next:
self._to_game_end_button.Enable()
self._next_button.Enable()
self._to_game_end_menu_item.Enable()
self._next_menu_item.Enable()
