site_tps/settings/models.py
2018-01-20 23:08:46 +01:00

47 lines
1.2 KiB
Python

from django.db import models
import binascii
from site_tps import qaes
from django.conf import settings
class AESEncryptedField(models.CharField):
def save_form_data(self, instance, data):
setattr(instance, self.name,
binascii.b2a_base64(qaes.encrypt(settings.AES_KEY, data)))
def value_from_object(self, obj):
return qaes.decrypt(settings.AES_KEY,
binascii.a2b_base64(getattr(obj, self.attname)))
class ContentSettings(models.Model):
ftp_url = models.URLField(
max_length=255,
verbose_name="URL du FTP",
default="",
)
ftp_id = models.CharField(
max_length=255,
verbose_name="Identifiant sur le FTP",
default=""
)
ftp_pass = AESEncryptedField(
max_length=255,
verbose_name="Mot de passe",
default=""
)
class SiteSettings(models.Model):
allow_upload = models.BooleanField(
verbose_name="Autoriser l'upload de vidéos.",
default=False,
)
home_message = models.TextField(
verbose_name="Message de la page d'accueil",
default=""
)
@classmethod
def get_settings(cls):
return cls.objects.get_or_create()[0]