make it nicer

Author: lukasIO

livekit-rtc/livekit/rtc/data_stream.py

@@ -70,10 +70,8 @@ async def _on_chunk_update(self, chunk: proto_DataStream.Chunk):
         await self._queue.put(chunk)
 
     async def _on_stream_close(self, trailer: proto_DataStream.Trailer):
-        if self.info.attributes is None:
-            self.info.attributes = dict(trailer.attributes)
-        else:
-            self.info.attributes.update(trailer.attributes)
+        self.info.attributes = self.info.attributes or {}
+        self.info.attributes.update(trailer.attributes)
         await self._queue.put(None)
 
     def __aiter__(self) -> AsyncIterator[str]:
@@ -122,10 +120,8 @@ async def _on_chunk_update(self, chunk: proto_DataStream.Chunk):
         await self._queue.put(chunk)
 
     async def _on_stream_close(self, trailer: proto_DataStream.Trailer):
-        if self.info.attributes is None:
-            self.info.attributes = dict(trailer.attributes)
-        else:
-            self.info.attributes.update(trailer.attributes)
+        self.info.attributes = self.info.attributes or {}
+        self.info.attributes.update(trailer.attributes)
         await self._queue.put(None)
 
     def __aiter__(self) -> AsyncIterator[bytes]: