from flask_sqlalchemy import SQLAlchemy
from uuid import uuid4

db = SQLAlchemy()

def get_uuid():
    return uuid4().hex

class User(db.Model):
    __tablename__ = "users"
    id = db.Column(db.String(32), primary_key=True, unique=True, default=get_uuid)
    username = db.Column(db.String(345), unique=True)
    email = db.Column(db.String(345), unique=True)
    password = db.Column(db.Text, nullable=False)

class Dataset(db.Model):
    __tablename__ = "datasets"
    id = db.Column(db.String(32), primary_key=True, unique=True, default=get_uuid)
    name = db.Column(db.String(345), unique=True)
    file_path = db.Column(db.String(345), unique=True)
    model_path = db.Column(db.String(345), unique=True)