)
users_cameras_relationship = db.Table('users_cameras_relationship',
db.Column('user_id', db.Integer, db.ForeignKey('users.id'), nullable=False),
db.Column('camera_id', db.Integer, db.ForeignKey('cameras.id'), nullable=False),
db.PrimaryKeyConstraint('user_id', 'camera_id')
)
class UsersCoursesRelationship():
def __init__(self, user_id, course_id):
self.user_id = user_id
self.course_id = course_id
db.mapper(UsersCoursesRelationship, users_courses_relationship)
class UsersCamerasRelationship():
def __init__(self, user_id, camera_id):
self.user_id = user_id
self.camera_id = camera_id
db.mapper(UsersCamerasRelationship, users_cameras_relationship)
class User(UserMixin, db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
social_id = db.Column(db.String(64), nullable=False) #ids for oAuth
nickname = db.Column(db.String(64), nullable=False)
def __init__(self, value, type, tlp, tag, source, added_on):
self.value = value
self.type = type
self.tlp = tlp
self.tag = tag
self.source = source
self.added_on = added_on
class Whitelist(db.Model):
def __init__(self, element, type, source, added_on):
self.element = element
self.type = type
self.source = source
self.added_on = added_on
class MISPInst(db.Model):
def __init__(self, name, url, key, ssl, added_on, last_sync):
self.name = name
self.url = url
self.apikey = key
self.verifycert = ssl
self.added_on = added_on
self.last_sync = last_sync
db.mapper(Whitelist, db.Table('whitelist', db.metadata, autoload=True))
db.mapper(Ioc, db.Table('iocs', db.metadata, autoload=True))
db.mapper(MISPInst, db.Table('misp', db.metadata, autoload=True))
return session_commit()
term_relationships=db.Table('term_relationships',
db.Column('post_id', db.Integer,db.ForeignKey('posts.id'), nullable=False),
db.Column('term_id',db.Integer,db.ForeignKey('terms.id'),nullable=False),
db.PrimaryKeyConstraint('post_id', 'term_id')
)
class TermRelationships():
def __init__(self,post_id,term_id):
self.post_id=post_id
self.term_id=term_id
db.mapper(TermRelationships, term_relationships)
class Posts(db.Model):
id = db.Column(db.Integer, primary_key=True)
author = db.Column(db.Integer, db.ForeignKey('users.id'))
title = db.Column(db.String(255),nullable=False)
slug = db.Column(db.String(255), nullable=False)
created_on=db.Column(db.TIMESTAMP,server_default=db.func.current_timestamp(), nullable=False)
content = db.Column(db.Text)
status = db.Column(db.String(20), server_default='publish', nullable=False)
#Need to make not null
modified = db.Column(db.TIMESTAMP,server_default=db.func.current_timestamp(),nullable=False)
post_type=db.Column(db.String(20), server_default='post', nullable=False)
terms=db.relationship('Terms', secondary=term_relationships, backref='posts' )
comments = db.relationship('Comments', backref="post", cascade="all, delete-orphan" , lazy='dynamic')
db.ForeignKey('courses.id'),
nullable=False),
db.Column('module_id',
db.Integer,
db.ForeignKey('modules.id'),
nullable=False), db.PrimaryKeyConstraint('course_id',
'module_id'))
class CompulsoryModules():
def __init__(self, course_id, module_id):
self.course_id = course_id
self.module_id = module_id
db.mapper(CompulsoryModules, compulsory_modules)
optional_modules = db.Table(
'optional_modules',
db.Column('course_id',
db.Integer,
db.ForeignKey('courses.id'),
nullable=False),
db.Column('module_id',
db.Integer,
db.ForeignKey('modules.id'),
nullable=False), db.PrimaryKeyConstraint('course_id',
'module_id'))
class OptionalModules():
holiday_members = db.Table(
"holiday_members",
db.metadata,
db.Column("id", db.Integer, primary_key=True),
db.Column("user_id", db.Integer, db.ForeignKey("users.id")),
db.Column("holiday_id", db.Integer, db.ForeignKey("holidays.id")),
)
# unique index of hippie_id and dog_id
db.Index("member",
holiday_members.c.user_id,
holiday_members.c.holiday_id,
unique=True)
db.mapper(Holiday_Members, holiday_members)
def create_user(new_name, new_email, new_password):
hashPass = password = bcrypt.hashpw(new_password.encode('utf8'),
bcrypt.gensalt())
user = Users(username=new_name,
email=new_email,
password=hashPass.decode('utf-8'))
db.session.add(user)
db.session.commit()
return user
def auth_user(new_name, new_password):
user = Users.query.filter_by(username=new_name).first()
term_relationships = db.Table(
'term_relationships',
db.Column('post_id', db.Integer, db.ForeignKey('posts.id'),
nullable=False),
db.Column('term_id', db.Integer, db.ForeignKey('terms.id'),
nullable=False), db.PrimaryKeyConstraint('post_id', 'term_id'))
class TermRelationships():
def __init__(self, post_id, term_id):
self.post_id = post_id
self.term_id = term_id
db.mapper(TermRelationships, term_relationships)
class Posts(db.Model):
id = db.Column(db.Integer, primary_key=True)
author = db.Column(db.Integer, db.ForeignKey('users.id'))
title = db.Column(db.String(255), nullable=False)
slug = db.Column(db.String(255), nullable=False)
created_on = db.Column(db.TIMESTAMP,
server_default=db.func.current_timestamp(),
nullable=False)
content = db.Column(db.Text)
status = db.Column(db.String(20), server_default='publish', nullable=False)
#Need to make not null
modified = db.Column(db.TIMESTAMP,
server_default=db.func.current_timestamp(),
return state
#this is very important to jsonpickle.
def __setstate__(self, state):
self.__dict__.update(state)
def toJson(self):
data = jsonify(number=self.number,
message=str(self.message),
name=self.name,
comment=self.comment,
company=self.company.name)
return data.get_data(as_text=True)
#return jsonpickle.encode(self, unpicklable=False)
class Company(db.Model):
__tablename__ = 'Company'
__name__ = 'Company'
idCompany = db.Column(db.Integer, primary_key=True,server_default=text("nextval('seqproject'::regclass)"))
name = db.Column(db.String(30))
def __init__(self, name):
self.name = name
def toJson(self):
return '{"idCompany" :"'+ str(self.idCompany) +'",\n "name" : "'+self.name+'"}\n'
#imporatnt to map the relationship.
db.mapper(UserHasProject,t_UserHasProject)
'users_cameras_relationship',
db.Column('user_id', db.Integer, db.ForeignKey('users.id'),
nullable=False),
db.Column('camera_id',
db.Integer,
db.ForeignKey('cameras.id'),
nullable=False), db.PrimaryKeyConstraint('user_id', 'camera_id'))
class UsersCoursesRelationship():
def __init__(self, user_id, course_id):
self.user_id = user_id
self.course_id = course_id
db.mapper(UsersCoursesRelationship, users_courses_relationship)
class UsersCamerasRelationship():
def __init__(self, user_id, camera_id):
self.user_id = user_id
self.camera_id = camera_id
db.mapper(UsersCamerasRelationship, users_cameras_relationship)
class User(UserMixin, db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
social_id = db.Column(db.String(64), nullable=False) #ids for oAuth
class Roletopermission(object):
"""
Roletopermission object the "roletopermission" table.
"""
def __init__(self, role_id, permission_id):
self.role_id = role_id
self.permission_id = permission_id
roletopermission = db.Table("roletopermission",
db.metadata,
db.Column("id", db.Integer, primary_key=True),
db.Column("role_id", db.Integer, db.ForeignKey("role.id")),
db.Column("permission_id", db.Integer, db.ForeignKey("permission.id")),
)
db.mapper(Roletopermission, roletopermission)
db.Index("roletopermission_index", roletopermission.c.role_id, roletopermission.c.permission_id, unique=True)
class Role(db.Model):
__tablename__ = 'role'
id = db.Column(db.Integer, primary_key=True)
rolename = db.Column(db.String(64), index=True, unique=True)
permission = db.relationship("Permission",
secondary=roletopermission,
backref=db.backref("Role", lazy="dynamic"),)
class Usertorole(object):
"""
Dogs object the "dogs" table.
"""
def __init__(self, role_id, user_id):