class AccessGroup(db.Model):
__tablename__ = 'access_group'
id = db.Column(db.Integer, primary_key=True)
description = db.Column(db.String(15), unique=True)
users = db.relationship('User', backref='user_group', lazy='dynamic')
def __repr__(self):
return "<Group (name={})".format(self.description)
class Logs(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
origin = db.Column(db.String(20))
operation = db.Column(db.String(10))
timestamp = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
def __init__(self, operation, user_id, origin):
self.operation = operation
self.user_id = user_id
self.origin = origin
def __repr__(self):
return "<Log date: {} operation: {} User: {}".format(
self.timestamp, self.operation, self.user_id)
class EventLog(db.Model):
id = db.Column(db.Integer, primary_key=True)
ob_id = db.Column(db.Integer)
date_added = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
ob_type = db.Column(db.String(30))
obs = db.Column(db.String(50))
def save_to_db(self):
try:
db.session.add(self)
db.session.commit()
except:
raise
db.session.rollback()
finally:
pass
def delete_from_db(self):
db.session.delete(self)
db.session.commit()
class User(PaginatedAPIMixin, UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(15), unique=True)
email = db.Column(db.String(50), unique=True)
password = db.Column(db.String(80))
isadmin = db.Column(db.Boolean)
createdAt = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
updatedAt = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow,
onupdate=datetime.utcnow)
lastlogin = db.Column(db.DateTime)
isActivated = db.Column(db.Boolean, default=False)
acess_group = db.Column(db.Integer, db.ForeignKey('access_group.id'))
deleted_on = db.Column(db.DateTime, default=datetime.utcnow)
#user resource
token = db.Column(db.String(32), index=True, unique=True)
token_expiration = db.Column(db.DateTime)
logs = db.relationship('Logs', backref='log_history', lazy='dynamic')
def __repr__(self):
return "<User(name='%s', createdAt='%s')>" % (self.username,
self.createdAt)
def to_dict(self, include_sensitivedata=False):
data = {
'id': self.id,
'username': self.username,
'lastlogin': self.lastlogin,
'last_updated': self.updatedAt,
'logs': self.logs.count(),
'programs': 0,
'_links': {
'self':
'https://hswebapp.com' + url_for('apiv0.get_user', id=self.id),
'obs':
'add more links'
}
}
if include_sensitivedata:
data['email'] = self.email
return data
def from_dict(self, data, new_user=False):
for field in ['username', 'email']:
if field in data:
setattr(self, field, data[field])
if new_user and 'password' in data:
self.set_password(data['password'])
def set_password(self, new_password):
self.password = generate_password_hash(new_password, method='sha256')
def check_password(self, check_password):
return check_password_hash(self.password, check_password)
def get_token(self, expires_in=3610):
now = datetime.utcnow()
if self.token and self.token_expiration > (now +
timedelta(seconds=30)):
return self.token
self.token = base64.b64encode(os.urandom(24)).decode('utf-8')
self.token_expiration = now + timedelta(seconds=expires_in)
db.session.add(self)
db.session.commit()
return self.token
def revoke_token(self):
self.token_expiration = datetime.utcnow() - timedelta(seconds=1)
db.session.commit()
def check_token_is_valid(self):
return True if self.token_expiration > datetime.utcnow() else False
def check_token_duration(self):
return self.token_expiration - datetime.utcnow()
@staticmethod
def check_token(token):
print(token)
user = User.query.filter_by(token=token).first()
if user is None or user.token_expiration < datetime.utcnow():
return None
return user
class TempLog(db.Model):
__tablename__ = 'templog'
id = db.Column(db.Integer, primary_key=True)
sensor = db.Column(db.String(30))
rdate = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
sensorLocation = db.Column(db.String(15))
value = db.Column(db.Float(precision=2))
sensorType = db.Column(db.String(15))
exported = db.Column(db.DateTime)
def __init__(self, sensor, rdate, sensorLocation, sensorType, value):
self.sensor = sensor
self.rdate = rdate
self.sensorLocation = sensorLocation
self.sensorType = sensorType
self.value = value
def __repr__(self):
return '<TempLog {}>'.format(self.sensorType)
def json(self):
return {
'data': self.data,
'sensorLocation': self.sensorLocation,
'value': self.value,
'sensorType': self.sensorType
}
@classmethod
def get_max_value(cls, sensorType):
return cls.query.filter_by(sensorType=sensorType).order_by(
cls.value.desc()).first()
@classmethod
def get_min_value(cls, sensorType):
return cls.query.filter_by(sensorType=sensorType).order_by(
cls.value.asc()).first()
def save_to_db(self):
try:
db.session.add(self)
db.session.commit()
model_saved.send(self, message=self)
except:
raise
db.session.rollback()
finally:
pass
def close_session(self):
db.session.close()
def delete_from_db(self):
db.session.delete(self)
db.session.commit()
def close_session(self):
db.session.close()
def get_temperature(self):
return str(self.value) + 'C'
def get_type(self):
return type(self).__name__
def get_id(self):
return self.id