"""
Storage Drive Asset Folder model.

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

from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import String

from assets.connectors import mysql


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

    __tablename__ = 'storage_drive_asset_folder'

    storage_drive_asset_folder_id = Column(
        'storage_drive_asset_folder_id', Integer,
        primary_key=True, autoincrement=True)
    storage_drive_id = Column(Integer)
    asset_type_id = Column(Integer)
    initial_folder = Column(String)