class User(db.Model, BaseModelMixin):
__tablename__ = 'user'
username = db.Column(db.String(200), unique=True, index=True)
name = db.Column(db.String())
email = db.Column(db.String(), unique=True, index=True)
password_hash = db.Column(db.String(128))
avatar_url = db.Column(db.String())
github_nb_followers = db.Column(db.Integer())
github_nb_following = db.Column(db.Integer())
github_access_token = db.Column(db.String(200))
pivots = db.relationship('Pivot', lazy='dynamic')
def __init__(self, username=None, email=None, github_access_token=None):
self.username = username
self.email = email
self.github_access_token = github_access_token
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
@property
def nb_pivots(self):
"""
Returns the number of pivot the user manages
"""
return db.session.query(Pivot).filter_by(created_by=self.id,
deleted=False).count()
def can_create_more_pivot(self):
"""
This method returns False if the user has reached
the maximum number of pivot he is allowed to
"""
return self.nb_pivots < 5
def avatar_url(self, size):
"""
The URL of the avatar for the given image size
"""
try:
size = int(size)
except Exception:
raise ValueError('Invalid size')
digest = md5(self.email.lower().encode('utf-8')).hexdigest()
return 'https://www.gravatar.com/avatar/{}?d=retro&s={}&r=pg'.format(
digest, size)
class Hook(db.Model):
__tablename__ = 'hook'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer())
repo_id = db.Column(db.Integer())
repo_name = db.Column(db.String())
repo_owner = db.Column(db.String())
subscribe_uuid = db.Column(db.String())
github_hook_secret = db.Column(db.String())
github_hook_id = db.Column(db.Integer())
def __init__(self, user_id, repo_id, repo_name, repo_owner, subscribe_uuid,
github_hook_secret, github_hook_id):
self.user_id = user_id
self.repo_id = repo_id
self.repo_name = repo_name
self.repo_owner = repo_owner
self.subscribe_uuid = subscribe_uuid
self.github_hook_secret = github_hook_secret
self.github_hook_id = github_hook_id
class Consumer(db.Model, BaseModelMixin, NchanChannelModelMixin):
__tablename__ = 'consumer'
pivot_id = db.Column(db.Integer(), db.ForeignKey('pivot.id'))
url_path = db.Column(db.String(), nullable=False)
ctype = db.Column(db.String(), nullable=False)
def __init__(self):
self.uuid = str(uuid.uuid4())
@property
def root_url(self):
return '{}/c'.format(Config.PUBLIC_ROOT_URL)
@property
def url(self):
return '{}/c/{}'.format(Config.PUBLIC_ROOT_URL, self.uuid)
class Producer(db.Model, BaseModelMixin, NchanChannelModelMixin):
__tablename__ = 'producer'
pivot_id = db.Column(db.Integer(), db.ForeignKey('pivot.id'))
url_path = db.Column(db.String(), nullable=False)
ptype = db.Column(db.String(), nullable=False)
def __init__(self):
self.uuid = str(uuid.uuid4())
def is_github(self):
return self.ptype == 'github'
@property
def url(self):
return '{}/p/{}'.format(Config.PUBLIC_ROOT_URL, self.uuid)
@property
def github_secret(self):
return 'my-super-secret'
class Pivot(db.Model, BaseModelMixin, NchanChannelModelMixin):
__tablename__ = 'pivot'
producers = db.relationship('Producer', lazy='dynamic', backref='pivot')
consumers = db.relationship('Consumer', lazy='dynamic', backref='pivot')
created_by = db.Column(db.Integer(), db.ForeignKey('user.id'))
def __init__(self):
self.uuid = str(uuid.uuid4())
def can_have_more_producer(self):
"""
This method returns False if the pivot has reached
the maximum number of producer it is allowed to
"""
return self.nb_producers < 5
def can_have_more_consumer(self):
"""
This method returns False if the pivot has reached
the maximum number of consumer it is allowed to
"""
return self.nb_consumers < 5
@property
def nb_producers(self):
return db.session.query(Producer).filter_by(pivot_id=self.id).count()
@property
def nb_consumers(self):
return db.session.query(Consumer).filter_by(pivot_id=self.id).count()
def get_last_webcalls(self):
return WebCall.query.filter_by(published_on=self.uuid).order_by(
WebCall.published_at.desc()).limit(10).all()