Files
postgres-web/pgweb/core/models.py
2019-01-26 16:19:26 +01:00

218 lines
8.4 KiB
Python

from django.db import models
from django.core.validators import ValidationError
from django.contrib.auth.models import User
from pgweb.util.misc import varnish_purge
import base64
TESTING_CHOICES = (
(0, 'Release'),
(1, 'Release candidate'),
(2, 'Beta'),
(3, 'Alpha'),
)
TESTING_SHORTSTRING = ('', 'rc', 'beta', 'alpha')
class Version(models.Model):
tree = models.DecimalField(max_digits=3, decimal_places=1, null=False, blank=False, unique=True)
latestminor = models.IntegerField(null=False, blank=False, default=0, help_text="For testing versions, latestminor means latest beta/rc number. For other releases, it's the latest minor release number in the tree.")
reldate = models.DateField(null=False, blank=False)
relnotes = models.CharField(max_length=32, null=False, blank=False)
current = models.BooleanField(null=False, blank=False, default=False)
supported = models.BooleanField(null=False, blank=False, default=True)
testing = models.IntegerField(null=False, blank=False, default=0, help_text="Testing level of this release. latestminor indicates beta/rc number", choices=TESTING_CHOICES)
docsloaded = models.DateTimeField(null=True, blank=True, help_text="The timestamp of the latest docs load. Used to control indexing and info on developer docs.")
firstreldate = models.DateField(null=False, blank=False, help_text="The date of the .0 release in this tree")
eoldate = models.DateField(null=False, blank=False, help_text="The final release date for this tree")
def __str__(self):
return self.versionstring
@property
def versionstring(self):
return self.buildversionstring(self.latestminor)
@property
def numtree(self):
# Return the proper numeric tree version, taking into account that PostgreSQL 10
# changed from x.y to x for major version.
if self.tree >= 10:
return int(self.tree)
else:
return self.tree
def buildversionstring(self, minor):
if not self.testing:
return "%s.%s" % (self.numtree, minor)
else:
return "%s%s%s" % (self.numtree, TESTING_SHORTSTRING[self.testing], minor)
@property
def treestring(self):
if not self.testing:
return "%s" % self.numtree
else:
return "%s %s" % (self.numtree, TESTING_SHORTSTRING[self.testing])
def save(self):
# Make sure only one version at a time can be the current one.
# (there may be some small race conditions here, but the likelyhood
# that two admins are editing the version list at the same time...)
if self.current:
previous = Version.objects.filter(current=True)
for p in previous:
if not p == self:
p.current = False
p.save() # primary key check avoids recursion
# Now that we've made any previously current ones non-current, we are
# free to save this one.
super(Version, self).save()
class Meta:
ordering = ('-tree', )
def purge_urls(self):
yield '/$'
yield '/support/versioning'
yield '/support/security'
yield '/docs/$'
yield '/docs/manuals'
yield '/about/featurematrix/$'
yield '/versions.rss'
class Country(models.Model):
name = models.CharField(max_length=100, null=False, blank=False)
tld = models.CharField(max_length=3, null=False, blank=False)
class Meta:
db_table = 'countries'
ordering = ('name',)
verbose_name = 'Country'
verbose_name_plural = 'Countries'
def __str__(self):
return self.name
class Language(models.Model):
# Import data from http://www.loc.gov/standards/iso639-2/ISO-639-2_utf-8.txt
# (yes, there is a UTF16 BOM in the UTF8 file)
# (and yes, there is a 7 length value in a field specified as 3 chars)
alpha3 = models.CharField(max_length=7, null=False, blank=False, primary_key=True)
alpha3term = models.CharField(max_length=3, null=False, blank=True)
alpha2 = models.CharField(max_length=2, null=False, blank=True)
name = models.CharField(max_length=100, null=False, blank=False)
frenchname = models.CharField(max_length=100, null=False, blank=False)
class Meta:
ordering = ('name', )
def __str__(self):
return self.name
class OrganisationType(models.Model):
typename = models.CharField(max_length=32, null=False, blank=False)
def __str__(self):
return self.typename
class Organisation(models.Model):
name = models.CharField(max_length=100, null=False, blank=False, unique=True)
approved = models.BooleanField(null=False, default=False)
address = models.TextField(null=False, blank=True)
url = models.URLField(null=False, blank=False)
email = models.EmailField(null=False, blank=True)
phone = models.CharField(max_length=100, null=False, blank=True)
orgtype = models.ForeignKey(OrganisationType, null=False, blank=False, verbose_name="Organisation type")
managers = models.ManyToManyField(User, blank=False)
lastconfirmed = models.DateTimeField(null=False, blank=False, auto_now_add=True)
send_notification = True
send_m2m_notification = True
def __str__(self):
return self.name
class Meta:
ordering = ('name',)
# Basic classes for importing external RSS feeds, such as planet
class ImportedRSSFeed(models.Model):
internalname = models.CharField(max_length=32, null=False, blank=False, unique=True)
url = models.URLField(null=False, blank=False)
purgepattern = models.CharField(max_length=512, null=False, blank=True, help_text="NOTE! Pattern will be automatically anchored with ^ at the beginning, but you must lead with a slash in most cases - and don't forget to include the trailing $ in most cases")
def purge_related(self):
if self.purgepattern:
varnish_purge(self.purgepattern)
def __str__(self):
return self.internalname
class ImportedRSSItem(models.Model):
feed = models.ForeignKey(ImportedRSSFeed)
title = models.CharField(max_length=100, null=False, blank=False)
url = models.URLField(null=False, blank=False)
posttime = models.DateTimeField(null=False, blank=False)
def __str__(self):
return self.title
@property
def date(self):
return self.posttime.strftime("%Y-%m-%d")
# From man sshd, except for ssh-dss
_valid_keytypes = ['ecdsa-sha2-nistp256', 'ecdsa-sha2-nistp384', 'ecdsa-sha2-nistp521', 'ssh-ed25519', 'ssh-rsa']
# Options, keytype, key, comment. But we don't support options.
def validate_sshkey(key):
lines = key.splitlines()
for k in lines:
pieces = k.split()
if len(pieces) == 0:
raise ValidationError("Empty keys are not allowed")
if len(pieces) > 3:
raise ValidationError('Paste each ssh key without options, e.g. "ssh-rsa AAAAbbbcc mykey@machine"')
if pieces[0] == 'ssh-dss':
raise ValidationError("For security reasons, ssh-dss keys are not supported")
if pieces[0] not in _valid_keytypes:
raise ValidationError("Only keys of types {0} are supported, not {1}.".format(", ".join(_valid_keytypes), pieces[0]))
try:
base64.b64decode(pieces[1])
except:
raise ValidationError("Incorrect base64 encoded key!")
# Extra attributes for users (if they have them)
class UserProfile(models.Model):
user = models.OneToOneField(User, null=False, blank=False, primary_key=True)
sshkey = models.TextField(null=False, blank=True, verbose_name="SSH key", help_text="Paste one or more public keys in OpenSSH format, one per line.", validators=[validate_sshkey, ])
lastmodified = models.DateTimeField(null=False, blank=False, auto_now=True)
# Notifications sent for any moderated content.
# Yes, we uglify it by storing the type of object as a string, so we don't
# end up with a bazillion fields being foreign keys. Ugly, but works.
class ModerationNotification(models.Model):
objectid = models.IntegerField(null=False, blank=False, db_index=True)
objecttype = models.CharField(null=False, blank=False, max_length=100)
text = models.TextField(null=False, blank=False)
author = models.CharField(null=False, blank=False, max_length=100)
date = models.DateTimeField(null=False, blank=False, auto_now=True)
def __str__(self):
return "%s id %s (%s): %s" % (self.objecttype, self.objectid, self.date, self.text[:50])
class Meta:
ordering = ('-date', )