Добавлена обработка заголовков запроса X-Forwarded...

[vpproxy.git] / vlcclient / vlcclient.py

diff --git a/vlcclient/vlcclient.py b/vlcclient/vlcclient.py
index b2226997e7a2d38d4cfdb0b3b2bc45ebdc122ff9..1fa5e4330d5feb7640febe5c42575956efea45ed 100644 (file)
--- a/vlcclient/vlcclient.py
+++ b/vlcclient/vlcclient.py
@@ -8,7 +8,7 @@ import gevent.coros
 import telnetlib
 import logging
 from vlcmessages import *
 import telnetlib
 import logging
 from vlcmessages import *

-
+import time

 
 class VlcException(Exception):
 
 
 class VlcException(Exception):
 


@@ -25,8 +25,8 @@ class VlcClient(object):
     '''
 
     def __init__(
     '''
 
     def __init__(

-        self, host='127.0.0.1', port=4212, password='admin', connect_timeout=5,
-            result_timeout=5, out_port=8081):
+        self, host='127.0.0.1', port=4212, password='admin', connect_timeout=10,
+            result_timeout=10, out_port=8081):

         # Receive buffer
         self._recvbuffer = None
         # Output port
         # Receive buffer
         self._recvbuffer = None
         # Output port


@@ -51,10 +51,13 @@ class VlcClient(object):
         # Logger
         logger = logging.getLogger('VlcClient_init')
 
         # Logger
         logger = logging.getLogger('VlcClient_init')
 

+        # Streams
+        self.streams=dict()
+

         # Making connection
         try:
             self._socket = telnetlib.Telnet(host, port, connect_timeout)
         # Making connection
         try:
             self._socket = telnetlib.Telnet(host, port, connect_timeout)

-            logger.debug("Successfully connected with VLC socket!")
+            logger.info("Successfully connected with VLC socket!")

         except Exception as e:
             raise VlcException(
                 "Socket creation error! VLC is not running? ERROR: " + repr(e))
         except Exception as e:
             raise VlcException(
                 "Socket creation error! VLC is not running? ERROR: " + repr(e))


@@ -89,7 +92,7 @@ class VlcClient(object):
         # If socket is still alive (connected)
         if self._socket:
             try:
         # If socket is still alive (connected)
         if self._socket:
             try:

-                logger.debug("Destroying VlcClient...")
+                logger.info("Destroying VlcClient...")

                 self._write(VlcMessage.request.SHUTDOWN)
                 # Set shuttingDown flag for recvData
                 self._shuttingDown.set()
                 self._write(VlcMessage.request.SHUTDOWN)
                 # Set shuttingDown flag for recvData
                 self._shuttingDown.set()


@@ -98,13 +101,16 @@ class VlcClient(object):
                 pass
 
     def _write(self, message):
                 pass
 
     def _write(self, message):

+
+        logger = logging.getLogger("VlcClient_write")
+

         # Return if in the middle of destroying
         if self._shuttingDown.isSet():
             return
 
         try:
             # Write message
         # Return if in the middle of destroying
         if self._shuttingDown.isSet():
             return
 
         try:
             # Write message

-            print 'VLC command:',message
+            logger.info('VLC command: ' + message)

             self._socket.write(message + "\r\n")
         except EOFError as e:
             raise VlcException("Vlc Write error! ERROR: " + repr(e))
             self._socket.write(message + "\r\n")
         except EOFError as e:
             raise VlcException("Vlc Write error! ERROR: " + repr(e))


@@ -130,6 +136,10 @@ class VlcClient(object):
             msg = VlcMessage.request.startBroadcast(stream_name, input, self._out_port, muxer, pre_access, qtype)
             self._write(msg)
         else:
             msg = VlcMessage.request.startBroadcast(stream_name, input, self._out_port, muxer, pre_access, qtype)
             self._write(msg)
         else:

+            if stream_name not in self.streams:
+                self._resultlock.release()
+                logger.error("Attempting to delete not existing stream %s" % stream_name)
+                return

             self._write(VlcMessage.request.stopBroadcast(stream_name))
 
         try:
             self._write(VlcMessage.request.stopBroadcast(stream_name))
 
         try:


@@ -142,20 +152,40 @@ class VlcClient(object):
             logger.error(broadcast + " result timeout")
             raise VlcException(broadcast + " result timeout")
         finally:
             logger.error(broadcast + " result timeout")
             raise VlcException(broadcast + " result timeout")
         finally:

+            logger.info("working with %s stream: %s" % (stream_name,broadcast))
+            if brtype == True:
+                self.streams[stream_name]=time.time()
+            else:
+                del self.streams[stream_name]

             self._resultlock.release()
             self._resultlock.release()

+            logger.info("worked with %s stream: %s" % (stream_name,broadcast))

 
         if brtype == True:
 
         if brtype == True:

-            logger.debug("Broadcast started")
+            logger.info("Broadcast started")

         else:
         else:

-            logger.debug("Broadcast stopped")
+            logger.info("Broadcast stopped")

 
     def startBroadcast(self, stream_name, input, muxer='ts', pre_access='', qtype='default'):
 
     def startBroadcast(self, stream_name, input, muxer='ts', pre_access='', qtype='default'):

-        print "Starting broadcast......"
+        logger = logging.getLogger("VlcClient_startBroadcast")
+        logger.debug("Starting broadcast......")

         return self._broadcast(True, stream_name, input, muxer, pre_access, qtype)
 
     def stopBroadcast(self, stream_name):
         return self._broadcast(False, stream_name)
 
         return self._broadcast(True, stream_name, input, muxer, pre_access, qtype)
 
     def stopBroadcast(self, stream_name):
         return self._broadcast(False, stream_name)
 

+    def mark(self,stream_name):
+      self.streams[stream_name]=time.time()
+
+    def clean_streams(self,timeout=15):
+      self._resultlock.acquire()
+      to_stop=set()
+      for stream,lasttime in self.streams.iteritems():
+        if time.time()-lasttime>timeout:
+          to_stop.add(stream)
+      for stream in to_stop:    
+        self.stopBroadcast(stream)
+      self._resultlock.release()
+

     def pauseBroadcast(self, stream_name):
         return self._write(VlcMessage.request.pauseBroadcast(stream_name))
 
     def pauseBroadcast(self, stream_name):
         return self._write(VlcMessage.request.pauseBroadcast(stream_name))
 


@@ -216,10 +246,10 @@ class VlcClient(object):
                 # Do not move this before error handlers!
                 elif self._recvbuffer.startswith(VlcMessage.response.STARTOK):
                     # Broadcast started
                 # Do not move this before error handlers!
                 elif self._recvbuffer.startswith(VlcMessage.response.STARTOK):
                     # Broadcast started

-                    logger.debug("Broadcast started")
+                    logger.info("Broadcast started")

                     self._result.set(True)
 
                 elif self._recvbuffer.startswith(VlcMessage.response.STOPOK):
                     # Broadcast stopped
                     self._result.set(True)
 
                 elif self._recvbuffer.startswith(VlcMessage.response.STOPOK):
                     # Broadcast stopped

-                    logger.debug("Broadcast stopped")
+                    logger.info("Broadcast stopped")

                     self._result.set(True)
                     self._result.set(True)