parent
5c16001824
commit
222bf53b25
@ -0,0 +1,87 @@
|
||||
##########################################################################
|
||||
# Copyright (C) 2022-2023 Kostiantyn Klochko <kostya_klochko@ukr.net> #
|
||||
# #
|
||||
# This file is part of Anitube Simple Notification. #
|
||||
# #
|
||||
# Anitube Simple Notification is free software: you can redistribute #
|
||||
# it and/or modify it under the terms of the GNU General Public #
|
||||
# License as published by the Free Software Foundation, either version #
|
||||
# 3 of the License, or (at your option) any later version. #
|
||||
# #
|
||||
# Anitube Simple Notification is distributed in the hope that it will #
|
||||
# be useful, but WITHOUT ANY WARRANTY; without even the implied #
|
||||
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See #
|
||||
# the GNU General Public License for more details. #
|
||||
# #
|
||||
# You should have received a copy of the GNU General Public License #
|
||||
# along with Anitube Simple Notification. If not, see #
|
||||
# <https://www.gnu.org/licenses/>. #
|
||||
##########################################################################
|
||||
|
||||
"""
|
||||
This module has all for simplify work with the toml configuration file.
|
||||
"""
|
||||
|
||||
from rich.console import Console
|
||||
import tomli
|
||||
import os
|
||||
|
||||
class Config:
|
||||
@staticmethod
|
||||
def config_validation(config: dict) -> bool:
|
||||
"""Return False if config is invalid. Empty values is valid."""
|
||||
validations = {
|
||||
"POSTERS": lambda v: isinstance(v, bool),
|
||||
"WAITING_PERIOD": lambda v: isinstance(v, int) and v >= 0,
|
||||
"URLS": lambda l: isinstance(l, list) and len(l) == len(list(filter(str,l))),
|
||||
}
|
||||
return all(validations.get(k, lambda x: False)(v) for (k, v) in config.items())
|
||||
|
||||
@staticmethod
|
||||
def config_validation_detail(config: dict) -> dict:
|
||||
"""Return a dict where key is a field and value is valid status."""
|
||||
validations = {
|
||||
"POSTERS": lambda v: isinstance(v, bool),
|
||||
"WAITING_PERIOD": lambda v: isinstance(v, int) and v >= 0,
|
||||
"URLS": lambda l: isinstance(l, list) and
|
||||
len(list(filter(lambda v: isinstance(v,str), l))) and
|
||||
len(list(filter(lambda v: v.startswith("https://anitube.in.ua/"), l)))
|
||||
#len(l) == len(list(filter(str,l))),
|
||||
}
|
||||
return {k:validations.get(k, lambda x: False)(v) for (k, v) in config.items()}
|
||||
|
||||
@staticmethod
|
||||
def config_validation_error(config: dict):
|
||||
"""Show error and remove invalid fields."""
|
||||
validation = Config.config_validation_detail(config)
|
||||
if not validation.pop("POSTERS", True):
|
||||
Console().print(f"[red][ERROR] Please, POSTER must be false or true.[/]")
|
||||
del config["POSTERS"]
|
||||
if not validation.pop("WAITING_PERIOD", True):
|
||||
Console().print(
|
||||
''.join(["[red][ERROR] ",
|
||||
"Please, WAITING_PERIOD must be an integer that greater",
|
||||
"than zero or equals to.[/]"]))
|
||||
del config["WAITING_PERIOD"]
|
||||
if not validation.pop("URLS", True):
|
||||
Console().print(
|
||||
"".join(["[red][ERROR] Please, URLS must be a list of strings"
|
||||
"which ones are page urls of anitube.in.ua.[/]"]))
|
||||
del config["URLS"]
|
||||
|
||||
@staticmethod
|
||||
def get_config(config_path:str, console:Console) -> dict:
|
||||
"""
|
||||
Read the configuration file and return dict as configuration.
|
||||
The configuration file must be in the application folder.
|
||||
"""
|
||||
config = {}
|
||||
try:
|
||||
with open(config_path, "rb") as file:
|
||||
config = tomli.load(file)
|
||||
except FileNotFoundError:
|
||||
console.print(f"[red][ERROR] Please, create configuration file.[/]")
|
||||
except tomli.TOMLDecodeError:
|
||||
console.print(f"[red][ERROR] Please, check configuration file for correctness.[/]")
|
||||
Config.config_validation_error(config)
|
||||
return config
|
||||
Loading…
Reference in new issue