53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from typing import List, Tuple
|
|
|
|
import logging
|
|
|
|
logging.basicConfig(
|
|
filename="config.log",
|
|
filemode="w",
|
|
level=logging.INFO,
|
|
format="%(name)s - %(levelname)s - %(message)s",
|
|
)
|
|
# General
|
|
CONF_DEBUG: bool = True
|
|
# Google
|
|
GOOGLE_API_KEY: str = ""
|
|
GOOGLE_SEARCH_ENGINE_ID: str = ""
|
|
# NLP
|
|
NLP_CONF_MODE: str = "default"
|
|
|
|
|
|
def GOOGLE_API_CONFIG(_GOOGLE_API_KEY: str, _GOOGLE_SEARCH_ENGINE_ID: str) -> None:
|
|
global GOOGLE_SEARCH_ENGINE_ID, GOOGLE_API_KEY
|
|
GOOGLE_API_KEY = _GOOGLE_API_KEY
|
|
GOOGLE_SEARCH_ENGINE_ID = _GOOGLE_SEARCH_ENGINE_ID
|
|
if CONF_DEBUG and _GOOGLE_API_KEY != "":
|
|
logging.info(f"API_KEY set")
|
|
if CONF_DEBUG and _GOOGLE_SEARCH_ENGINE_ID != "":
|
|
logging.info(f"SEARCH_ENGINE_ID set")
|
|
|
|
|
|
def GET_GOOGLE_API_CONFIG() -> tuple[str, str]:
|
|
global GOOGLE_SEARCH_ENGINE_ID, GOOGLE_API_KEY
|
|
return (GOOGLE_API_KEY, GOOGLE_SEARCH_ENGINE_ID)
|
|
|
|
|
|
# TODO: work in progress
|
|
# class GoogleAPI:
|
|
# def __init__(self) -> None:
|
|
# self.GOOGLE_SEARCH_API_KEY: str = ""
|
|
# self.GOOGLE_SEARCH_ENGINE_ID: str = ""
|
|
|
|
# @property
|
|
# def google_search_api_key
|
|
|
|
|
|
def NLP_config(mode: str = "default", debug: bool = True) -> None:
|
|
global NLP_CONF_MODE, CONF_DEBUG
|
|
CONF_DEBUG = debug
|
|
if mode == "accuracy" or mode == "speed":
|
|
NLP_CONF_MODE = mode
|
|
else:
|
|
if CONF_DEBUG:
|
|
logging.warn(f"mode: {mode} does not exist")
|