[beers] Connect beers with producers

This commit is contained in:
2024-10-22 17:34:20 -04:00
parent 08b48371bc
commit 59765b14ca
3 changed files with 72 additions and 3 deletions

View File

@ -2,6 +2,8 @@ from django.apps import apps
from django.db import models
from django.urls import reverse
from django_extensions.db.models import TimeStampedModel
from imagekit.models import ImageSpecField
from imagekit.processors import ResizeToFit
from scrobbles.dataclasses import BeerLogData
from scrobbles.mixins import ScrobblableMixin
@ -11,6 +13,10 @@ BNULL = {"blank": True, "null": True}
class BeerProducer(TimeStampedModel):
description = models.TextField(**BNULL)
location = models.CharField(max_length=255, **BNULL)
beeradvocate_id = models.CharField(max_length=255, **BNULL)
def find_or_create(cls, title: str) -> "BeerProducer":
return cls.objects.filter(title=title).first()
class Beer(ScrobblableMixin):
@ -21,11 +27,41 @@ class Beer(ScrobblableMixin):
non_alcoholic = models.BooleanField(default=False)
beeradvocate_id = models.CharField(max_length=255, **BNULL)
beeradvocate_score = models.SmallIntegerField(**BNULL)
beeradvocate_image = models.ImageField(
upload_to="beers/beeradvcoate/", **BNULL
)
beeradvocate_image_small = ImageSpecField(
source="cover",
processors=[ResizeToFit(100, 100)],
format="JPEG",
options={"quality": 60},
)
beeradvocate_image_medium = ImageSpecField(
source="cover",
processors=[ResizeToFit(300, 300)],
format="JPEG",
options={"quality": 75},
)
untappd_id = models.CharField(max_length=255, **BNULL)
producer = models.ForeignKey(
BeerProducer, on_delete=models.DO_NOTHING, **BNULL
)
def get_absolute_url(self):
return reverse("beers:beer_detail", kwargs={"slug": self.uuid})
def beeradvocate_link(self):
link = ""
if self.beeradvocate_id and self.producer:
link = f"https://www.beeradvocate.com/beer/profile/{self.producer.beeradvocate_id}/{self.beeradvocate_id}/"
return link
def primary_image_url(self) -> str:
url = ""
if self.beeradvocate_image:
url = self.beeradvocate_image.url
return url
@property
def logdata_cls(self):
return BeerLogData