from flask_login import UserMixin


class User(UserMixin):
    """Lightweight user object wrapping a DB row dict."""

    def __init__(self, row: dict):
        self.id = row['id']
        self.email = row['email']
        self.password_hash = row['password_hash']
        self.created_at = row.get('created_at', '')

    def get_id(self):
        return str(self.id)