Ensure all maps get local record and karma information linked in memo…

…ry. (#1174)
PyPlanet · May 22, 2022 · e687b37 · e687b37
1 parent e150fac
commit e687b37
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/pyplanet/apps/contrib/karma/__init__.py b/pyplanet/apps/contrib/karma/__init__.py
@@ -86,9 +86,11 @@ async def load_map_votes(self, map=None):
 			)
 
 			# Group by map.
+			# Make sure all maps have an entry in the dictionary.
+			for list_map_id in maps:
+				map_karmas[list_map_id] = list()
+
 			for row in rows:
-				if row.map_id not in map_karmas:
-					map_karmas[row.map_id] = list()
 				map_karmas[row.map_id].append(row)
 
 			# Map karma stats.

diff --git a/pyplanet/apps/contrib/local_records/__init__.py b/pyplanet/apps/contrib/local_records/__init__.py
@@ -83,9 +83,11 @@ async def load_map_locals(self, map=None):
 			)
 
 			# Group by map.
+			# Make sure all maps have an entry in the dictionary.
+			for list_map_id in maps:
+				map_locals[list_map_id] = list()
+
 			for row in rows:
-				if row.map_id not in map_locals:
-					map_locals[row.map_id] = list()
 				map_locals[row.map_id].append(row)
 
 			# Map local stats.