forked from platypush/platypush
Fabio Manganiello
d872835093
- Check if it's part of the metadata through a function call rather than checking `Base.metadata` in every single module. - Make it possible to override them (mostly for doc generation logic that needs to be able to import those classes). - Make it possible to extend them.
36 lines
840 B
Python
36 lines
840 B
Python
from sqlalchemy import Column, Integer, ForeignKey
|
|
|
|
from platypush.common.db import is_defined
|
|
|
|
from .dimmers import Dimmer
|
|
from .switches import Switch
|
|
|
|
|
|
if not is_defined('volume'):
|
|
|
|
class Volume(Dimmer):
|
|
__tablename__ = 'volume'
|
|
|
|
id = Column(
|
|
Integer, ForeignKey(Dimmer.id, ondelete='CASCADE'), primary_key=True
|
|
)
|
|
|
|
__table_args__ = {'extend_existing': True}
|
|
__mapper_args__ = {
|
|
'polymorphic_identity': __tablename__,
|
|
}
|
|
|
|
|
|
if not is_defined('muted'):
|
|
|
|
class Muted(Switch):
|
|
__tablename__ = 'muted'
|
|
|
|
id = Column(
|
|
Integer, ForeignKey(Switch.id, ondelete='CASCADE'), primary_key=True
|
|
)
|
|
|
|
__table_args__ = {'extend_existing': True}
|
|
__mapper_args__ = {
|
|
'polymorphic_identity': __tablename__,
|
|
}
|