added toJSON func

This commit is contained in:
2019-04-02 16:58:59 +02:00
parent 8cf8632c8c
commit cfa12717e0
2 changed files with 17 additions and 2 deletions

View File

@@ -2,6 +2,8 @@
"""
Example user model and related models
"""
import json
from sqlalchemy.orm import relation
from sqlalchemy import MetaData
@@ -347,7 +349,12 @@ class User(UserMixin, db.Model):
followers.c.follower_id == self.id).order_by(Post.timestamp.desc())
def to_dict(self):
return dict(id=self.id, email=self.email, groups=self.groups)
#return self.__dict__
return dict(id=self.id, email=self.email, groups=[g.to_dict() for g in self.groups])
def toJSON(self):
return json.dumps(self.to_dict(), default=lambda o: o.__dict__,
sort_keys=True, indent=4)
def __repr__(self):
return '<User %r>' % self.nickname
@@ -403,3 +410,10 @@ class Group(db.Model):
def __str__(self):
return self.name
def to_dict(self):
return dict(id=self.id, name=self.name)
def toJSON(self):
return json.dumps(self.to_dict(), default=lambda o: o.__dict__,
sort_keys=True, indent=4)