From 4d9e2c3226a10729f3401f4e75b9b10aa90a2610 Mon Sep 17 00:00:00 2001 From: Aaron Manaloto Date: Fri, 8 Sep 2023 14:46:19 +0800 Subject: [PATCH 1/2] Return false if DB hasn't been loaded yet --- lib/dradis/plugins/settings/adapters/db.rb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/dradis/plugins/settings/adapters/db.rb b/lib/dradis/plugins/settings/adapters/db.rb index 2430cee..5864374 100644 --- a/lib/dradis/plugins/settings/adapters/db.rb +++ b/lib/dradis/plugins/settings/adapters/db.rb @@ -9,7 +9,8 @@ def delete(key) end def exists?(key) - Configuration.exists?(name: namespaced_key(key)) + (ActiveRecord::Base.connection rescue false) && + Configuration.exists?(name: namespaced_key(key)) end def read(key) From 7049f0aae88f1255be16e07a75fa91a36cdbc4bb Mon Sep 17 00:00:00 2001 From: Aaron Manaloto Date: Fri, 8 Sep 2023 14:54:09 +0800 Subject: [PATCH 2/2] Add table_exists? check --- lib/dradis/plugins/settings/adapters/db.rb | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/dradis/plugins/settings/adapters/db.rb b/lib/dradis/plugins/settings/adapters/db.rb index 5864374..e04e0ed 100644 --- a/lib/dradis/plugins/settings/adapters/db.rb +++ b/lib/dradis/plugins/settings/adapters/db.rb @@ -10,6 +10,7 @@ def delete(key) def exists?(key) (ActiveRecord::Base.connection rescue false) && + Configuration.table_exists? && Configuration.exists?(name: namespaced_key(key)) end