from django.db import models from django.db.models import Q from django.contrib.auth.models import User from django.contrib.sites.models import Site from main.middleware import get_user from packages.models import PackageRelation ########################### ### 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, send user 'flag out of date' notifications") alias = models.CharField( max_length=50, help_text="Required field") public_email = models.CharField( max_length=50, help_text="Required field") other_contact = models.CharField(max_length=100, null=True, blank=True) website = models.CharField(max_length=200, null=True, blank=True) yob = models.IntegerField(null=True, blank=True) location = models.CharField(max_length=50, null=True, blank=True) languages = models.CharField(max_length=50, null=True, blank=True) interests = models.CharField(max_length=255, null=True, blank=True) occupation = models.CharField(max_length=50, null=True, blank=True) roles = models.CharField(max_length=255, null=True, blank=True) favorite_distros = models.CharField(max_length=255, null=True, blank=True) picture = models.FileField(upload_to='devs', default='devs/silhouette.png') user = models.ForeignKey( User, related_name='userprofile_user', unique=True) allowed_repos = models.ManyToManyField('Repo', blank=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 incomplete(self): return self.filter(todolistpkg__complete=False).distinct() class PackageManager(models.Manager): def flagged(self): return self.get_query_set().filter(needupdate=True) ############################# ### General Model Classes ### ############################# TIER_CHOICES = ( (0, 'Tier 0'), (1, 'Tier 1'), (2, 'Tier 2'), (-1, 'Untiered'), ) class Mirror(models.Model): name = models.CharField(max_length=255) tier = models.SmallIntegerField(default=2, choices=TIER_CHOICES) upstream = models.ForeignKey('self', null=True) country = models.CharField(max_length=255, db_index=True) admin_email = models.EmailField(max_length=255, blank=True) public = models.BooleanField(default=True) active = models.BooleanField(default=True) isos = models.BooleanField(default=True) rsync_user = models.CharField(max_length=50, blank=True, default='') rsync_password = models.CharField(max_length=50, blank=True, default='') notes = models.TextField(blank=True) class Meta: ordering = ('country', 'name') def __unicode__(self): return self.name def supported_protocols(self): protocols = MirrorProtocol.objects.filter(urls__mirror=self).distinct() return ", ".join([p.protocol for p in protocols]) class MirrorProtocol(models.Model): protocol = models.CharField(max_length=10, unique=True) def __unicode__(self): return self.protocol class Meta: verbose_name = 'Mirror Protocol' class MirrorUrl(models.Model): url = models.CharField(max_length=255) protocol = models.ForeignKey(MirrorProtocol, related_name="urls") mirror = models.ForeignKey(Mirror, related_name="urls") def __unicode__(self): return self.url class Meta: verbose_name = 'Mirror URL' class MirrorRsync(models.Model): ip = models.CharField(max_length=24) mirror = models.ForeignKey(Mirror, related_name="rsync_ips") def __unicode__(self): return "%s" % (self.ip) class Meta: verbose_name = 'Mirror Rsync IP' class Donor(models.Model): id = models.AutoField(primary_key=True) name = models.CharField(max_length=255, unique=True) def __unicode__(self): return self.name class Meta: db_table = 'donors' ordering = ['name'] class News(models.Model): id = models.AutoField(primary_key=True) author = models.ForeignKey(User, related_name='news_author', default=get_user) postdate = models.DateField(auto_now_add=True) title = models.CharField(max_length=255) content = models.TextField() def __unicode__(self): return self.title 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(max_length=255,unique=True) def __unicode__(self): return self.name class Meta: db_table = 'arches' ordering = ['name'] verbose_name_plural = 'arches' class Repo(models.Model): id = models.AutoField(primary_key=True) name = models.CharField(max_length=255,unique=True) testing = models.BooleanField(default=False, help_text="Is this repo meant for package testing?") def __unicode__(self): return self.name class Meta: db_table = 'repos' ordering = ['name'] verbose_name_plural = 'repos' class Package(models.Model): id = models.AutoField(primary_key=True) repo = models.ForeignKey(Repo, related_name="packages") arch = models.ForeignKey(Arch, related_name="packages") needupdate = models.BooleanField(default=False) pkgname = models.CharField(max_length=255, db_index=True) pkgbase = models.CharField(max_length=255, db_index=True) pkgver = models.CharField(max_length=255) pkgrel = models.CharField(max_length=255) pkgdesc = models.CharField(max_length=255, null=True) url = models.CharField(max_length=255, null=True) filename = models.CharField(max_length=255) compressed_size = models.PositiveIntegerField(null=True) installed_size = models.PositiveIntegerField(null=True) build_date = models.DateTimeField(null=True) last_update = models.DateTimeField(null=True, blank=True) files_last_update = models.DateTimeField(null=True, blank=True) license = models.CharField(max_length=255, null=True) objects = PackageManager() class Meta: db_table = 'packages' ordering = ('pkgname',) #get_latest_by = 'last_update' #ordering = ('-last_update',) def __unicode__(self): return self.pkgname def get_absolute_url(self): return '/packages/%s/%s/%s/' % (self.repo.name.lower(), self.arch.name, self.pkgname) def get_full_url(self, proto='http'): '''get a URL suitable for things like email including the domain''' domain = Site.objects.get_current().domain return '%s://%s%s' % (proto, domain, self.get_absolute_url()) @property def maintainers(self): return User.objects.filter( package_relations__pkgbase=self.pkgbase, package_relations__type=PackageRelation.MAINTAINER) @property def signoffs(self): if 'signoffs_cache' in dir(self): if len(self.signoffs_cache) > 0: print self.signoffs_cache return self.signoffs_cache self.signoffs_cache = list(Signoff.objects.filter( pkg=self, pkgver=self.pkgver, pkgrel=self.pkgrel)) return self.signoffs_cache def approved_for_signoff(self): return len(self.signoffs) >= 2 def get_requiredby(self): """ Returns a list of package objects. """ requiredby = Package.objects.select_related('arch', 'repo').filter( packagedepend__depname=self.pkgname, arch__name__in=(self.arch.name, 'any')).distinct() return requiredby.order_by('pkgname') def get_depends(self): """ Returns a list of dicts. Each dict contains ('pkg' and 'dep'). If it represents a found package both vars will be available; else pkg will be None if it is a 'virtual' dependency. """ # object level cache. Doesn't last long, but helps for items rendered # twice in the same template. if 'deps_cache' in dir(self): return self.deps_cache deps = [] # TODO: we can use list comprehension and an 'in' query to make this more effective for dep in self.packagedepend_set.order_by('depname'): pkgs = Package.objects.select_related('arch', 'repo').filter(pkgname=dep.depname) if self.arch.name != 'any': # make sure we match architectures if possible pkgs = pkgs.filter(arch__name__in=(self.arch.name, 'any')) if len(pkgs) == 0: # couldn't find a package in the DB # it should be a virtual depend (or a removed package) pkg = None elif len(pkgs) == 1: pkg = pkgs[0] else: # more than one package, see if we can't shrink it down # grab the first though in case we fail pkg = pkgs[0] if self.repo.testing: pkgs = pkgs.filter(repo__testing=True) else: pkgs = pkgs.filter(repo__testing=False) if len(pkgs) > 0: pkg = pkgs[0] deps.append({'dep': dep, 'pkg': pkg}) self.deps_cache = deps return deps def get_svn_link(self, svnpath): linkbase = "http://repos.archlinux.org/wsvn/%s/%s/%s/" repo = self.repo.name.lower() if repo.startswith('community'): root = 'community' else: root = 'packages' return linkbase % (root, self.pkgbase, svnpath) def get_arch_svn_link(self): repo = self.repo.name.lower() return self.get_svn_link("repos/%s-%s" % (repo, self.arch.name)) def get_trunk_svn_link(self): return self.get_svn_link("trunk") def get_bugs_link(self): repo = self.repo.name.lower() if repo.startswith('community'): project = 5 else: project = 1 return "http://bugs.archlinux.org/?project=%d&string=%s" % \ (project, self.pkgname) def is_same_version(self, other): 'is this package similar, name and version-wise, to another' return self.pkgname == other.pkgname \ and self.pkgver == other.pkgver \ and self.pkgrel == other.pkgrel class Signoff(models.Model): pkg = models.ForeignKey(Package) pkgver = models.CharField(max_length=255) pkgrel = models.CharField(max_length=255) packager = models.ForeignKey(User) class PackageFile(models.Model): id = models.AutoField(primary_key=True) pkg = models.ForeignKey('Package') path = models.CharField(max_length=255) class Meta: db_table = 'package_files' class PackageDepend(models.Model): id = models.AutoField(primary_key=True) pkg = models.ForeignKey('Package') depname = models.CharField(db_index=True, max_length=255) depvcmp = models.CharField(max_length=255) class Meta: db_table = 'package_depends' class Todolist(models.Model): id = models.AutoField(primary_key=True) creator = models.ForeignKey(User) name = models.CharField(max_length=255) description = models.TextField() date_added = models.DateField(auto_now_add=True) objects = TodolistManager() def __unicode__(self): return self.name @property def packages(self): # select_related() does not use LEFT OUTER JOIN for nullable ForeignKey # fields. That is why we need to explicitly list the ones we want. return TodolistPkg.objects.select_related( 'pkg__repo', 'pkg__arch').filter(list=self).order_by('pkg') @property def package_names(self): return '\n'.join(set([p.pkg.pkgname for p in self.packages])) 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 = 'todolist_pkgs' unique_together = (('list','pkg'),) # vim: set ts=4 sw=4 et: