django-rest-framework/rest_framework/tests/models.py

109 lines
2.7 KiB
Python
Raw Normal View History

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.generic import GenericForeignKey, GenericRelation
2012-09-28 17:28:50 +04:00
# from django.contrib.auth.models import Group
# class CustomUser(models.Model):
# """
# A custom user model, which uses a 'through' table for the foreign key
# """
# username = models.CharField(max_length=255, unique=True)
# groups = models.ManyToManyField(
# to=Group, blank=True, null=True, through='UserGroupMap'
# )
# @models.permalink
# def get_absolute_url(self):
# return ('custom_user', (), {
# 'pk': self.id
# })
# class UserGroupMap(models.Model):
# user = models.ForeignKey(to=CustomUser)
# group = models.ForeignKey(to=Group)
# @models.permalink
# def get_absolute_url(self):
# return ('user_group_map', (), {
# 'pk': self.id
# })
2012-10-04 12:29:12 +04:00
def foobar():
return 'foobar'
class RESTFrameworkModel(models.Model):
2012-10-03 13:51:38 +04:00
"""
Base for test models that sets app_label, so they play nicely.
2012-10-03 13:51:38 +04:00
"""
class Meta:
app_label = 'rest_framework'
abstract = True
2012-10-03 13:51:38 +04:00
2012-09-28 17:28:50 +04:00
class Anchor(RESTFrameworkModel):
text = models.CharField(max_length=100, default='anchor')
2012-09-28 17:28:50 +04:00
2012-10-03 13:51:38 +04:00
class BasicModel(RESTFrameworkModel):
text = models.CharField(max_length=100)
2012-10-03 13:51:38 +04:00
2012-10-04 12:29:12 +04:00
class DefaultValueModel(RESTFrameworkModel):
text = models.CharField(default='foobar', max_length=100)
class CallableDefaultValueModel(RESTFrameworkModel):
text = models.CharField(default=foobar, max_length=100)
class ManyToManyModel(RESTFrameworkModel):
rel = models.ManyToManyField(Anchor)
# Models to test generic relations
class Tag(RESTFrameworkModel):
tag_name = models.SlugField()
class TaggedItem(RESTFrameworkModel):
tag = models.ForeignKey(Tag, related_name='items')
content_type = models.ForeignKey(ContentType)
object_id = models.PositiveIntegerField()
content_object = GenericForeignKey('content_type', 'object_id')
def __unicode__(self):
return self.tag.tag_name
class Bookmark(RESTFrameworkModel):
url = models.URLField()
tags = GenericRelation(TaggedItem)
2012-10-08 15:52:56 +04:00
# Model for regression test for #285
class Comment(RESTFrameworkModel):
email = models.EmailField()
content = models.CharField(max_length=200)
created = models.DateTimeField(auto_now_add=True)
class ActionItem(RESTFrameworkModel):
title = models.CharField(max_length=200)
done = models.BooleanField(default=False)
# Models for reverse relations
class BlogPost(RESTFrameworkModel):
title = models.CharField(max_length=100)
class BlogPostComment(RESTFrameworkModel):
text = models.TextField()
blog_post = models.ForeignKey(BlogPost)