mirror of
https://github.com/daniviga/django-ram.git
synced 2025-08-04 13:17:50 +02:00
Use slugs to filter
This commit is contained in:
93
ram/metadata/migrations/0011_company_slug_and_more.py
Normal file
93
ram/metadata/migrations/0011_company_slug_and_more.py
Normal file
@@ -0,0 +1,93 @@
|
||||
# Generated by Django 4.1.5 on 2023-01-09 11:22
|
||||
|
||||
from django.db import migrations, models
|
||||
|
||||
|
||||
from ram.utils import slugify
|
||||
|
||||
|
||||
def create_slug(apps, schema_editor):
|
||||
fields = ["Company", "Manufacturer", "RollingStockType", "Scale"]
|
||||
|
||||
for m in fields:
|
||||
model = apps.get_model("metadata", m)
|
||||
|
||||
for row in model.objects.all():
|
||||
if hasattr(row, "type"):
|
||||
row.slug = slugify("{} {}".format(row.type, row.category))
|
||||
elif hasattr(row, "scale"):
|
||||
row.slug = slugify(row.scale)
|
||||
else:
|
||||
row.slug = slugify(row.name)
|
||||
|
||||
row.save(update_fields=["slug"])
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
dependencies = [
|
||||
("metadata", "0010_alter_manufacturer_category"),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.AddField(
|
||||
model_name="company",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=64, blank=True
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="manufacturer",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=128, blank=True
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="rollingstocktype",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=128, blank=True
|
||||
),
|
||||
),
|
||||
migrations.AddField(
|
||||
model_name="scale",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=32, blank=True
|
||||
),
|
||||
),
|
||||
migrations.RunPython(
|
||||
create_slug,
|
||||
reverse_code=migrations.RunPython.noop
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="company",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=64, unique=True
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="manufacturer",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=128, unique=True
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="rollingstocktype",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=128, unique=True
|
||||
),
|
||||
),
|
||||
migrations.AlterField(
|
||||
model_name="scale",
|
||||
name="slug",
|
||||
field=models.CharField(
|
||||
editable=False, max_length=32, unique=True
|
||||
),
|
||||
),
|
||||
]
|
@@ -1,6 +1,7 @@
|
||||
from urllib.parse import quote
|
||||
|
||||
from django.db import models
|
||||
from django.urls import reverse
|
||||
from django.conf import settings
|
||||
from django.dispatch.dispatcher import receiver
|
||||
from django_countries.fields import CountryField
|
||||
@@ -22,6 +23,7 @@ class Property(models.Model):
|
||||
|
||||
class Manufacturer(models.Model):
|
||||
name = models.CharField(max_length=128, unique=True)
|
||||
slug = models.CharField(max_length=128, unique=True, editable=False)
|
||||
category = models.CharField(
|
||||
max_length=64, choices=settings.MANUFACTURER_TYPES
|
||||
)
|
||||
@@ -36,8 +38,13 @@ class Manufacturer(models.Model):
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def safe_name(self):
|
||||
return quote(self.__str__().lower(), safe="& ")
|
||||
def get_absolute_url(self):
|
||||
return reverse(
|
||||
"filtered", kwargs={
|
||||
"_filter": "manufacturer",
|
||||
"search": self.slug,
|
||||
}
|
||||
)
|
||||
|
||||
def logo_thumbnail(self):
|
||||
return get_image_preview(self.logo.url)
|
||||
@@ -47,6 +54,7 @@ class Manufacturer(models.Model):
|
||||
|
||||
class Company(models.Model):
|
||||
name = models.CharField(max_length=64, unique=True)
|
||||
slug = models.CharField(max_length=64, unique=True, editable=False)
|
||||
extended_name = models.CharField(max_length=128, blank=True)
|
||||
country = CountryField()
|
||||
freelance = models.BooleanField(default=False)
|
||||
@@ -61,8 +69,13 @@ class Company(models.Model):
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def safe_name(self):
|
||||
return quote(self.__str__().lower(), safe="& ")
|
||||
def get_absolute_url(self):
|
||||
return reverse(
|
||||
"filtered", kwargs={
|
||||
"_filter": "company",
|
||||
"search": self.slug,
|
||||
}
|
||||
)
|
||||
|
||||
def logo_thumbnail(self):
|
||||
return get_image_preview(self.logo.url)
|
||||
@@ -94,6 +107,7 @@ class Decoder(models.Model):
|
||||
|
||||
class Scale(models.Model):
|
||||
scale = models.CharField(max_length=32, unique=True)
|
||||
slug = models.CharField(max_length=32, unique=True, editable=False)
|
||||
ratio = models.CharField(max_length=16, blank=True)
|
||||
gauge = models.CharField(max_length=16, blank=True)
|
||||
tracks = models.CharField(max_length=16, blank=True)
|
||||
@@ -101,11 +115,40 @@ class Scale(models.Model):
|
||||
class Meta:
|
||||
ordering = ["scale"]
|
||||
|
||||
def get_absolute_url(self):
|
||||
return reverse(
|
||||
"filtered", kwargs={
|
||||
"_filter": "scale",
|
||||
"search": self.slug,
|
||||
}
|
||||
)
|
||||
|
||||
def __str__(self):
|
||||
return str(self.scale)
|
||||
|
||||
def safe_name(self):
|
||||
return quote(self.__str__(), safe="& ")
|
||||
|
||||
class RollingStockType(models.Model):
|
||||
type = models.CharField(max_length=64)
|
||||
order = models.PositiveSmallIntegerField()
|
||||
category = models.CharField(
|
||||
max_length=64, choices=settings.ROLLING_STOCK_TYPES
|
||||
)
|
||||
slug = models.CharField(max_length=128, unique=True, editable=False)
|
||||
|
||||
class Meta(object):
|
||||
unique_together = ("category", "type")
|
||||
ordering = ["order"]
|
||||
|
||||
def get_absolute_url(self):
|
||||
return reverse(
|
||||
"filtered", kwargs={
|
||||
"_filter": "type",
|
||||
"search": self.slug,
|
||||
}
|
||||
)
|
||||
|
||||
def __str__(self):
|
||||
return "{0} {1}".format(self.type, self.category)
|
||||
|
||||
|
||||
class Tag(models.Model):
|
||||
@@ -115,28 +158,20 @@ class Tag(models.Model):
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def safe_name(self):
|
||||
return self.slug
|
||||
def get_absolute_url(self):
|
||||
return reverse(
|
||||
"filtered", kwargs={
|
||||
"_filter": "tag",
|
||||
"search": self.slug,
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
|
||||
@receiver(models.signals.pre_save, sender=Manufacturer)
|
||||
@receiver(models.signals.pre_save, sender=Company)
|
||||
@receiver(models.signals.pre_save, sender=Scale)
|
||||
@receiver(models.signals.pre_save, sender=RollingStockType)
|
||||
@receiver(models.signals.pre_save, sender=Tag)
|
||||
def slug_pre_save(sender, instance, **kwargs):
|
||||
instance.slug = slugify(instance.name)
|
||||
|
||||
|
||||
class RollingStockType(models.Model):
|
||||
type = models.CharField(max_length=64)
|
||||
order = models.PositiveSmallIntegerField()
|
||||
category = models.CharField(
|
||||
max_length=64, choices=settings.ROLLING_STOCK_TYPES
|
||||
)
|
||||
|
||||
class Meta(object):
|
||||
unique_together = ("category", "type")
|
||||
ordering = ["order"]
|
||||
|
||||
def __str__(self):
|
||||
return "{0} {1}".format(self.type, self.category)
|
||||
|
||||
def safe_name(self):
|
||||
return quote(self.__str__().lower(), safe="& ")
|
||||
instance.slug = slugify(instance.__str__())
|
||||
|
Reference in New Issue
Block a user