from sqlalchemy import Column, String, Integer, ForeignKey
from sqlalchemy.orm import relationship
from crudkit.core.base import CRUDMixin
from test_app.db import Base

class User(CRUDMixin, Base):
    __tablename__ = 'users'
    name = Column(String)
    email = Column(String)
    supervisor_id = Column(Integer, ForeignKey('users.id'))
    supervisor = relationship('User', remote_side='User.id')

class Device(CRUDMixin, Base):
    __tablename__ = 'devices'
    name = Column(String)
    serial = Column(String)
    assigned_to_id = Column(Integer, ForeignKey('users.id'))
    assigned_to = relationship('User')