class Reference(db.Model):
__tablename__ = 'rule_references'
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.String(140))
rule_id = db.Column(db.Integer, db.ForeignKey('rules.id'))
class DeployScript(db.Model, APIModel):
all_fields = {
'script': {'required': True, 'editable': True},
'name': {'required': True, 'editable': True},
'date': {'required': False, 'editable': False},
'notes': {'required': True, 'editable': True},
}
__tablename__ = 'deploy_scripts'
id = db.Column(db.Integer, primary_key=True)
script = db.Column(db.String(102400))
date = db.Column(
db.DateTime(), default=datetime.utcnow)
notes = db.Column(db.String(140))
name = db.Column(db.String(140))
user_id = db.Column(db.Integer, db.ForeignKey(User.id))
user = db.relationship(User, uselist=False)
def __init__(self, name=None, script=None, notes=None):
self.name = name
self.script = script
self.notes = notes
def __repr__(self):
return '<DeployScript>{}'.format(self.to_dict())
def to_dict(self):
return dict(script=self.script, date=self.date, notes=self.notes,
user=self.user.email, id=self.id)
class ApiKey(db.Model):
all_fields = {
'api_key': {'required': True, 'editable': False},
'user_id': {'required': True, 'editable': False}
}
id = db.Column(db.Integer, primary_key=True)
api_key = db.Column(db.String(32), unique=True)
user_id = db.Column('user_id', db.Integer, db.ForeignKey("user.id"), nullable=False)
class PasswdReset(db.Model):
id = db.Column(db.Integer, primary_key=True)
hashstr = db.Column(db.String(40))
created = db.Column(db.DateTime(), default=datetime.utcnow)
active = db.Column(db.Boolean())
user_id = db.Column(db.Integer, db.ForeignKey(User.id))
user = db.relationship(User, uselist=False)
@property
def email_body(self):
from mhn import mhn
return render_template(
'auth/reset-email.html', hashstr=self.hashstr,
server_url=mhn.config['SERVER_BASE_URL'],
email=self.user.email)
from datetime import datetime
from flask_security import UserMixin, RoleMixin
from flask import render_template
from mhn import db
from mhn.api import APIModel
roles_users = db.Table(
'roles_users', db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))
class Role(db.Model, RoleMixin):
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(80), unique=True)
description = db.Column(db.String(255))
class User(db.Model, APIModel, UserMixin):
all_fields = {
'email': {'required': True, 'editable': False},
'password': {'required': True, 'editable': True},
'active': {'required': False, 'editable': True}
}
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(255), unique=True)
password = db.Column(db.String(255))
active = db.Column(db.Boolean())