42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
"""Provide the ORM's Restaurant model."""
|
||
|
|
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from sqlalchemy import orm
|
||
|
|
|
||
|
|
from urban_meal_delivery.db import meta
|
||
|
|
|
||
|
|
|
||
|
|
class Restaurant(meta.Base):
|
||
|
|
"""A Restaurant selling meals on the UDP."""
|
||
|
|
|
||
|
|
# pylint:disable=too-few-public-methods
|
||
|
|
|
||
|
|
__tablename__ = 'restaurants'
|
||
|
|
|
||
|
|
# Columns
|
||
|
|
id = sa.Column( # noqa:WPS125
|
||
|
|
sa.SmallInteger, primary_key=True, autoincrement=False,
|
||
|
|
)
|
||
|
|
created_at = sa.Column(sa.DateTime, nullable=False)
|
||
|
|
name = sa.Column(sa.Unicode(length=45), nullable=False) # noqa:WPS432
|
||
|
|
_address_id = sa.Column('address_id', sa.Integer, nullable=False, index=True)
|
||
|
|
estimated_prep_duration = sa.Column(sa.SmallInteger, nullable=False)
|
||
|
|
|
||
|
|
# Constraints
|
||
|
|
__table_args__ = (
|
||
|
|
sa.ForeignKeyConstraint(
|
||
|
|
['address_id'], ['addresses.id'], onupdate='RESTRICT', ondelete='RESTRICT',
|
||
|
|
),
|
||
|
|
sa.CheckConstraint(
|
||
|
|
'0 <= estimated_prep_duration AND estimated_prep_duration <= 2400',
|
||
|
|
name='realistic_estimated_prep_duration',
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
# Relationships
|
||
|
|
address = orm.relationship('Address', back_populates='restaurant')
|
||
|
|
orders = orm.relationship('Order', back_populates='restaurant')
|
||
|
|
|
||
|
|
def __repr__(self) -> str:
|
||
|
|
"""Non-literal text representation."""
|
||
|
|
return '<{cls}({name})>'.format(cls=self.__class__.__name__, name=self.name)
|