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 Press(models.Model): id = models.AutoField(primary_key=True) name = models.CharField(maxlength=255) url = models.CharField(maxlength=255) def __str__(self): return self.name class Meta: db_table = 'press' class Admin: ordering = ['name'] search_fields = ('name') pass class AltForums(models.Model): id = models.AutoField(primary_key=True) language = models.CharField(maxlength=255) url = models.CharField(maxlength=255) name = models.CharField(maxlength=255) def __str__(self): return self.name class Meta: db_table = 'alt_forums' class Admin: list_display = ('language', 'name') list_filter = ('language',) ordering = ['name'] search_fields = ('name') pass class Donor(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 Arch(models.Model): id = models.AutoField(primary_key=True) name = models.CharField(maxlength=255) class Meta: db_table = 'arch' ordering = ['name'] 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) arch = models.ForeignKey(Arch) maintainer = models.ForeignKey(User, related_name='package_maintainer') 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() depends = models.ForeignKey(PackageDepends) 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.packagedepends_set.all(): try: p = Package.objects.filter(pkgname=dep.depname) except IndexError: # couldn't find a package in the DB # it might be a virtual depend urls.append('