mirror of
https://github.com/CCOSTAN/Home-AssistantConfig.git
synced 2025-08-13 17:17:51 +00:00
You can find the entire repo here.
This commit is contained in:
82
config/custom_components/sensor/minecraft.py
Executable file
82
config/custom_components/sensor/minecraft.py
Executable file
@@ -0,0 +1,82 @@
|
||||
"""
|
||||
Sensor to check the status of a Minecraft server.
|
||||
|
||||
"""
|
||||
import logging
|
||||
from homeassistant.helpers.entity import Entity
|
||||
ATTR_USERS = 'users_online'
|
||||
ATTR_MAX = 'users_max'
|
||||
ATTR_MOTD = 'MOTD'
|
||||
ATTR_VERSION = 'Version'
|
||||
ICON = 'mdi:minecraft'
|
||||
REQUIREMENTS = ['mcstatus==2.1']
|
||||
|
||||
# pylint: disable=unused-argument
|
||||
def setup_platform(hass, config, add_devices, discovery_info=None):
|
||||
"""Setup the Minecraft server platform."""
|
||||
from mcstatus import MinecraftServer as mcserver
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
server = config.get('server')
|
||||
name = config.get('name')
|
||||
|
||||
if server is None:
|
||||
logger.error('No server specified')
|
||||
return False
|
||||
elif name is None:
|
||||
logger.error('No name specified')
|
||||
return False
|
||||
|
||||
add_devices([
|
||||
MCServerSensor(server, name, mcserver)
|
||||
])
|
||||
|
||||
|
||||
class MCServerSensor(Entity):
|
||||
"""A class for the Minecraft server."""
|
||||
|
||||
# pylint: disable=abstract-method
|
||||
def __init__(self, server, name, mcserver):
|
||||
"""Initialize the sensor."""
|
||||
self._mcserver = mcserver
|
||||
self._server = server
|
||||
self._name = name
|
||||
self.update()
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the server."""
|
||||
return self._name
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return the state of the sensor."""
|
||||
return self._state
|
||||
|
||||
# pylint: disable=no-member
|
||||
def update(self):
|
||||
"""Update device state."""
|
||||
status = self._mcserver.lookup(self._server).status()
|
||||
query = self._mcserver.lookup(self._server).query()
|
||||
self._state = status.players.online
|
||||
#self._max = str(status.players.max)
|
||||
self._max = status.players.max
|
||||
self._users = query.players.names
|
||||
self._motd = query.motd
|
||||
self._version = query.software.version
|
||||
|
||||
|
||||
@property
|
||||
def device_state_attributes(self):
|
||||
"""Return the state attributes."""
|
||||
return {
|
||||
ATTR_USERS: self._users,
|
||||
ATTR_MAX: self._max,
|
||||
ATTR_MOTD: self._motd,
|
||||
ATTR_VERSION: self._version
|
||||
}
|
||||
|
||||
@property
|
||||
def icon(self):
|
||||
"""Return the icon to use in the frontend."""
|
||||
return ICON
|
Reference in New Issue
Block a user