added setup.py
This commit is contained in:
Giovanni Harting
285
ledd/daemon.py
Normal file
285
ledd/daemon.py
Normal file
@@ -0,0 +1,285 @@
|
||||
# LEDD Project
|
||||
# Copyright (C) 2015 LEDD Team
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import asyncore
|
||||
import socket
|
||||
import configparser
|
||||
import json
|
||||
import sqlite3
|
||||
import os
|
||||
import sys
|
||||
import traceback
|
||||
import time
|
||||
import logging
|
||||
from multiprocessing import Process
|
||||
|
||||
import nose
|
||||
|
||||
from ledd import controller, VERSION
|
||||
from ledd.decorators import add_action
|
||||
|
||||
|
||||
class Daemon:
|
||||
daemonSection = 'daemon'
|
||||
databaseSection = 'db'
|
||||
instance = None
|
||||
""":type : Daemon """
|
||||
action_dict = {}
|
||||
|
||||
def __init__(self):
|
||||
Daemon.instance = self
|
||||
logging.basicConfig(level=logging.DEBUG,
|
||||
format="[%(asctime)s] %(levelname)s [%(name)s.%(funcName)s:%(lineno)d] %(message)s",
|
||||
datefmt="%H:%M:%S")
|
||||
try:
|
||||
self.config = configparser.ConfigParser()
|
||||
try:
|
||||
with open('ledd.config', 'w+') as f:
|
||||
self.config.read_file(f)
|
||||
except FileNotFoundError:
|
||||
logging.info("No config file found!")
|
||||
|
||||
self.sqldb = sqlite3.connect(self.config.get(self.databaseSection, 'name', fallback='ledd.sqlite'))
|
||||
self.sqldb.row_factory = sqlite3.Row
|
||||
|
||||
if not self.check_db():
|
||||
self.init_db()
|
||||
|
||||
self.sqldb.commit()
|
||||
|
||||
self.controllers = controller.Controller.from_db(self.sqldb)
|
||||
logging.debug(self.controllers)
|
||||
|
||||
server = self.SocketServer(self.config.get(self.daemonSection, 'host', fallback='0.0.0.0'),
|
||||
self.config.get(self.daemonSection, 'port', fallback=1425))
|
||||
asyncore.loop()
|
||||
except (KeyboardInterrupt, SystemExit):
|
||||
logging.info("Exiting")
|
||||
self.sqldb.close()
|
||||
sys.exit(0)
|
||||
|
||||
def check_db(self):
|
||||
"""
|
||||
Checks database version
|
||||
:return: database validity
|
||||
:rtype: bool
|
||||
"""
|
||||
c = self.sqldb.cursor()
|
||||
try:
|
||||
c.execute("SELECT value FROM meta WHERE option = 'db_version'")
|
||||
db_version = c.fetchone()
|
||||
c.close()
|
||||
|
||||
if db_version is not None:
|
||||
logging.info("DB connection established; db-version=%s", db_version[0])
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
except sqlite3.OperationalError:
|
||||
c.close()
|
||||
return False
|
||||
|
||||
def init_db(self):
|
||||
self.sqldb.close()
|
||||
if os.path.exists("ledd.sqlite"):
|
||||
os.remove("ledd.sqlite")
|
||||
self.sqldb = sqlite3.connect(self.config.get(self.databaseSection, 'name', fallback='ledd.sqlite'))
|
||||
self.sqldb.row_factory = sqlite3.Row
|
||||
with open("ledd/sql/ledd.sql", "r") as sqlfile:
|
||||
c = self.sqldb.cursor()
|
||||
c.executescript(sqlfile.read())
|
||||
c.close()
|
||||
self.check_db()
|
||||
|
||||
@add_action(action_dict)
|
||||
def set_color(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to set color of a stripe.
|
||||
Required JSON parameters: stripe ID: sid; HSV values: h,s,v
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
# TODO: add adapter setting stripe with color here
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
|
||||
@add_action(action_dict)
|
||||
def add_controller(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to add a controller.
|
||||
Required JSON parameters: channels; i2c_dev: number of i2c device (e.g. /dev/i2c-1 would be i2c_dev = 1);
|
||||
address: hexdecimal address of controller on i2c bus, e.g. 0x40
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
try:
|
||||
ncontroller = controller.Controller(Daemon.instance.sqldb, req_json['channels'],
|
||||
req_json['i2c_dev'], req_json['address'])
|
||||
except OSError as e:
|
||||
logging.error("Error opening i2c device: %s", req_json['i2c_dev'])
|
||||
rjson = {
|
||||
'success': False,
|
||||
'message': "Error while opening i2c device",
|
||||
'message_detail': os.strerror(e.errno),
|
||||
'ref': req_json['ref']
|
||||
}
|
||||
return json.dumps(rjson)
|
||||
|
||||
self.controllers.append(ncontroller)
|
||||
|
||||
rjson = {
|
||||
'success': True,
|
||||
'cid': ncontroller.id,
|
||||
'ref': req_json['ref']
|
||||
}
|
||||
|
||||
return json.dumps(rjson)
|
||||
|
||||
@add_action(action_dict)
|
||||
def get_color(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to get the currect color of an stripe.
|
||||
Required JSON parameters: stripeid: sid
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
# TODO: Add get color logic
|
||||
|
||||
@add_action(action_dict)
|
||||
def add_stripes(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to add stripes.
|
||||
Required JSON parameters:
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
if "stripes" in req_json:
|
||||
for stripe in req_json['stripes']:
|
||||
# TODO: add stripe here
|
||||
logging.debug(len(req_json['stripes']))
|
||||
|
||||
@add_action(action_dict)
|
||||
def get_controllers(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to get all registered controllers known to the daemon.
|
||||
Required JSON parameters: none
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
|
||||
rjson = {
|
||||
'success': True,
|
||||
'ccount': len(Daemon.instance.controllers),
|
||||
'controller': Daemon.instance.controllers,
|
||||
'ref': req_json['ref']
|
||||
}
|
||||
|
||||
return json.dumps(rjson, cls=controller.ControllerEncoder)
|
||||
|
||||
@add_action(action_dict)
|
||||
def connection_check(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used to query all channels on a specified controller.
|
||||
Required JSON parameters: controller id: cid
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
|
||||
result = next(filter(lambda x: x.id == req_json['cid'], self.controllers), None)
|
||||
""" :type : Controller """
|
||||
|
||||
if result is not None:
|
||||
for i in range(result.channels):
|
||||
logging.debug("set channel %d=%s", i, "1")
|
||||
result.set_channel(i, 1)
|
||||
time.sleep(10)
|
||||
result.set_channel(i, 0)
|
||||
|
||||
rjson = {
|
||||
'success': True,
|
||||
'ref': req_json['ref']
|
||||
}
|
||||
|
||||
return json.dumps(rjson)
|
||||
|
||||
@add_action(action_dict)
|
||||
def discover(self, req_json):
|
||||
"""
|
||||
Part of the Color API. Used by mobile applications to find the controller.
|
||||
Required JSON parameters: none
|
||||
:param req_json: dict of request json
|
||||
"""
|
||||
logging.debug("recieved action: %s", req_json['action'])
|
||||
|
||||
rjson = {
|
||||
'success': True,
|
||||
'ref': req_json['ref'],
|
||||
'version': VERSION
|
||||
}
|
||||
|
||||
return json.dumps(rjson)
|
||||
|
||||
class ConnectionHandler(asyncore.dispatcher_with_send):
|
||||
def handle_read(self):
|
||||
data = self.recv(5120)
|
||||
self.debug = True
|
||||
|
||||
def no_action_found(self, req_json):
|
||||
rjson = {
|
||||
'success': False,
|
||||
'message': "No action found",
|
||||
'ref': req_json['ref']
|
||||
}
|
||||
return json.dumps(rjson)
|
||||
|
||||
if data:
|
||||
try:
|
||||
json_decoded = json.loads(data.decode())
|
||||
logging.debug(json.dumps(json_decoded, sort_keys=True))
|
||||
|
||||
if "action" in json_decoded and "ref" in json_decoded:
|
||||
return_data = Daemon.instance.action_dict.get(json_decoded['action'], no_action_found)(
|
||||
self=Daemon.instance,
|
||||
req_json=json_decoded)
|
||||
|
||||
if return_data is not None:
|
||||
self.send("{}\n".format(return_data).encode())
|
||||
else:
|
||||
logging.warning("no action or ref value found in JSON, ignoring")
|
||||
except TypeError:
|
||||
logging.error("No valid JSON found: %s", traceback.format_exc())
|
||||
except ValueError:
|
||||
logging.error("No valid JSON detected: %s", traceback.format_exc())
|
||||
|
||||
class SocketServer(asyncore.dispatcher):
|
||||
def __init__(self, host, port):
|
||||
asyncore.dispatcher.__init__(self)
|
||||
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
self.set_reuse_addr()
|
||||
self.bind((host, port))
|
||||
self.listen(5)
|
||||
|
||||
p = Process(target=self.run_tests)
|
||||
p.start()
|
||||
|
||||
@staticmethod
|
||||
def run_tests():
|
||||
nose.run()
|
||||
|
||||
def handle_accept(self):
|
||||
pair = self.accept()
|
||||
if pair is not None:
|
||||
sock, addr = pair
|
||||
logging.debug('Incoming connection from %s' % repr(addr))
|
||||
handler = Daemon.ConnectionHandler(sock)
|
||||
Reference in New Issue
Block a user