34 lines
1 KiB
Python
34 lines
1 KiB
Python
from flask_sqlalchemy import SQLAlchemy
|
|
|
|
db = SQLAlchemy()
|
|
|
|
class MatrixUser(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
matrix_id = db.Column(db.Text, unique=True)
|
|
room_id = db.Column(db.Text)
|
|
pnut_id = db.Column(db.Text, unique=True)
|
|
pnut_token = db.Column(db.Text)
|
|
|
|
def __init__(self, matrix_id, room_id, pnut_id, pnut_token):
|
|
self.matrix_id = matrix_id
|
|
self.room_id = room_id
|
|
self.pnut_id = pnut_id
|
|
self.pnut_token = pnut_token
|
|
|
|
def __repr__(self):
|
|
return '<MatrixUser %r>' % self.matrix_id
|
|
|
|
class MatrixRoom(db.Model):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
room_id = db.Column(db.Text, unique=True)
|
|
pnut_chan = db.Column(db.Text)
|
|
pnut_since = db.Column(db.Text)
|
|
pnut_write = db.Column(db.Boolean, default=True)
|
|
|
|
def __init__(self, room_id, pnut_chan, pnut_write=True):
|
|
self.room_id = room_id
|
|
self.pnut_chan = pnut_chan
|
|
self.pnut_write = pnut_write
|
|
|
|
def __repr__(self):
|
|
return '<MatrixRoom %r>' % self.room_id
|