self-hosted/Mailu
1
0
Fork 0
mirror of https://github.com/Mailu/Mailu.git synced 2025-01-14 02:34:22 +02:00
Code Issues Releases Activity

Add a sqlalchemy custom type for unicode to idna conersion of domain names

Browse Source
This commit is contained in:
Stefan Auditor 2018-04-12 19:07:48 +02:00
parent db0cd8efb4
commit 1b666cd25b
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
core/admin/mailu/models.py
View File

@ -38,6 +38,19 @@ class CommaSeparatedList(db.TypeDecorator):
return filter(bool, value.split(","))
class Idna(db.TypeDecorator):
""" Stores a Unicode string in it's IDNA representation (ASCII only)
"""
impl = db.String
def process_bind_param(self, value, dialect):
return value.encode("idna")
def process_result_value(self, value, dialect):
return value.decode("idna")
class Base(db.Model):
""" Base class for all models
"""
@ -54,7 +67,7 @@ class Domain(Base):
"""
__tablename__ = "domain"
name = db.Column(db.String(80), primary_key=True, nullable=False)
name = db.Column(Idna, primary_key=True, nullable=False)
managers = db.relationship('User', secondary=managers,
backref=db.backref('manager_of'), lazy='dynamic')
max_users = db.Column(db.Integer, nullable=False, default=0)