"""
Asset Location Model.

This Asset Location model uses sqlalchemy
"""

from sqlalchemy import Column
from sqlalchemy import Enum
from sqlalchemy import Integer

from assets.connectors import mysql
from assets.constants import asset_legacy


class AssetLocation(mysql.DdModel):
    """Table defination for asset_location table."""

    __tablename__ = 'asset_location'

    asset_location_id = Column(Integer, primary_key=True)
    asset_id = Column(Integer)
    storage_drive_id = Column(Integer)
    status = Column(Enum(*asset_legacy.LOCATION_STATUSES))