config_db search methods

This commit is contained in:
Alexander Strizhakov 2020-12-03 19:34:23 +03:00
parent 7fd4f4908b
commit a02eb88396
No known key found for this signature in database
GPG Key ID: 022896A53AEF1381
2 changed files with 24 additions and 18 deletions

View File

@ -62,9 +62,8 @@ defmodule Mix.Tasks.Pleroma.Config do
group = maybe_atomize(group)
key = maybe_atomize(key)
%{group: group, key: key}
|> ConfigDB.get_by_params()
|> Repo.all()
group
|> ConfigDB.get_all_by_group_and_key(key)
|> Enum.each(&dump/1)
end)
end
@ -290,17 +289,15 @@ defmodule Mix.Tasks.Pleroma.Config do
end
defp dump_group(group) when is_atom(group) do
%{group: group}
|> ConfigDB.get_by_params()
|> Repo.all()
group
|> ConfigDB.get_all_by_group()
|> Enum.each(&dump/1)
end
defp group_exists?(group) do
%{group: group}
|> ConfigDB.get_by_params()
|> Repo.all()
|> Enum.empty?()
group
|> ConfigDB.get_all_by_group()
|> Enum.empty?()
end
defp maybe_atomize(arg) when is_atom(arg), do: arg
@ -326,8 +323,8 @@ defmodule Mix.Tasks.Pleroma.Config do
defp delete_key(group, key) do
check_configdb(fn ->
ConfigDB.get_by_params(%{group: group, key: key})
|> Repo.all()
group
|> ConfigDB.get_all_by_group_and_key(key)
|> Enum.each(&delete(&1, true))
end)
end
@ -338,10 +335,9 @@ defmodule Mix.Tasks.Pleroma.Config do
shell_info("The following settings will be removed from ConfigDB:\n")
dump_group(group)
ConfigDB.get_by_params(%{group: group})
|> Repo.all()
|> Enum.each(&delete(&1, true))
group
|> ConfigDB.get_all_by_group()
|> Enum.each(&delete(&1, true))
else
_ -> shell_error("No settings in ConfigDB for #{inspect(group)}. Aborting.")
end

View File

@ -6,7 +6,7 @@ defmodule Pleroma.ConfigDB do
use Ecto.Schema
import Ecto.Changeset
import Ecto.Query, only: [select: 3]
import Ecto.Query, only: [select: 3, from: 2]
import Pleroma.Web.Gettext
alias __MODULE__
@ -41,6 +41,16 @@ defmodule Pleroma.ConfigDB do
end)
end
@spec get_all_by_group(atom() | String.t()) :: [t()]
def get_all_by_group(group) do
from(c in ConfigDB, where: c.group == ^group) |> Repo.all()
end
@spec get_all_by_group_and_key(atom() | String.t(), atom() | String.t()) :: [t()]
def get_all_by_group_and_key(group, key) do
from(c in ConfigDB, where: c.group == ^group and c.key == ^key) |> Repo.all()
end
@spec get_by_params(map()) :: ConfigDB.t() | nil
def get_by_params(params), do: Repo.get_by(ConfigDB, params)