From a06ce53fc1c4603d33a89a1334aa5687dc0e7e6f Mon Sep 17 00:00:00 2001
From: Marcin Kostrzewski <komar86311@gmail.com>
Date: Sun, 5 Apr 2020 23:42:41 +0200
Subject: [PATCH] Implemented getEntityOnCoord method

---
 src/game/Map.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/game/Map.py b/src/game/Map.py
index a8408d7..e70e439 100644
--- a/src/game/Map.py
+++ b/src/game/Map.py
@@ -1,8 +1,8 @@
 import json
 
 import pygame
-from game.TerrainTile import TerrainTile
-from game.Screen import Locations
+from src.game.TerrainTile import TerrainTile
+from src.game.Screen import Locations
 
 from src.entities.Entity import Entity
 from src.entities.Pickupable import Pickupable
@@ -75,6 +75,13 @@ class Map:
                     self.screen.draw(object, Locations.MAP, 0, 0)
                     self.collidables.add(object)
 
+    def getEntityOnCoord(self, coord):
+        result = None
+        for entity in self.collidables:
+            if entity.rect.x == coord[0] and entity.rect.y == coord[1]
+                result = entity
+        return result
+
     def addEntity(self, entity):
         self.screen.draw(entity, Locations.MAP, 0, 0)
         self.collidables.add(entity)