@@ -26,6 +26,8 @@ import asyncio
|
|||||||
|
|
||||||
import spectra
|
import spectra
|
||||||
|
|
||||||
|
from zeroconf import Zeroconf, ServiceInfo
|
||||||
|
|
||||||
from ledd import controller, VERSION
|
from ledd import controller, VERSION
|
||||||
from ledd.decorators import ledd_protocol
|
from ledd.decorators import ledd_protocol
|
||||||
from ledd.stripe import Stripe
|
from ledd.stripe import Stripe
|
||||||
@@ -46,6 +48,7 @@ class Daemon:
|
|||||||
Daemon.instance = self
|
Daemon.instance = self
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
# read config
|
||||||
self.config = configparser.ConfigParser()
|
self.config = configparser.ConfigParser()
|
||||||
try:
|
try:
|
||||||
with open('ledd.config', 'w+') as f:
|
with open('ledd.config', 'w+') as f:
|
||||||
@@ -53,6 +56,11 @@ class Daemon:
|
|||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
log.info("No config file found!")
|
log.info("No config file found!")
|
||||||
|
|
||||||
|
# create zeroconf service info
|
||||||
|
self.zinfo = ServiceInfo("_ledd._tcp", "LedD Daemon",
|
||||||
|
port=self.config.get(self.daemonSection, 'port', fallback=1425))
|
||||||
|
|
||||||
|
# SQL init
|
||||||
self.sqldb = sqlite3.connect(self.config.get(self.databaseSection, 'name', fallback='ledd.sqlite'))
|
self.sqldb = sqlite3.connect(self.config.get(self.databaseSection, 'name', fallback='ledd.sqlite'))
|
||||||
self.sqldb.row_factory = sqlite3.Row
|
self.sqldb.row_factory = sqlite3.Row
|
||||||
|
|
||||||
@@ -61,10 +69,15 @@ class Daemon:
|
|||||||
|
|
||||||
self.sqldb.commit()
|
self.sqldb.commit()
|
||||||
|
|
||||||
|
# init controllers from db
|
||||||
self.controllers = controller.Controller.from_db(self.sqldb)
|
self.controllers = controller.Controller.from_db(self.sqldb)
|
||||||
log.debug(self.controllers)
|
log.debug(self.controllers)
|
||||||
logging.getLogger("asyncio").setLevel(logging.DEBUG)
|
logging.getLogger("asyncio").setLevel(logging.DEBUG)
|
||||||
|
|
||||||
|
# announce server to network
|
||||||
|
self.register_zeroconf()
|
||||||
|
|
||||||
|
# main loop
|
||||||
self.loop = asyncio.get_event_loop()
|
self.loop = asyncio.get_event_loop()
|
||||||
coro = self.loop.create_server(LedDProtocol,
|
coro = self.loop.create_server(LedDProtocol,
|
||||||
self.config.get(self.daemonSection, 'host', fallback='0.0.0.0'),
|
self.config.get(self.daemonSection, 'host', fallback='0.0.0.0'),
|
||||||
@@ -73,6 +86,7 @@ class Daemon:
|
|||||||
self.loop.run_forever()
|
self.loop.run_forever()
|
||||||
except (KeyboardInterrupt, SystemExit):
|
except (KeyboardInterrupt, SystemExit):
|
||||||
log.info("Exiting")
|
log.info("Exiting")
|
||||||
|
self.deregister_zeroconf()
|
||||||
self.sqldb.close()
|
self.sqldb.close()
|
||||||
self.server.close()
|
self.server.close()
|
||||||
self.loop.run_until_complete(self.server.wait_closed())
|
self.loop.run_until_complete(self.server.wait_closed())
|
||||||
@@ -112,6 +126,16 @@ class Daemon:
|
|||||||
c.close()
|
c.close()
|
||||||
self.check_db()
|
self.check_db()
|
||||||
|
|
||||||
|
def register_zeroconf(self):
|
||||||
|
zeroconf = Zeroconf()
|
||||||
|
zeroconf.register_service(self.zinfo)
|
||||||
|
log.info("Registered ledd daemon with zeroconf")
|
||||||
|
|
||||||
|
def deregister_zeroconf(self):
|
||||||
|
zeroconf = Zeroconf()
|
||||||
|
zeroconf.unregister_service(self.zinfo)
|
||||||
|
log.info("Unregistered ledd daemon with zeroconf")
|
||||||
|
|
||||||
@ledd_protocol(protocol)
|
@ledd_protocol(protocol)
|
||||||
def set_color(self, req_json):
|
def set_color(self, req_json):
|
||||||
"""
|
"""
|
||||||
|
Reference in New Issue
Block a user