97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
import json
|
|
import uuid
|
|
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext as _
|
|
from django.db.models.signals import pre_delete
|
|
from portier.common import handlers
|
|
from . import signals_shared
|
|
|
|
|
|
class Stream(models.Model):
|
|
stream = models.UUIDField(unique=True, default=uuid.uuid4, help_text=_('stream_stream_help'))
|
|
name = models.CharField(max_length=100, help_text=_('stream_name_help'))
|
|
|
|
# the same stream uuid can be published multiple times to different origin
|
|
# servers. this is a valid scheme to achieve a failover on the origin layer.
|
|
# thus we need to keep track if a stream is published at least once,
|
|
# and only send signals when we are going to / coming from 0 published streams.
|
|
publish_counter = models.PositiveIntegerField(default=0)
|
|
|
|
def on_publish(self, param):
|
|
# if so far there were less than one incoming streams, this stream
|
|
# is now being considered active
|
|
if self.publish_counter < 1:
|
|
signals_shared.stream_active.send(sender=self.__class__,
|
|
stream=str(self.stream),
|
|
param=param
|
|
)
|
|
|
|
# keep track of this incoming stream
|
|
self.publish_counter += 1
|
|
self.save()
|
|
|
|
def on_unpublish(self, param):
|
|
# note that we now have on less incoming stream
|
|
if self.publish_counter > 0:
|
|
self.publish_counter -= 1
|
|
|
|
# if we now have less than one incoming stream, this stream is being
|
|
# considered inactive
|
|
if self.publish_counter < 1:
|
|
signals_shared.stream_inactive.send(sender=self.__class__,
|
|
stream=str(self.stream),
|
|
param=param
|
|
)
|
|
self.save()
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('config:stream_detail', kwargs={'pk': self.pk})
|
|
|
|
def class_name(self):
|
|
return _('stream_class_name')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
pre_delete.connect(handlers.remove_obj_perms_connected_with_user, sender=Stream)
|
|
|
|
class Restream(models.Model):
|
|
FORMATS = (
|
|
('flv', 'flv (RTMP)'),
|
|
('mpegts', 'mpegts (SRT)'),
|
|
)
|
|
stream = models.ForeignKey(Stream, on_delete=models.CASCADE, help_text=_('restream_stream_help'))
|
|
target = models.CharField(max_length=500, help_text=_('restream_target_help'))
|
|
name = models.CharField(max_length=100, help_text=_('restream_name_help'))
|
|
active = models.BooleanField(help_text=_('restream_activate_help'))
|
|
format = models.CharField(max_length=6, choices=FORMATS, default='flv', help_text=_('restream_format_help'))
|
|
|
|
class Meta:
|
|
verbose_name = _('restream_verbose_name')
|
|
verbose_name_plural = _('restream_verbose_name_plural')
|
|
|
|
def class_name(self):
|
|
return _('restream_class_name')
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('config:restream_detail', kwargs={'pk': self.pk})
|
|
|
|
def __str__(self):
|
|
return '{} to {}'.format(self.stream, self.name)
|
|
|
|
def get_json_config(self):
|
|
config = {
|
|
'name': self.name,
|
|
'app': settings.GLOBAL_STREAM_NAMESPACE,
|
|
'stream': str(self.stream.stream),
|
|
'target': self.target,
|
|
'format': self.format
|
|
}
|
|
return json.dumps(config)
|
|
|
|
|
|
pre_delete.connect(handlers.remove_obj_perms_connected_with_user, sender=Restream) |