From f713fc238cafa2ba00f5dcb51df28d35a6449d9d Mon Sep 17 00:00:00 2001
From: compiling <8335770+compiling@users.noreply.github.com>
Date: Sat, 11 Jan 2020 20:00:58 +1100
Subject: [PATCH] Add doors to mystery.

---
 Mystery.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/Mystery.py b/Mystery.py
index 0c5c848c..022802d8 100644
--- a/Mystery.py
+++ b/Mystery.py
@@ -4,8 +4,8 @@ import random
 import urllib.request
 import urllib.parse
 
-from EntranceRandomizer import parse_arguments
-from Main import main as ERmain
+from DungeonRandomizer import parse_arguments
+from Main import main as DRMain
 
 def parse_yaml(txt):
     def strip(s):
@@ -97,7 +97,7 @@ def main():
     loglevel = {'error': logging.ERROR, 'info': logging.INFO, 'warning': logging.WARNING, 'debug': logging.DEBUG}[erargs.loglevel]
     logging.basicConfig(format='%(message)s', level=loglevel)
 
-    ERmain(erargs, seed)
+    DRMain(erargs, seed)
 
 def get_weights(path):
     try:
@@ -143,6 +143,8 @@ def roll_settings(weights):
 
     entrance_shuffle = get_choice('entrance_shuffle')
     ret.shuffle = entrance_shuffle if entrance_shuffle != 'none' else 'vanilla'
+    door_shuffle = get_choice('door_shuffle')
+    ret.door_shuffle = door_shuffle if door_shuffle != 'none' else 'vanilla'
 
     ret.goal = {'ganon': 'ganon',
                 'fast_ganon': 'crystals',