From b3444201fac9628d7b50656f79dedd4cbcd0790f Mon Sep 17 00:00:00 2001
From: happyleavesaoc <happyleaves.tfr@gmail.com>
Date: Sun, 2 Oct 2022 15:42:48 +0000
Subject: [PATCH] use resolved map id

---
 mgz/common/map.py   | 5 ++++-
 mgz/summary/full.py | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/mgz/common/map.py b/mgz/common/map.py
index d76a140..31ca9bf 100644
--- a/mgz/common/map.py
+++ b/mgz/common/map.py
@@ -105,8 +105,9 @@ def lookup_name(map_id, name, version, reference):
     is_de = version == Version.DE
     is_hd = version == Version.HD
     # 59: RM custom
+    # 137: custom map pool
     # 138: DM custom
-    if (map_id != 44 and not (is_de or is_hd)) or (map_id not in [59, 138] and (is_de or is_hd)):
+    if (map_id != 44 and not (is_de or is_hd)) or (map_id not in [59, 137, 138] and (is_de or is_hd)):
         map_keys = [int(k) for k in reference['maps'].keys()]
         if map_id in map_keys:
             name = reference['maps'][str(map_id)]
@@ -174,6 +175,8 @@ def get_water_percent(tiles, dataset_id):
 def get_mod_id(strings):
     """Extract RMS mod id."""
     for obj in strings:
+        if not obj.string:
+            continue
         s = obj.string.value.decode('utf-8').split(':')
         if s[0] == 'SUBSCRIBEDMODS' and s[1] == 'RANDOM_MAPS':
             return s[3].split('_')[0]
diff --git a/mgz/summary/full.py b/mgz/summary/full.py
index f2869e3..9a0924e 100644
--- a/mgz/summary/full.py
+++ b/mgz/summary/full.py
@@ -316,7 +316,7 @@ def get_map_id(self):
         if self._header.hd:
             return self._header.hd.selected_map_id
         if self._header.de:
-            return self._header.de.selected_map_id
+            return self._header.de.resolved_map_id
         return self._header.scenario.game_settings.map_id
 
     def get_map(self):