111 lines
2.7 KiB
Python
111 lines
2.7 KiB
Python
"""
|
|
The lookup API
|
|
|
|
This demonstrates features of the database API.
|
|
"""
|
|
|
|
from django.db import models
|
|
from django.db.models.lookups import IsNull
|
|
|
|
|
|
class Alarm(models.Model):
|
|
desc = models.CharField(max_length=100)
|
|
time = models.TimeField()
|
|
|
|
def __str__(self):
|
|
return '%s (%s)' % (self.time, self.desc)
|
|
|
|
|
|
class Author(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
alias = models.CharField(max_length=50, null=True, blank=True)
|
|
|
|
class Meta:
|
|
ordering = ('name',)
|
|
|
|
|
|
class Article(models.Model):
|
|
headline = models.CharField(max_length=100)
|
|
pub_date = models.DateTimeField()
|
|
author = models.ForeignKey(Author, models.SET_NULL, blank=True, null=True)
|
|
slug = models.SlugField(unique=True, blank=True, null=True)
|
|
|
|
class Meta:
|
|
ordering = ('-pub_date', 'headline')
|
|
|
|
def __str__(self):
|
|
return self.headline
|
|
|
|
|
|
class Tag(models.Model):
|
|
articles = models.ManyToManyField(Article)
|
|
name = models.CharField(max_length=100)
|
|
|
|
class Meta:
|
|
ordering = ('name',)
|
|
|
|
|
|
class NulledTextField(models.TextField):
|
|
def get_prep_value(self, value):
|
|
return None if value == '' else value
|
|
|
|
|
|
@NulledTextField.register_lookup
|
|
class NulledTransform(models.Transform):
|
|
lookup_name = 'nulled'
|
|
template = 'NULL'
|
|
|
|
|
|
@NulledTextField.register_lookup
|
|
class IsNullWithNoneAsRHS(IsNull):
|
|
lookup_name = 'isnull_none_rhs'
|
|
can_use_none_as_rhs = True
|
|
|
|
|
|
class Season(models.Model):
|
|
year = models.PositiveSmallIntegerField()
|
|
gt = models.IntegerField(null=True, blank=True)
|
|
nulled_text_field = NulledTextField(null=True)
|
|
|
|
class Meta:
|
|
constraints = [
|
|
models.UniqueConstraint(fields=['year'], name='season_year_unique'),
|
|
]
|
|
|
|
def __str__(self):
|
|
return str(self.year)
|
|
|
|
|
|
class Game(models.Model):
|
|
season = models.ForeignKey(Season, models.CASCADE, related_name='games')
|
|
home = models.CharField(max_length=100)
|
|
away = models.CharField(max_length=100)
|
|
|
|
|
|
class Player(models.Model):
|
|
name = models.CharField(max_length=100)
|
|
games = models.ManyToManyField(Game, related_name='players')
|
|
|
|
|
|
class Product(models.Model):
|
|
name = models.CharField(max_length=80)
|
|
qty_target = models.DecimalField(max_digits=6, decimal_places=2)
|
|
|
|
|
|
class Stock(models.Model):
|
|
product = models.ForeignKey(Product, models.CASCADE)
|
|
short = models.BooleanField(default=False)
|
|
qty_available = models.DecimalField(max_digits=6, decimal_places=2)
|
|
|
|
|
|
class Freebie(models.Model):
|
|
gift_product = models.ForeignKey(Product, models.CASCADE)
|
|
stock_id = models.IntegerField(blank=True, null=True)
|
|
|
|
stock = models.ForeignObject(
|
|
Stock,
|
|
from_fields=['stock_id', 'gift_product'],
|
|
to_fields=['id', 'product'],
|
|
on_delete=models.CASCADE,
|
|
)
|