Mercurial > public > sg101
view weblinks/models.py @ 1188:cb712b2c34af
Add JSON option to dlcatreport management command.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Mon, 27 Dec 2021 17:30:26 -0600 (2021-12-27) |
parents | eeaf387803c6 |
children |
line wrap: on
line source
""" This module contains the models for the weblinks application. """ import datetime from django.contrib.auth.models import User from django.core.urlresolvers import reverse from django.db import models class Category(models.Model): """Links belong to categories""" title = models.CharField(max_length=64) slug = models.SlugField(max_length=64) description = models.TextField(blank=True) count = models.IntegerField(default=0) def __unicode__(self): return self.title class Meta: verbose_name_plural = 'Categories' ordering = ('title', ) class PublicLinkManager(models.Manager): """The manager for all public links.""" def get_queryset(self): return super(PublicLinkManager, self).get_queryset().filter( is_public=True).select_related() class LinkBase(models.Model): """Abstract model to aggregate common fields of a web link.""" category = models.ForeignKey(Category) title = models.CharField(max_length=128) url = models.URLField(db_index=True) description = models.TextField(blank=True) user = models.ForeignKey(User) date_added = models.DateTimeField(db_index=True) update_date = models.DateTimeField(db_index=True, blank=True) class Meta: abstract = True class Link(LinkBase): """Model to represent a web link""" hits = models.IntegerField(default=0) is_public = models.BooleanField(default=False, db_index=True) # Managers: objects = models.Manager() public_objects = PublicLinkManager() class Meta: ordering = ('title', ) def __unicode__(self): return self.title def save(self, *args, **kwargs): if not self.pk: if not self.date_added: self.date_added = datetime.datetime.now() self.update_date = self.date_added else: self.update_date = datetime.datetime.now() super(Link, self).save(*args, **kwargs) def get_absolute_url(self): return reverse('weblinks-link_detail', args=[str(self.pk)]) def search_title(self): return self.title def search_summary(self): return self.description class PendingLink(LinkBase): """This model represents links that users submit. They must be approved by an admin before they become visible on the site. """ class Meta: ordering = ('date_added', ) def __unicode__(self): return self.title def save(self, *args, **kwargs): if not self.pk: self.date_added = datetime.datetime.now() self.update_date = self.date_added else: self.update_date = datetime.datetime.now() super(PendingLink, self).save(*args, **kwargs) class FlaggedLinkManager(models.Manager): def create(self, link, user): flagged_link = FlaggedLink(link = link, user = user, approved = False) flagged_link.save() class FlaggedLink(models.Model): """Model to represent links that have been flagged as broken by users""" link = models.ForeignKey(Link) user = models.ForeignKey(User) date_flagged = models.DateField(auto_now_add = True) approved = models.BooleanField(default = False, help_text = 'Check this and save to remove the referenced link from the database') objects = FlaggedLinkManager() def save(self, *args, **kwargs): if self.approved: self.link.delete() self.delete() else: super(FlaggedLink, self).save(*args, **kwargs) def url(self): return self.link.url def get_link_url(self): return '<a href="%s">Link #%d</a>' % (self.link.get_absolute_url(), self.link.id) get_link_url.allow_tags = True get_link_url.short_description = "View Link on Site" def __unicode__(self): return self.link.title class Meta: ordering = ('-date_flagged', )