changeset 1969:fc66b28302a5

some types for subclient Ignore-this: a01ce3ca8427241d5f543e427167891d
author drewp@bigasterisk.com
date Sat, 08 Jun 2019 03:51:49 +0000
parents 6e1b06199cff
children 3767ae7187d9
files light9/subclient.py
diffstat 1 files changed, 8 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/light9/subclient.py	Sat Jun 08 03:50:57 2019 +0000
+++ b/light9/subclient.py	Sat Jun 08 03:51:49 2019 +0000
@@ -1,13 +1,18 @@
 from light9.collector.collector_client import sendToCollector
 from twisted.internet import reactor
+from twisted.internet.defer import Deferred
 import traceback
 import time
 import logging
+from rdflib import URIRef
+from rdfdb.syncedgraph import SyncedGraph
 log = logging.getLogger()
 
 
 class SubClient:
-
+    graph: SyncedGraph
+    session: URIRef
+    
     def __init__(self):
         """assumed that your init saves self.graph"""
         pass  # we may later need init code for network setup
@@ -19,7 +24,7 @@
     def send_levels(self):
         self._send_sub()
 
-    def send_levels_loop(self, delay=1000):
+    def send_levels_loop(self, delay=1000) -> None:
         now = time.time()
 
         def done(sec):
@@ -34,7 +39,7 @@
         d = self._send_sub()
         d.addCallbacks(done, err)
 
-    def _send_sub(self):
+    def _send_sub(self) -> Deferred:
         try:
             with self.graph.currentState() as g:
                 outputSettings = self.get_output_settings(_graph=g)