Skip to content
Snippets Groups Projects
Commit 72c20107 authored by Julian's avatar Julian
Browse files

Replace custom code with ldapalchemys 1-n-relationships

parent a721ff1b
Branches
No related tags found
No related merge requests found
Subproject commit 6a0b85f2c6a11951d99d4b4e84b1df992580d964 Subproject commit db135ec7a4940a384e82a4500cfb0e73ecd2f557
...@@ -13,6 +13,7 @@ import crypt ...@@ -13,6 +13,7 @@ import crypt
from flask import request, current_app from flask import request, current_app
from sqlalchemy import Column, Integer, Enum, String, DateTime, Text from sqlalchemy import Column, Integer, Enum, String, DateTime, Text
from ldapalchemy.dbutils import DBRelationship
from uffd.database import db from uffd.database import db
from uffd.user.models import User from uffd.user.models import User
...@@ -29,6 +30,7 @@ class MFAMethod(db.Model): ...@@ -29,6 +30,7 @@ class MFAMethod(db.Model):
created = Column(DateTime()) created = Column(DateTime())
name = Column(String(128)) name = Column(String(128))
dn = Column(String(128)) dn = Column(String(128))
user = DBRelationship('dn', User, backref='mfa_methods')
__mapper_args__ = { __mapper_args__ = {
'polymorphic_on': type, 'polymorphic_on': type,
...@@ -39,14 +41,6 @@ class MFAMethod(db.Model): ...@@ -39,14 +41,6 @@ class MFAMethod(db.Model):
self.name = name self.name = name
self.created = datetime.datetime.now() self.created = datetime.datetime.now()
@property
def user(self):
return User.query.get(self.dn)
@user.setter
def user(self, new_user):
self.dn = new_user.dn
class RecoveryCodeMethod(MFAMethod): class RecoveryCodeMethod(MFAMethod):
code_salt = Column('recovery_salt', String(64)) code_salt = Column('recovery_salt', String(64))
code_hash = Column('recovery_hash', String(256)) code_hash = Column('recovery_hash', String(256))
......
from flask import current_app from flask import current_app
from sqlalchemy import Column, Integer, String, DateTime, Text from sqlalchemy import Column, Integer, String, DateTime, Text
from ldapalchemy.dbutils import DBRelationship
from uffd.database import db from uffd.database import db
from uffd.user.models import User from uffd.user.models import User
...@@ -36,14 +37,7 @@ class OAuth2Grant(db.Model): ...@@ -36,14 +37,7 @@ class OAuth2Grant(db.Model):
id = Column(Integer, primary_key=True) id = Column(Integer, primary_key=True)
user_dn = Column(String(128)) user_dn = Column(String(128))
user = DBRelationship('user_dn', User, backref='oauth2_grants')
@property
def user(self):
return User.query.get(self.user_dn)
@user.setter
def user(self, newuser):
self.user_dn = newuser.dn
client_id = Column(String(40)) client_id = Column(String(40))
...@@ -76,13 +70,7 @@ class OAuth2Token(db.Model): ...@@ -76,13 +70,7 @@ class OAuth2Token(db.Model):
id = Column(Integer, primary_key=True) id = Column(Integer, primary_key=True)
user_dn = Column(String(128)) user_dn = Column(String(128))
@property user = DBRelationship('user_dn', User, backref='oauth2_tokens')
def user(self):
return User.query.get(self.user_dn)
@user.setter
def user(self, newuser):
self.user_dn = newuser.dn
client_id = Column(String(40)) client_id = Column(String(40))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment