From: Roland Häder Date: Thu, 9 Jan 2025 01:14:14 +0000 (+0100) Subject: Continued: X-Git-Url: https://git.mxchange.org/?a=commitdiff_plain;h=f5338382ff7032244874aa2fafe74e8a50b3b87e;p=fba.git Continued: - added needed timeout parameter (pylint) - removed unused imports (pylint) --- diff --git a/daemon.py b/daemon.py index e732851..f4d8f43 100755 --- a/daemon.py +++ b/daemon.py @@ -45,6 +45,9 @@ from fba.helpers import tidyup from fba.models import blocks from fba.models import instances +# Timeout +_timeout = (config.get("connection_timeout"), config.get("read_timeout")) + router = fastapi.FastAPI(docs_url=config.get("base_url") + "/docs", redoc_url=config.get("base_url") + "/redoc") router.mount( "/static", @@ -360,7 +363,7 @@ def scoreboard(request: Request, mode: str, amount: int) -> None: elif amount <= 0: raise HTTPException(status_code=500, detail="Invalid amount specified") - response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/scoreboard.json?mode={mode}&amount={amount}") + response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/scoreboard.json?mode={mode}&amount={amount}", timeout=_timeout) if response is None: raise HTTPException(status_code=500, detail="Could not determine scores") @@ -383,7 +386,7 @@ def list_domains(request: Request, mode: str, value: str, amount: int = config.g if mode == "detection_mode" and not instances.valid(value, "detection_mode"): raise HTTPException(status_code=500, detail="Invalid detection mode provided") - response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/list.json?mode={mode}&value={value}&amount={amount}") + response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/list.json?mode={mode}&value={value}&amount={amount}", timeout=_timeout) domainlist = list() if response is not None and response.ok: @@ -411,7 +414,7 @@ def top(request: Request, mode: str, value: str, amount: int = config.get("api_l elif mode in ["domain", "reverse"] and not domain_helper.is_wanted(value): raise HTTPException(status_code=500, detail="Invalid or blocked domain specified") - response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/top.json?mode={mode}&value={value}&amount={amount}") + response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/top.json?mode={mode}&value={value}&amount={amount}", timeout=_timeout) found = 0 blocklist = list() @@ -447,7 +450,7 @@ def infos(request: Request, domain: str) -> None: if not domain_helper.is_wanted(domain): raise HTTPException(status_code=500, detail=f"domain='{domain}' is not wanted") - response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/domain.json?domain={domain}") + response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/domain.json?domain={domain}", timeout=_timeout) if not response.ok or response.status_code > 200 or response.text.strip() == "": raise HTTPException(status_code=response.status_code, detail=response.reason) @@ -533,7 +536,7 @@ def robots(request: Request) -> None: @router.get(config.get("base_url") + "/") def index(request: Request) -> None: # Get info - response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/info.json") + response = requests.get(f"http://{config.get('host')}:{config.get('port')}{config.get('base_url')}/api/info.json", timeout=_timeout) if not response.ok: raise HTTPException(status_code=response.status_code, detail=response.text) diff --git a/fba/helpers/domain.py b/fba/helpers/domain.py index bf40064..7d3ad06 100644 --- a/fba/helpers/domain.py +++ b/fba/helpers/domain.py @@ -18,7 +18,6 @@ import logging from functools import lru_cache from urllib.parse import urlparse -from urllib.parse import urlunparse import validators diff --git a/fba/http/csrf.py b/fba/http/csrf.py index 96c02f2..b004c16 100644 --- a/fba/http/csrf.py +++ b/fba/http/csrf.py @@ -17,7 +17,6 @@ import logging import bs4 -import reqto import requests import validators diff --git a/fba/http/network.py b/fba/http/network.py index 5ca77bc..5237d87 100644 --- a/fba/http/network.py +++ b/fba/http/network.py @@ -25,8 +25,6 @@ import requests import urllib3 import validators -from fba import utils - from fba.helpers import blacklist from fba.helpers import config from fba.helpers import cookies diff --git a/fba/models/instances.py b/fba/models/instances.py index 5672494..d3cde89 100644 --- a/fba/models/instances.py +++ b/fba/models/instances.py @@ -28,7 +28,6 @@ from fba.helpers import blacklist from fba.helpers import cache from fba.helpers import config from fba.helpers import domain as domain_helper -from fba.helpers import tidyup from fba.http import federation from fba.http import network diff --git a/fba/networks/peertube.py b/fba/networks/peertube.py index 63123b3..4b5baf0 100644 --- a/fba/networks/peertube.py +++ b/fba/networks/peertube.py @@ -17,7 +17,6 @@ import logging from fba.helpers import blacklist -from fba.helpers import config from fba.helpers import domain as domain_helper from fba.http import csrf diff --git a/fba/networks/pleroma.py b/fba/networks/pleroma.py index ba6f5e4..3938e53 100644 --- a/fba/networks/pleroma.py +++ b/fba/networks/pleroma.py @@ -20,7 +20,6 @@ import bs4 import validators from fba import database -from fba import utils from fba.helpers import blacklist from fba.helpers import config