"""
Storage Drive model.

This storage drive model uses sqlalchemy.
It's used to store assets storage drives info in DD database
"""

from sqlalchemy import Column
from sqlalchemy import Integer

from assets.connectors import mysql


class StorageDrive(mysql.DdModel):
    """Table definition for storage drive table."""

    __tablename__ = 'storage_drive'

    storage_drive_id = Column(
        'storage_drive_id', Integer, primary_key=True, autoincrement=True)
    storage_id = Column(Integer)