class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(60), nullable=False)
messages = db.relationship('ChatSchema', backref='author', lazy=True)
def __repr__(self):
return f"User('{self.username}', '{self.email}')"
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(16), index=True, unique=True)
password_hash = db.Column(db.String(128))
public_key = db.Column(db.String(1024))
online = db.Column(db.Boolean, default=False)
def __repr__(self):
return '<User {}>'.format(self.username)
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
sender_id = db.Column(db.Integer, db.ForeignKey('user.id'))
receiver_id = db.Column(db.Integer, db.ForeignKey('user.id'))
post_time = db.Column(db.DateTime, index=True)
content = db.Column(db.String(512), index=True)
def __repr__(self):
return '<Post {}>'.format(self.body)
class ChatRoom(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(20), unique=True, nullable=False)
password = db.Column(db.String(60), nullable=False)
messages = db.relationship('ChatSchema', backref='msgs', lazy=True)
# add user relationship
# users = db.relationship('User', backref='users', lazy=True)
def __repr__(self):
return f"ChatRoom('{self.id}', '{self.name}')"
class ChatSchema(db.Model):
SID = db.Column(db.Integer, primary_key=True)
message = db.Column(db.Text, nullable=False)
time_sent = db.Column(db.DateTime, nullable=False, default=datetime.now)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
room_id = db.Column(db.Integer,
db.ForeignKey('chat_room.id'),
nullable=False)
def __repr__(self):
return f"ChatSchema('{self.message}', '{self.time_sent}')"
class Users(db.Model,UserMixin):
id= db.Column(db.Integer,primary_key=True)
firstname=db.Column(db.String(20),nullable=False)
lastname=db.Column(db.String(20),nullable=False)
username=db.Column(db.String(20),unique=True,nullable=False)
email=db.Column(db.String(120),unique=True,nullable=False)
password=db.Column(db.String(20),nullable=False)
def __repr__(self):
return f"Users('{self.firstname}','{self.lastname}','{self.username}','{self.email}')"