forked from platypush/platypush
27 lines
751 B
Python
27 lines
751 B
Python
from sqlalchemy import Column, Integer, ForeignKey
|
|
|
|
from .devices import entity_types_registry
|
|
from .sensors import NumericSensor
|
|
|
|
|
|
if not entity_types_registry.get('Battery'):
|
|
|
|
class Battery(NumericSensor):
|
|
__tablename__ = 'battery'
|
|
|
|
def __init__(
|
|
self, *args, unit: str = '%', min: float = 0, max: float = 100, **kwargs
|
|
):
|
|
super().__init__(*args, min=min, max=max, unit=unit, **kwargs)
|
|
|
|
id = Column(
|
|
Integer, ForeignKey(NumericSensor.id, ondelete='CASCADE'), primary_key=True
|
|
)
|
|
|
|
__mapper_args__ = {
|
|
'polymorphic_identity': __tablename__,
|
|
}
|
|
|
|
entity_types_registry['Battery'] = Battery
|
|
else:
|
|
Battery = entity_types_registry['Battery']
|