Skip to content

Commit

Permalink
!squash settings
Browse files Browse the repository at this point in the history
  • Loading branch information
tony committed Oct 15, 2022
1 parent 44f1644 commit c312edd
Showing 1 changed file with 42 additions and 7 deletions.
49 changes: 42 additions & 7 deletions src/robots/settings.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,44 @@
import sys
import typing as t

from django.conf import settings
from django.test.signals import setting_changed

DEFAULT_ROBOTS_SETTINGS = {
"SITEMAP_URLS": [],
"USE_SITEMAP": True,
"USE_HOST": True,
"CACHE_TIMEOUT": None,
"SITE_BY_REQUEST": False,
"USE_SCHEME_IN_HOST": False,
"SITEMAP_VIEW_NAME": False,
}


def getdefault(setting: str) -> t.Any:
return getattr(settings, f"ROBOTS_{setting}", DEFAULT_ROBOTS_SETTINGS[setting])


SITEMAP_URLS: t.List[str] = getdefault("SITEMAP_URLS")
USE_SITEMAP: bool = getdefault("USE_SITEMAP")
USE_HOST: bool = getdefault("USE_HOST")
CACHE_TIMEOUT: t.Optional[int] = getdefault("CACHE_TIMEOUT")
SITE_BY_REQUEST: bool = getdefault("SITE_BY_REQUEST")
USE_SCHEME_IN_HOST: bool = getdefault("USE_SCHEME_IN_HOST")
SITEMAP_VIEW_NAME: bool = getdefault("SITEMAP_VIEW_NAME")


def reload_robots_settings(
setting: str, value: t.Any, enter: bool, **kwargs: object
) -> None:
if not setting.startswith("ROBOTS_"): # pragma: nocover
return

setting = setting.replace("ROBOTS_", "")
if enter: # setting applied
setattr(sys.modules[__name__], setting, value)
else:
setattr(sys.modules[__name__], setting, getdefault(setting))


SITEMAP_URLS = getattr(settings, "ROBOTS_SITEMAP_URLS", [])
USE_SITEMAP = getattr(settings, "ROBOTS_USE_SITEMAP", True)
USE_HOST = getattr(settings, "ROBOTS_USE_HOST", True)
CACHE_TIMEOUT = getattr(settings, "ROBOTS_CACHE_TIMEOUT", None)
SITE_BY_REQUEST = getattr(settings, "ROBOTS_SITE_BY_REQUEST", False)
USE_SCHEME_IN_HOST = getattr(settings, "ROBOTS_USE_SCHEME_IN_HOST", False)
SITEMAP_VIEW_NAME = getattr(settings, "ROBOTS_SITEMAP_VIEW_NAME", False)
setting_changed.connect(reload_robots_settings)

0 comments on commit c312edd

Please sign in to comment.