|
4 | 4 |
|
5 | 5 | from viam import logging |
6 | 6 | from viam.proto.app.data import ( |
7 | | - AddTagsToBinaryDataByIDsRequest, |
8 | | - AddTagsToBinaryDataByIDsResponse, |
9 | 7 | AddTagsToBinaryDataByFilterRequest, |
10 | 8 | AddTagsToBinaryDataByFilterResponse, |
11 | | - RemoveTagsFromBinaryDataByIDsRequest, |
12 | | - RemoveTagsFromBinaryDataByIDsResponse, |
13 | | - RemoveTagsFromBinaryDataByFilterRequest, |
14 | | - RemoveTagsFromBinaryDataByFilterResponse, |
15 | | - TagsByFilterRequest, |
16 | | - TagsByFilterResponse, |
17 | | - BoundingBoxLabelsByFilterRequest, |
18 | | - BoundingBoxLabelsByFilterResponse, |
| 9 | + AddTagsToBinaryDataByIDsRequest, |
| 10 | + AddTagsToBinaryDataByIDsResponse, |
19 | 11 | BinaryDataByFilterRequest, |
20 | 12 | BinaryDataByFilterResponse, |
21 | 13 | BinaryDataByIDsRequest, |
22 | 14 | BinaryDataByIDsResponse, |
23 | 15 | BinaryID, |
24 | | - DataServiceStub, |
| 16 | + BoundingBoxLabelsByFilterRequest, |
| 17 | + BoundingBoxLabelsByFilterResponse, |
25 | 18 | DataRequest, |
26 | | - DeleteBinaryDataByIDsRequest, |
27 | | - DeleteBinaryDataByIDsResponse, |
| 19 | + DataServiceStub, |
28 | 20 | DeleteBinaryDataByFilterRequest, |
29 | 21 | DeleteBinaryDataByFilterResponse, |
| 22 | + DeleteBinaryDataByIDsRequest, |
| 23 | + DeleteBinaryDataByIDsResponse, |
30 | 24 | DeleteTabularDataByFilterRequest, |
31 | 25 | DeleteTabularDataByFilterResponse, |
32 | 26 | Filter, |
| 27 | + RemoveTagsFromBinaryDataByFilterRequest, |
| 28 | + RemoveTagsFromBinaryDataByFilterResponse, |
| 29 | + RemoveTagsFromBinaryDataByIDsRequest, |
| 30 | + RemoveTagsFromBinaryDataByIDsResponse, |
33 | 31 | TabularDataByFilterRequest, |
34 | 32 | TabularDataByFilterResponse, |
| 33 | + TagsByFilterRequest, |
| 34 | + TagsByFilterResponse, |
35 | 35 | ) |
36 | 36 | from viam.proto.app.datasync import DataSyncServiceStub, FileData, SensorData, UploadMetadata |
37 | 37 |
|
@@ -92,7 +92,7 @@ async def tabular_data_by_filter( |
92 | 92 |
|
93 | 93 | if dest: |
94 | 94 | try: |
95 | | - file = open(dest, 'w') |
| 95 | + file = open(dest, "w") |
96 | 96 | file.write(f"{data}") |
97 | 97 | except Exception as e: |
98 | 98 | LOGGER.error(f"Failed to write tabular data to file {dest}", exc_info=e) |
@@ -130,7 +130,7 @@ async def binary_data_by_filter( |
130 | 130 |
|
131 | 131 | if dest: |
132 | 132 | try: |
133 | | - file = open(dest, 'w') |
| 133 | + file = open(dest, "w") |
134 | 134 | file.write(f"{data}") |
135 | 135 | except Exception as e: |
136 | 136 | LOGGER.error(f"Failed to write binary data to file {dest}", exc_info=e) |
@@ -158,7 +158,7 @@ async def binary_data_by_ids( |
158 | 158 | response: BinaryDataByIDsResponse = await self._data_client.BinaryDataByIDs(request, metadata=self._metadata) |
159 | 159 | if dest: |
160 | 160 | try: |
161 | | - file = open(dest, 'w') |
| 161 | + file = open(dest, "w") |
162 | 162 | file.write(f"{response.data}") |
163 | 163 | except Exception as e: |
164 | 164 | LOGGER.error(f"Failed to write binary data to file {dest}", exc_info=e) |
@@ -251,8 +251,7 @@ async def remove_tags_from_binary_data_by_filter(self, tags: List[str], filter: |
251 | 251 | filter = filter if filter else Filter() |
252 | 252 | request = RemoveTagsFromBinaryDataByFilterRequest(filter=filter, tags=tags) |
253 | 253 | _: RemoveTagsFromBinaryDataByFilterResponse = await self._data_client.RemoveTagsFromBinaryDataByFilter( |
254 | | - request, |
255 | | - metadata=self._metadata |
| 254 | + request, metadata=self._metadata |
256 | 255 | ) |
257 | 256 |
|
258 | 257 | async def tags_by_filter(self, filter: Optional[Filter] = None) -> List[str]: |
@@ -292,7 +291,7 @@ async def bounding_box_labels_by_filter(self, filter: Optional[Filter] = None) - |
292 | 291 | return response.labels |
293 | 292 |
|
294 | 293 | # TODO(RSDK-3637): Implement |
295 | | - async def data_capture_upload(self, metadata: UploadMetadata , sensor_contents: Optional[List[SensorData]] = None) -> None: |
| 294 | + async def data_capture_upload(self, metadata: UploadMetadata, sensor_contents: Optional[List[SensorData]] = None) -> None: |
296 | 295 | raise NotImplementedError() |
297 | 296 |
|
298 | 297 | # TODO(RSDK-3637): Implement |
|
0 commit comments