light9 Changeset - a0abf52b06d6

Changeset - a0abf52b06d6

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

drewp@bigasterisk.com - 20 months ago 2023-06-01 21:16:45
drewp@bigasterisk.com

collector: auto find dmx address (at startup) if it reconnects usb with a new one

1 file changed with 13 insertions and 2 deletions:

light9/collector/service.py

0 comments (0 inline, 0 general)

light9/collector/service.py

➞

Show inline comments

 #!bin/python
 """
 Collector receives device attrs from multiple senders, combines
 them, and sends output attrs to hardware. The combining part has
 custom code for some attributes.
 Input can be over http or zmq.
 """
 import asyncio
 import functools
 import logging
 import subprocess
 import traceback
 from typing import List
 from light9 import networking
 from light9.collector.collector import Collector
 from light9.collector.output import ArtnetDmx, DummyOutput, Output, Udmx  # noqa
 from light9.collector.weblisteners import UiListener, WebListeners
 from light9.namespaces import L9
 from light9.run_local import log
 from light9.zmqtransport import parseJsonMessage
 from prometheus_client import Summary
 from rdfdb.syncedgraph.syncedgraph import SyncedGraph
@@ @@ -26,24 +27,25 @@ from starlette.endpoints import WebSocke @@
 from starlette.requests import ClientDisconnect
 from starlette.responses import Response
 from starlette.routing import Route, WebSocketRoute
 from starlette.types import Receive, Scope, Send
 from starlette.websockets import WebSocket
 from starlette_exporter import PrometheusMiddleware, handle_metrics
 import zmq
 import zmq.asyncio
 STAT_SETATTR = Summary('set_attr', 'setAttr calls')
 # this is the rate sent to usb
 RATE = 20
 class Updates(WebSocketEndpoint, UiListener):
     def __init__(self, listeners, scope: Scope, receive: Receive, send: Send) -> None:
         super().__init__(scope, receive, send)
         self.listeners = listeners
     async def on_connect(self, websocket: WebSocket):
         await websocket.accept()
         log.info('socket connect %s', self.scope['client'])
@@ @@ -82,42 +84,51 @@ async def zmqListener(collector): @@
         sock.subscribe(b'setAttr')
         while True:
             [topic, msg] = await sock.recv_multipart()
             if topic != b'setAttr':
                 raise ValueError(topic)
             # log.info(f'zmq recv {len(msg)}')
             client, clientSession, settings, sendTime = parseJsonMessage(collector.graph, msg)
             collector.setAttrs(client, clientSession, settings, sendTime)
     except:
         traceback.print_exc()
         raise
 def findDevice():
     for line in subprocess.check_output("lsusb").decode('utf8').splitlines():
         if '16c0:05dc' in line:
             words = line.split(':')[0].split()
             dev = f'/dev/bus/usb/{words[1]}/{words[3]}'
             log.info(f'device will be {dev}')
             return dev ,int(words[3])
     raise ValueError("no matching uDMX found")
 def main():
     logging.getLogger('autodepgraphapi').setLevel(logging.INFO)
     logging.getLogger('syncedgraph').setLevel(logging.INFO)
     logging.getLogger('output.allDmx').setLevel(logging.WARNING)
     graph = SyncedGraph(networking.rdfdb.url, "collector")
     devPath, usbAddress = findDevice()
             # if user doesn't have r/w, fail now
     try:
         # todo: drive outputs with config files
         rate = 30
         outputs: List[Output] = [
             # ArtnetDmx(L9['output/dmxA/'],
             #           host='127.0.0.1',
             #           port=6445,
             #           rate=rate),
             #sudo chmod a+rw /dev/bus/usb/003/021
-            Udmx(L9['output/dmxA/'], bus=3, address=21, lastDmxChannel=200, rate=RATE),
+            Udmx(L9['output/dmxA/'], bus=1, address=usbAddress, lastDmxChannel=200, rate=RATE),
+        ]
     except Exception:
         log.error("setting up outputs:")
         traceback.print_exc()
         raise
     listeners = WebListeners()
     c = Collector(graph, outputs, listeners)
     zl = asyncio.create_task(zmqListener(c))
     app = Starlette(
         debug=True,
         routes=[
             # Route('/recentRequests', lambda req: get_recentRequests(req, db)),

0 comments (0 inline, 0 general)