Add DB execution context for gym_park. (#2508)

RocketMap · Mar 3, 2018 · ef6fd17 · ef6fd17
1 parent 15be085
commit ef6fd17
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pogom/models.py b/pogom/models.py
@@ -638,10 +638,11 @@ def set_gyms_in_park(gyms, park):
 
     @staticmethod
     def get_gyms_park(id):
-        gym_by_id = Gym.select(Gym.park).where(
-            Gym.gym_id == id).dicts()
-        if gym_by_id:
-            return gym_by_id[0]['park']
+        with Gym.database().execution_context():
+            gym_by_id = Gym.select(Gym.park).where(
+                Gym.gym_id == id).dicts()
+            if gym_by_id:
+                return gym_by_id[0]['park']
         return False