mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-11 04:07:57 +03:00
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
CHOICES = (
|
|
(1, 'this'),
|
|
(2, _('that'))
|
|
)
|
|
|
|
|
|
class Pet(models.Model):
|
|
name = models.CharField(max_length=30)
|
|
|
|
|
|
class FilmDetails(models.Model):
|
|
location = models.CharField(max_length=30)
|
|
film = models.OneToOneField('Film', on_delete=models.CASCADE, related_name='details')
|
|
|
|
|
|
class Film(models.Model):
|
|
reporters = models.ManyToManyField('Reporter',
|
|
related_name='films')
|
|
|
|
class DoeReporterManager(models.Manager):
|
|
def get_queryset(self):
|
|
return super(DoeReporterManager, self).get_queryset().filter(last_name="Doe")
|
|
|
|
class Reporter(models.Model):
|
|
first_name = models.CharField(max_length=30)
|
|
last_name = models.CharField(max_length=30)
|
|
email = models.EmailField()
|
|
pets = models.ManyToManyField('self')
|
|
a_choice = models.CharField(max_length=30, choices=CHOICES)
|
|
objects = models.Manager()
|
|
doe_objects = DoeReporterManager()
|
|
|
|
reporter_type = models.IntegerField(
|
|
'Reporter Type',
|
|
null=True,
|
|
blank=True,
|
|
choices=[(1, u'Regular'), (2, u'CNN Reporter')]
|
|
)
|
|
|
|
def __str__(self): # __unicode__ on Python 2
|
|
return "%s %s" % (self.first_name, self.last_name)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
"""
|
|
Override the init method so that during runtime, Django
|
|
can know that this object can be a CNNReporter by casting
|
|
it to the proxy model. Otherwise, as far as Django knows,
|
|
when a CNNReporter is pulled from the database, it is still
|
|
of type Reporter. This was added to test proxy model support.
|
|
"""
|
|
super(Reporter, self).__init__(*args, **kwargs)
|
|
if self.reporter_type == 2: # quick and dirty way without enums
|
|
self.__class__ = CNNReporter
|
|
|
|
class CNNReporter(Reporter):
|
|
"""
|
|
This class is a proxy model for Reporter, used for testing
|
|
proxy model support
|
|
"""
|
|
class Meta:
|
|
proxy = True
|
|
|
|
|
|
class Article(models.Model):
|
|
headline = models.CharField(max_length=100)
|
|
pub_date = models.DateField()
|
|
pub_date_time = models.DateTimeField()
|
|
reporter = models.ForeignKey(Reporter, on_delete=models.CASCADE, related_name='articles')
|
|
editor = models.ForeignKey(Reporter, on_delete=models.CASCADE, related_name='edited_articles_+')
|
|
lang = models.CharField(max_length=2, help_text='Language', choices=[
|
|
('es', 'Spanish'),
|
|
('en', 'English')
|
|
], default='es')
|
|
importance = models.IntegerField('Importance', null=True, blank=True,
|
|
choices=[(1, u'Very important'), (2, u'Not as important')])
|
|
|
|
def __str__(self): # __unicode__ on Python 2
|
|
return self.headline
|
|
|
|
class Meta:
|
|
ordering = ('headline',)
|