23 lines
1.0 KiB
Python
23 lines
1.0 KiB
Python
|
from database import db_connection
|
||
|
|
|
||
|
|
class Organization(db_connection.Model):
|
||
|
|
__tablename__ = 'organizations'
|
||
|
|
|
||
|
|
id = db_connection.Column(db_connection.Integer, primary_key=True, index=True)
|
||
|
|
name = db_connection.Column(db_connection.String, unique=True, index=True, nullable=False)
|
||
|
|
users = db_connection.Column(db_connection.JSON, nullable=False, default=dict)
|
||
|
|
users_id = db_connection.Column(db_connection.Integer, db_connection.ForeignKey('users.id'))
|
||
|
|
manager = db_connection.relationship('User', backref=db_connection.backref('owned_organization', uselist=False))
|
||
|
|
|
||
|
|
def to_dict(self):
|
||
|
|
return {
|
||
|
|
"id": self.id,
|
||
|
|
"name": self.name,
|
||
|
|
"manager": self.manager.to_dict(),
|
||
|
|
"users": [{"id": user_id, "user_data": {
|
||
|
|
"username": user_data["username"],
|
||
|
|
"full_name": user_data["full_name"],
|
||
|
|
"email": user_data["email"],
|
||
|
|
"status": user_data["status"]
|
||
|
|
}} for user_id, user_data in self.users.items()],
|
||
|
|
}
|