django-orchestra/orchestra/contrib/databases/models.py

95 lines
3.2 KiB
Python
Raw Permalink Normal View History

2014-05-08 16:59:35 +00:00
import hashlib
from django.db import models
from django.utils.translation import gettext_lazy as _
2014-05-08 16:59:35 +00:00
from orchestra.core import validators
2014-05-08 16:59:35 +00:00
from . import settings
class Database(models.Model):
""" Represents a basic database for a web application """
MYSQL = 'mysql'
POSTGRESQL = 'postgresql'
2014-10-09 17:04:12 +00:00
name = models.CharField(_("name"), max_length=64, # MySQL limit
2014-05-08 16:59:35 +00:00
validators=[validators.validate_name])
2015-04-21 13:12:48 +00:00
users = models.ManyToManyField('databases.DatabaseUser', blank=True,
verbose_name=_("users"),related_name='databases')
2014-05-08 16:59:35 +00:00
type = models.CharField(_("type"), max_length=32,
choices=settings.DATABASES_TYPE_CHOICES,
default=settings.DATABASES_DEFAULT_TYPE)
account = models.ForeignKey('accounts.Account', on_delete=models.CASCADE,
verbose_name=_("Account"), related_name='databases')
comments = models.TextField(default="", blank=True)
target_server = models.ForeignKey('orchestration.Server', on_delete=models.CASCADE,
verbose_name=_("Server"), default=3 )
2014-05-08 16:59:35 +00:00
class Meta:
unique_together = ('name', 'type', 'target_server')
2015-04-02 16:14:55 +00:00
def __str__(self):
2014-05-08 16:59:35 +00:00
return "%s" % self.name
2014-05-08 16:59:35 +00:00
@property
def owner(self):
""" database owner is the first user related to it """
# Accessing intermediary model to get which is the first user
users = Database.users.through.objects.filter(database_id=self.id)
2015-04-21 13:12:48 +00:00
user = users.order_by('id').first()
if user is not None:
return user.databaseuser
return None
2016-04-06 19:00:16 +00:00
@property
def active(self):
return self.account.is_active
2014-05-08 16:59:35 +00:00
2015-03-04 21:06:16 +00:00
Database.users.through._meta.unique_together = (
('database', 'databaseuser'),
)
2014-05-08 16:59:35 +00:00
class DatabaseUser(models.Model):
2014-10-17 10:04:47 +00:00
MYSQL = Database.MYSQL
POSTGRESQL = Database.POSTGRESQL
typeOfPermision = [
('all','all'),
('ro', 'read only'),
]
username = models.CharField(_("username"), max_length=32, # MySQL usernames 16 char long
2014-05-08 16:59:35 +00:00
validators=[validators.validate_name])
2014-10-09 17:04:12 +00:00
password = models.CharField(_("password"), max_length=256)
2014-05-08 16:59:35 +00:00
type = models.CharField(_("type"), max_length=32,
choices=settings.DATABASES_TYPE_CHOICES,
default=settings.DATABASES_DEFAULT_TYPE)
account = models.ForeignKey('accounts.Account', on_delete=models.CASCADE,
verbose_name=_("Account"), related_name='databaseusers')
target_server = models.ForeignKey('orchestration.Server', on_delete=models.CASCADE,
verbose_name=_("Server"), default=3 )
permision = models.CharField(verbose_name=_("Permisson"), max_length=20, choices=typeOfPermision, default='all')
2014-05-08 16:59:35 +00:00
class Meta:
verbose_name_plural = _("DB users")
unique_together = ('username', 'type', 'target_server')
2015-04-02 16:14:55 +00:00
def __str__(self):
2014-05-08 16:59:35 +00:00
return self.username
2014-05-08 16:59:35 +00:00
def get_username(self):
return self.username
2014-05-08 16:59:35 +00:00
def set_password(self, password):
if self.type == self.MYSQL:
# MySQL stores sha1(sha1(password).binary).hex
2015-04-02 16:14:55 +00:00
binary = hashlib.sha1(password.encode('utf-8')).digest()
2014-05-08 16:59:35 +00:00
hexdigest = hashlib.sha1(binary).hexdigest()
2014-10-09 17:04:12 +00:00
self.password = '*%s' % hexdigest.upper()
2014-05-08 16:59:35 +00:00
else:
raise TypeError("Database type '%s' not supported" % self.type)