22from .consts import getProjectEndpoint , initDatasetIncrement , addDatasetIncrement
33from .Dataset import Dataset
44import time
5+ import asyncio
56
67class DatasetReceiver :
78
@@ -17,10 +18,10 @@ def __init__(self, backendURL, readKey=None, writeKey=None):
1718 self .datasets = []
1819 res_data = res .json ()
1920 datasets = res_data ["datasets" ]
20- self .labeligns = res_data ["labelings" ]
21+ self .labelings = res_data ["labelings" ]
2122 for d in datasets :
2223 tmp_dataset = Dataset (backendURL , self ._readKey , self ._writeKey )
23- tmp_dataset .parse (d , self .labeligns )
24+ tmp_dataset .parse (d , self .labelings )
2425 self .datasets .append (tmp_dataset )
2526
2627 def loadData (self ):
@@ -45,12 +46,12 @@ def __str__(self) -> str:
4546
4647class DatasetCollector :
4748 def __init__ (
48- self , url , apiKey , name , useDeviceTime , timeSeries , metaData , datasetLabel = None
49+ self , url , write_key , name , use_own_timestamps , timeSeries , metaData , datasetLabel = None
4950 ):
5051 self .url = url
51- self .apiKey = apiKey
52+ self .apiKey = write_key
5253 self .name = name
53- self .useDeviceTime = useDeviceTime
54+ self .use_own_timestamps = use_own_timestamps
5455 self .timeSeries = timeSeries
5556 self .metaData = metaData
5657 self .datasetLabel = datasetLabel
@@ -59,7 +60,7 @@ def __init__(
5960 self .labeling = None
6061 self .lastChecked = time .time () * 1000
6162
62- if self .useDeviceTime :
63+ if self .use_own_timestamps :
6364 self .addDataPoint = self ._addDataPoint_DeviceTime
6465 else :
6566 self .addDataPoint = self ._addDataPoint_OwnTimeStamps
@@ -72,7 +73,7 @@ def __init__(
7273 self .error = None
7374
7475 res = req .post (
75- url + initDatasetIncrement + apiKey ,
76+ url + initDatasetIncrement + self . apiKey ,
7677 json = {
7778 "name" : self .name ,
7879 "metaData" : self .metaData ,
@@ -106,7 +107,7 @@ async def _addDataPoint_OwnTimeStamps(self, timestamp, name, value):
106107 self .dataStore [name ].append ([timestamp , value ])
107108
108109 if time .time () * 1000 - self .lastChecked > UPLOAD_INTERVAL :
109- self .upload (self .labeling )
110+ asyncio . create_task ( self .upload (self .labeling ) )
110111 self .lastChecked = time .time () * 1000
111112
112113
0 commit comments