From 95d5ec5f0e5642f21269e43fe092192578f6253c Mon Sep 17 00:00:00 2001 From: Will Date: Mon, 22 Apr 2019 18:33:52 -0400 Subject: [PATCH] [Setup] Fix the mongo name to create instances using new driver (#2594) * oops * damn you --- redbot/setup.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/redbot/setup.py b/redbot/setup.py index 05d249867..be6b75d8a 100644 --- a/redbot/setup.py +++ b/redbot/setup.py @@ -161,10 +161,11 @@ def basic_setup(): storage = get_storage_type() - storage_dict = {1: "JSON", 2: "MongoDB"} - default_dirs["STORAGE_TYPE"] = storage_dict.get(storage, 1) + storage_dict = {1: BackendType.JSON, 2: BackendType.MONGO} + storage_type: BackendType = storage_dict.get(storage, BackendType.JSON) + default_dirs["STORAGE_TYPE"] = storage_type.value - if storage_dict.get(storage, 1) == "MongoDB": + if storage_type == BackendType.MONGO: from redbot.core.drivers.red_mongo import get_config_details default_dirs["STORAGE_DETAILS"] = get_config_details()