38 lines
951 B
Python
38 lines
951 B
Python
from typing import TypedDict
|
|
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
from wagtail.search import index
|
|
|
|
|
|
class ImageURLs(TypedDict):
|
|
raw: str
|
|
full: str
|
|
regular: str
|
|
small: str
|
|
thumb: str
|
|
|
|
|
|
SIZES = {"full": 2000, "regular": 1080, "small": 400, "thumb": 200}
|
|
|
|
|
|
class UnsplashPhoto(index.Indexed, models.Model):
|
|
unsplash_id = models.CharField(unique=True, max_length=11, db_index=True)
|
|
data = models.JSONField()
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
data_last_updated = models.DateTimeField(default=timezone.now)
|
|
|
|
search_fields = [
|
|
index.SearchField("unsplash_id", boost=10),
|
|
index.SearchField("get_description"),
|
|
]
|
|
|
|
def get_description(self) -> str:
|
|
return self.data["description"]
|
|
|
|
def get_image_urls(self) -> ImageURLs:
|
|
return self.data["urls"]
|
|
|
|
def get_thumbnail_url(self) -> str:
|
|
return self.get_image_urls()["thumb"]
|