Export DataPacket in rtc (#133)

Co-authored-by: Théo Monnom <[email protected]>

Author: davidzhao

examples/basic_room.py

@@ -90,10 +90,8 @@ def on_track_unmuted(
         logging.info("track unmuted: %s", publication.sid)
 
     @room.on("data_received")
-    def on_data_received(
-        data: bytes, kind: rtc.DataPacketKind, participant: rtc.Participant
-    ):
-        logging.info("received data from %s: %s", participant.identity, data)
+    def on_data_received(data: rtc.DataPacket):
+        logging.info("received data from %s: %s", data.participant.identity, data.data)
 
     @room.on("connection_quality_changed")
     def on_connection_quality_changed(

livekit-rtc/livekit/rtc/__init__.py

@@ -15,7 +15,6 @@
 """LiveKit RTC SDK
 """
 
-# flake8: noqa
 from ._proto.room_pb2 import (
     ConnectionQuality,
     ConnectionState,
@@ -33,7 +32,7 @@
 from .audio_source import AudioSource
 from .audio_stream import AudioStream
 from .participant import LocalParticipant, Participant, RemoteParticipant
-from .room import ConnectError, Room, RoomOptions, RtcConfiguration
+from .room import ConnectError, Room, RoomOptions, RtcConfiguration, DataPacket
 from .track import (
     LocalAudioTrack,
     LocalVideoTrack,
@@ -75,3 +74,65 @@
 from .video_stream import VideoStream
 
 from .version import __version__
+
+__all__ = [
+    "ConnectionQuality",
+    "ConnectionState",
+    "DataPacketKind",
+    "TrackPublishOptions",
+    "IceTransportType",
+    "ContinualGatheringPolicy",
+    "IceServer",
+    "EncryptionType",
+    "EncryptionState",
+    "StreamState",
+    "TrackKind",
+    "TrackSource",
+    "VideoFormatType",
+    "VideoFrameBufferType",
+    "VideoRotation",
+    "stats",
+    "AudioFrame",
+    "AudioSource",
+    "AudioStream",
+    "LocalParticipant",
+    "Participant",
+    "RemoteParticipant",
+    "ConnectError",
+    "Room",
+    "RoomOptions",
+    "RtcConfiguration",
+    "DataPacket",
+    "LocalAudioTrack",
+    "LocalVideoTrack",
+    "RemoteAudioTrack",
+    "RemoteVideoTrack",
+    "Track",
+    "LocalTrack",
+    "RemoteTrack",
+    "AudioTrack",
+    "VideoTrack",
+    "E2EEManager",
+    "E2EEOptions",
+    "KeyProviderOptions",
+    "KeyProvider",
+    "FrameCryptor",
+    "LocalTrackPublication",
+    "RemoteTrackPublication",
+    "TrackPublication",
+    "ArgbFrame",
+    "I010Buffer",
+    "I420ABuffer",
+    "I420Buffer",
+    "I422Buffer",
+    "NativeVideoBuffer",
+    "NV12Buffer",
+    "PlanarYuv8Buffer",
+    "PlanarYuv16Buffer",
+    "PlanarYuvBuffer",
+    "VideoFrame",
+    "VideoFrameBuffer",
+    "VideoSource",
+    "VideoStream",
+    "__version__",
+]