from sqlalchemy import Column, Integer, String
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class DeliveryBatch(Base):
    __tablename__ = 'delivery_batch'
    delivery_batch_id = Column(Integer, primary_key=True)
    dms_master_master_id = Column(Integer)
    status = Column(String)