This repository has been archived on 2023-09-24. You can view files and clone it, but cannot push or open issues or pull requests.

60 lines
1.8 KiB
Python
Raw Normal View History

import sys
from urllib.parse import urljoin
from django.conf import settings
2022-12-15 00:35:04 -07:00
from django.contrib.staticfiles.storage import staticfiles_storage
class RelativeAbsoluteUrl:
"""
Represents a URL that can have both "relative" and "absolute" forms
for various use either locally or remotely.
"""
absolute: str
relative: str
def __init__(self, absolute: str, relative: str | None = None):
if "://" not in absolute:
raise ValueError(f"Absolute URL {absolute!r} is not absolute!")
self.absolute = absolute
self.relative = relative or absolute
class AutoAbsoluteUrl(RelativeAbsoluteUrl):
"""
Automatically makes the absolute variant by using either settings.MAIN_DOMAIN
or a passed identity's URI domain.
"""
def __init__(self, relative: str, identity=None):
self.relative = relative
if identity:
absolute_prefix = f"https://{identity.domain.uri_domain}/"
else:
absolute_prefix = f"https://{settings.MAIN_DOMAIN}/"
self.absolute = urljoin(absolute_prefix, self.relative)
2022-12-12 07:32:35 -07:00
class StaticAbsoluteUrl(RelativeAbsoluteUrl):
"""
Creates static URLs given only the static-relative path
"""
def __init__(self, path: str):
try:
static_url = staticfiles_storage.url(path)
except ValueError:
# Suppress static issues during the first collectstatic
# Yes, I know it's a big hack! Pull requests welcome :)
if "collectstatic" in sys.argv:
super().__init__("https://example.com/")
return
raise
2022-12-12 07:32:35 -07:00
if "://" in static_url:
super().__init__(static_url)
else:
super().__init__(
urljoin(f"https://{settings.MAIN_DOMAIN}/", static_url), static_url
)