fixes bug in pawn

Author: fearsd

main.py

@@ -1,5 +1,4 @@
-from models.gamefield import GameField
-from models.player import Player, Computer
+from models import Player, Computer, GameField
 
 def game_init():
     chess = GameField()

models/__init__.py

@@ -0,0 +1,22 @@
+from .basefigure import BaseFigure
+from .bishop import Bishop
+from .gamefield import GameField
+from .king import King
+from .knight import Knight
+from .pawn import Pawn
+from .player import Player, Computer
+from .queen import Queen
+from .rook import Rook
+
+__all__ = [
+    'BaseFigure',
+    'Bishop',
+    'GameField',
+    'King',
+    'Knight',
+    'Pawn',
+    'Player',
+    'Queen',
+    'Rook',
+    'Computer'
+]

models/pawn.py

@@ -84,5 +84,15 @@ def _available_moves(self, field):
                         moves.append(str(int(self.coord[0]) - 1) + chr(ord(self.coord[1]) - 1))
                 except KeyError:
                     pass
+
+        i = 0
+        while i <= len(moves) - 1:
+            if isinstance(field.field[moves[i]], field.figures):
+                if field.field[moves[i]].team == self.team:
+                    del moves[i]
+                else:
+                    i += 1
+            else:
+                i += 1
 
         return moves