215 lines
7.6 KiB
Python
215 lines
7.6 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
import re
|
|
|
|
|
|
###########################
|
|
### User Profile Class ####
|
|
###########################
|
|
class UserProfile(models.Model):
|
|
id = models.AutoField(primary_key=True) # not technically needed
|
|
notify = models.BooleanField("Send notifications", default=True, help_text="When enabled, user will recieve 'flag out of date' notifications")
|
|
alias = models.CharField(core=True, maxlength=50, help_text="Required field")
|
|
public_email = models.CharField(core=True, maxlength=50, help_text="Required field")
|
|
other_contact = models.CharField(maxlength=100, null=True, blank=True)
|
|
website = models.URLField(null=True, blank=True)
|
|
yob = models.IntegerField(null=True, blank=True)
|
|
location = models.CharField(maxlength=50, null=True, blank=True)
|
|
languages = models.CharField(maxlength=50, null=True, blank=True)
|
|
interests = models.CharField(maxlength=255, null=True, blank=True)
|
|
occupation = models.CharField(maxlength=50, null=True, blank=True)
|
|
roles = models.CharField(maxlength=255, null=True, blank=True)
|
|
favorite_distros = models.CharField(maxlength=255, null=True, blank=True)
|
|
picture = models.FileField(upload_to='devs', default='devs/silhouette.png')
|
|
user = models.ForeignKey(User, related_name='userprofile_user', edit_inline=models.STACKED, num_in_admin=1, min_num_in_admin=1, max_num_in_admin=1, num_extra_on_change=0, unique=True)
|
|
class Meta:
|
|
db_table = 'user_profiles'
|
|
verbose_name = 'Additional Profile Data'
|
|
verbose_name_plural = 'Additional Profile Data'
|
|
|
|
|
|
#######################
|
|
### Manager Classes ###
|
|
#######################
|
|
class TodolistManager(models.Manager):
|
|
def get_incomplete(self):
|
|
results = []
|
|
for l in self.all().order_by('-date_added'):
|
|
if TodolistPkg.objects.filter(list=l.id).filter(complete=False).count() > 0:
|
|
results.append(l)
|
|
return results
|
|
|
|
class PackageManager(models.Manager):
|
|
def get_flag_stats(self):
|
|
results = []
|
|
# first the orphans
|
|
unflagged = self.filter(maintainer=0).count()
|
|
flagged = self.filter(maintainer=0).filter(needupdate=True).count()
|
|
results.append((User(id=0,first_name='Orphans'), unflagged, flagged))
|
|
# now the rest
|
|
for maint in User.objects.all().order_by('first_name'):
|
|
unflagged = self.filter(maintainer=maint.id).count()
|
|
flagged = self.filter(maintainer=maint.id).filter(needupdate=True).count()
|
|
results.append((maint, unflagged, flagged))
|
|
return results
|
|
|
|
|
|
#############################
|
|
### General Model Classes ###
|
|
#############################
|
|
class Mirror(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
domain = models.CharField(maxlength=255)
|
|
country = models.CharField(maxlength=255)
|
|
url = models.CharField(maxlength=255)
|
|
protocol_list = models.CharField(maxlength=255, null=True, blank=True)
|
|
admin_email = models.CharField(maxlength=255, null=True, blank=True)
|
|
def __str__(self):
|
|
return self.domain
|
|
class Meta:
|
|
db_table = 'common_mirror'
|
|
class Admin:
|
|
list_display = ('domain', 'country')
|
|
list_filter = ('country',)
|
|
ordering = ['domain']
|
|
search_fields = ('domain')
|
|
pass
|
|
|
|
class Donator(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
name = models.CharField(maxlength=255)
|
|
def __str__(self):
|
|
return self.name
|
|
class Meta:
|
|
db_table = 'common_donator'
|
|
class Admin:
|
|
ordering = ['name']
|
|
search_fields = ('name')
|
|
pass
|
|
|
|
class News(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
author = models.ForeignKey(User, related_name='news_author')
|
|
postdate = models.DateField(auto_now_add=True)
|
|
title = models.CharField(maxlength=255)
|
|
content = models.TextField()
|
|
class Meta:
|
|
db_table = 'news'
|
|
verbose_name_plural = 'news'
|
|
get_latest_by = 'postdate'
|
|
ordering = ['-postdate', '-id']
|
|
|
|
def get_absolute_url(self):
|
|
return '/news/%i/' % self.id
|
|
|
|
class Category(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
category = models.CharField(maxlength=255)
|
|
class Meta:
|
|
db_table = 'categories'
|
|
verbose_name_plural = 'categories'
|
|
|
|
class Repo(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
name = models.CharField(maxlength=255)
|
|
class Meta:
|
|
db_table = 'repos'
|
|
ordering = ['name']
|
|
def last_update(self):
|
|
try:
|
|
latest = Package.objects.filter(
|
|
repo__name__exact=self.name).order_by('-last_update')[0]
|
|
return latest.last_update
|
|
except IndexError:
|
|
return "N/A"
|
|
|
|
class Package(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
repo = models.ForeignKey(Repo)
|
|
maintainer = models.ForeignKey(User, related_name='package_maintainer')
|
|
category = models.ForeignKey(Category)
|
|
needupdate = models.BooleanField(default=False)
|
|
pkgname = models.CharField(maxlength=255)
|
|
pkgver = models.CharField(maxlength=255)
|
|
pkgrel = models.CharField(maxlength=255)
|
|
pkgdesc = models.CharField(maxlength=255)
|
|
url = models.URLField()
|
|
sources = models.TextField()
|
|
depends = models.TextField()
|
|
last_update = models.DateTimeField(null=True, blank=True)
|
|
objects = PackageManager()
|
|
class Meta:
|
|
db_table = 'packages'
|
|
get_latest_by = 'last_update'
|
|
|
|
def get_absolute_url(self):
|
|
return '/packages/%i/' % self.id
|
|
|
|
def depends_urlize(self):
|
|
urls = ''
|
|
for dep in self.depends.split(' '):
|
|
# shave off any version qualifiers
|
|
nameonly = re.match(r"([a-z0-9-]+)", dep).group(1)
|
|
try:
|
|
p = Package.objects.filter(pkgname=nameonly)[0]
|
|
except IndexError:
|
|
# couldn't find a package in the DB -- it might be a virtual depend
|
|
urls = urls + '<li>' + dep + '</li>'
|
|
continue
|
|
url = '<li><a href="/packages/' + str(p.id) + '">' + dep + '</a></li>'
|
|
urls = urls + url
|
|
return urls
|
|
|
|
def sources_urlize(self):
|
|
urls = ''
|
|
for source in self.sources.split(' '):
|
|
if re.search('://', source):
|
|
url = '<li><a href="' + source + '">' + source + '</a></li>'
|
|
else:
|
|
url = '<li>' + source + '</li>'
|
|
urls = urls + url
|
|
return urls
|
|
|
|
class PackageFile(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
pkg = models.ForeignKey(Package)
|
|
path = models.CharField(maxlength=255)
|
|
class Meta:
|
|
db_table = 'packages_files'
|
|
|
|
class Todolist(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
creator = models.ForeignKey(User, related_name='todolist_creator')
|
|
name = models.CharField(maxlength=255)
|
|
description = models.TextField()
|
|
date_added = models.DateField(auto_now_add=True)
|
|
objects = TodolistManager()
|
|
class Meta:
|
|
db_table = 'todolists'
|
|
|
|
class TodolistPkg(models.Model):
|
|
id = models.AutoField(primary_key=True)
|
|
list = models.ForeignKey(Todolist)
|
|
pkg = models.ForeignKey(Package)
|
|
complete = models.BooleanField(default=False)
|
|
class Meta:
|
|
db_table = 'todolists_pkgs'
|
|
unique_together = (('list','pkg'),)
|
|
|
|
class Wikipage(models.Model):
|
|
"""Wiki page storage"""
|
|
title = models.CharField(maxlength=255)
|
|
content = models.TextField()
|
|
last_author = models.ForeignKey(User, related_name='wikipage_last_author')
|
|
class Meta:
|
|
db_table = 'wikipages'
|
|
|
|
def editurl(self):
|
|
return "/wiki/edit/" + self.title + "/"
|
|
|
|
def __repr__(self):
|
|
return self.title
|
|
|
|
# vim: set ts=4 sw=4 et:
|
|
|