from flask_jwt_extended import current_user
from remixvr.database import (Model, SurrogatePK, db, reference_col,
relationship)
followers_assoc = db.Table(
"followers_assoc",
db.Column("follower", db.Integer, db.ForeignKey("userprofile.user_id")),
db.Column("followed_by", db.Integer, db.ForeignKey("userprofile.user_id")))
class UserProfile(Model, SurrogatePK):
__tablename__ = 'userprofile'
# id is needed for primary join, it does work with SurrogatePK class
id = db.Column(db.Integer, primary_key=True)
user_id = reference_col('users', nullable=False, unique=True)
user = relationship('User', backref=db.backref('profile', uselist=False))
follows = relationship('UserProfile',
secondary=followers_assoc,
primaryjoin=id == followers_assoc.c.follower,
secondaryjoin=id == followers_assoc.c.followed_by,
backref='followed_by',
lazy='dynamic')
def __init__(self, user, **kwargs):
db.Model.__init__(self, user=user, **kwargs)
def is_following(self, profile):
return bool(
# -*- coding: utf-8 -*-
"""Project models."""
import datetime as dt
import uuid
from flask_jwt_extended import current_user
from slugify import slugify
from remixvr.database import (Model, SurrogatePK, relationship, reference_col,
Column, db)
from remixvr.profile.models import UserProfile
favoriter_assoc = db.Table(
"favoritor_assoc",
db.Column("favoriter", db.Integer, db.ForeignKey("userprofile.id")),
db.Column("favorited_project", db.Integer, db.ForeignKey("project.id")))
tag_assoc = db.Table(
"tag_assoc", db.Column("tag", db.Integer, db.ForeignKey("tags.id")),
db.Column("project", db.Integer, db.ForeignKey("project.id")))
class Tags(Model):
__tablename__ = 'tags'
id = db.Column(db.Integer, primary_key=True)
tagname = db.Column(db.String(100))
def __init__(self, tagname):
db.Model.__init__(self, tagname=tagname)