class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(64), index=True, unique=True)
email = db.Column(db.String(120), index=True, unique=True)
password_hash = db.Column(db.String(128))
token = db.Column(db.String(32), index=True, unique=True)
token_expiration = db.Column(db.DateTime)
def __repr__(self):
return '<User {}>'.format(self.username)
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)
def get_token(self, expires_in=3600):
now = datetime.utcnow()
if self.token and self.token_expiration > now + timedelta(seconds=60):
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)
return self.token
def revoke_token(self):
self.token_expiration = datetime.utcnow() - timedelta(seconds=1)
@staticmethod
def check_token(token):
user = User.query.filter_by(token=token).first()
if user is None or user.token_expiration < datetime.utcnow():
return None
return user
class Dataset(db.Model):
id = db.Column(db.String(20), primary_key=True, index=True, unique=True)
database_id = db.Column(db.String(20), db.ForeignKey("database.id"))
size_MB = db.Column(db.Integer, nullable=True)
ipfs = db.Column(db.String(60), nullable=True)
path = db.Column(db.String(340), nullable=True)
name = db.Column(db.String(160), nullable=True)
computed = db.Column(db.DateTime, nullable=True)
class Database(db.Model):
id = db.Column(db.String(20), primary_key=True, index=True, unique=True)
metadata_link = db.Column(db.String(128), nullable=False)
datasets = db.relationship("Dataset", backref="database", lazy="dynamic")
class Task(db.Model):
id = db.Column(db.String(36), primary_key=True)
name = db.Column(db.String(128), index=True)
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
complete = db.Column(db.Boolean, default=False)