mirror of
https://github.com/Dispatcharr/Dispatcharr.git
synced 2026-07-20 16:51:10 +00:00
129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
from rest_framework import serializers
|
|
from .models import Stream, Channel, ChannelGroup, ChannelStream
|
|
from core.models import StreamProfile
|
|
|
|
#
|
|
# Stream
|
|
#
|
|
class StreamSerializer(serializers.ModelSerializer):
|
|
stream_profile_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=StreamProfile.objects.all(),
|
|
source='stream_profile',
|
|
allow_null=True,
|
|
required=False
|
|
)
|
|
|
|
class Meta:
|
|
model = Stream
|
|
fields = [
|
|
'id',
|
|
'name',
|
|
'url',
|
|
'custom_url',
|
|
'm3u_account', # Uncomment if using M3U fields
|
|
'logo_url',
|
|
'tvg_id',
|
|
'local_file',
|
|
'current_viewers',
|
|
'updated_at',
|
|
'group_name',
|
|
'stream_profile_id',
|
|
]
|
|
|
|
def get_fields(self):
|
|
fields = super().get_fields()
|
|
|
|
# Unable to edit specific properties if this stream was created from an M3U account
|
|
if self.instance and getattr(self.instance, 'm3u_account', None):
|
|
fields['id'].read_only = True
|
|
fields['name'].read_only = True
|
|
fields['url'].read_only = True
|
|
fields['m3u_account'].read_only = True
|
|
fields['tvg_id'].read_only = True
|
|
fields['group_name'].read_only = True
|
|
|
|
return fields
|
|
|
|
#
|
|
# Channel Group
|
|
#
|
|
class ChannelGroupSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = ChannelGroup
|
|
fields = ['id', 'name']
|
|
|
|
|
|
#
|
|
# Channel
|
|
#
|
|
class ChannelSerializer(serializers.ModelSerializer):
|
|
# Show nested group data, or ID
|
|
channel_group = ChannelGroupSerializer(read_only=True)
|
|
channel_group_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=ChannelGroup.objects.all(),
|
|
source="channel_group",
|
|
write_only=True,
|
|
required=False
|
|
)
|
|
|
|
stream_profile_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=StreamProfile.objects.all(),
|
|
source='stream_profile',
|
|
allow_null=True,
|
|
required=False
|
|
)
|
|
|
|
streams = serializers.ListField(
|
|
child=serializers.IntegerField(), write_only=True
|
|
)
|
|
stream_ids = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Channel
|
|
fields = [
|
|
'id',
|
|
'channel_number',
|
|
'channel_name',
|
|
'logo_url',
|
|
'logo_file',
|
|
'channel_group',
|
|
'channel_group_id',
|
|
'tvg_id',
|
|
'tvg_name',
|
|
'streams',
|
|
'stream_ids',
|
|
'stream_profile_id',
|
|
]
|
|
|
|
def get_stream_ids(self, obj):
|
|
"""Retrieve ordered stream IDs for GET requests."""
|
|
return list(obj.streams.all().order_by('channelstream__order').values_list('id', flat=True))
|
|
|
|
def create(self, validated_data):
|
|
stream_ids = validated_data.pop('streams', [])
|
|
channel = Channel.objects.create(**validated_data)
|
|
|
|
# Add streams in the specified order
|
|
for index, stream_id in enumerate(stream_ids):
|
|
ChannelStream.objects.create(channel=channel, stream_id=stream_id, order=index)
|
|
|
|
return channel
|
|
|
|
def update(self, instance, validated_data):
|
|
print("Validated Data:", validated_data)
|
|
stream_ids = validated_data.get('streams', None)
|
|
print(f'stream ids: {stream_ids}')
|
|
|
|
# Update basic fields
|
|
instance.name = validated_data.get('channel_name', instance.channel_name)
|
|
instance.save()
|
|
|
|
if stream_ids is not None:
|
|
# Clear existing relationships
|
|
instance.channelstream_set.all().delete()
|
|
|
|
# Add new streams in order
|
|
for index, stream_id in enumerate(stream_ids):
|
|
ChannelStream.objects.create(channel=instance, stream_id=stream_id, order=index)
|
|
|
|
return instance
|