87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
|
from CameraModel import CameraModel
|
||
|
|
from CameraCommander import CameraCommander
|
||
|
|
from WaitAuthentication import WaitAuthentication
|
||
|
|
from WaitEnabledModeReply import WaitEnabledModeReply
|
||
|
|
from WaitDisabledModeReply import WaitDisabledModeReply
|
||
|
|
from WaitDisconnectedReply import WaitDisconnectedReply
|
||
|
|
from DeadState import DeadState
|
||
|
|
|
||
|
|
|
||
|
|
class CameraContext:
|
||
|
|
class InvalidStateException(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def __init__(self, connection):
|
||
|
|
self.camera = None
|
||
|
|
self.connection = connection
|
||
|
|
self.state = WaitAuthentication()
|
||
|
|
|
||
|
|
|
||
|
|
def handle(self):
|
||
|
|
return self.state.handle(self)
|
||
|
|
|
||
|
|
|
||
|
|
def tic(self):
|
||
|
|
self.state.tic(self)
|
||
|
|
|
||
|
|
|
||
|
|
def enable(self, userCommand = None):
|
||
|
|
if not self.camera or not self.connection:
|
||
|
|
raise CameraContext.InvalidStateException()
|
||
|
|
if not self.connection.send("enable"):
|
||
|
|
print("[CameraContext] : Connection lost with camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.close()
|
||
|
|
if userCommand:
|
||
|
|
userCommand.callback(False, "connection_lost")
|
||
|
|
return False
|
||
|
|
print("[CameraContext] : Ask for enabling RTMP streaming on camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.state = WaitEnabledModeReply(userCommand)
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def disable(self, userCommand = None):
|
||
|
|
if not self.camera or not self.connection:
|
||
|
|
raise CameraContext.InvalidStateException()
|
||
|
|
if not self.connection.send("disable"):
|
||
|
|
print("[CameraContext] : Connection lost with camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.close()
|
||
|
|
if userCommand:
|
||
|
|
userCommand.callback(False, "connection_lost")
|
||
|
|
return False
|
||
|
|
print("[CameraContext] : Ask for disabling RTMP streaming on camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.state = WaitDisabledModeReply(userCommand)
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def disconnect(self, userCommand = None):
|
||
|
|
if not self.camera or not self.connection:
|
||
|
|
raise CameraContext.InvalidStateException()
|
||
|
|
if not self.connection.send("disconnect"):
|
||
|
|
print("[CameraContext] : Connection lost with camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.close()
|
||
|
|
if userCommand:
|
||
|
|
userCommand.callback(False, "connection_lost")
|
||
|
|
return False
|
||
|
|
print("[CameraContext] : Ask for disconnecting camera ({}) at {}:{}".format(self.camera.id, self.connection.addr, self.connection.port))
|
||
|
|
self.state = WaitDisconnectedReply(userCommand)
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
if not self.connection:
|
||
|
|
raise CameraContext.InvalidStateException()
|
||
|
|
if CameraCommander.IsRegistered(self):
|
||
|
|
CameraCommander.Remove(self)
|
||
|
|
if not self.camera:
|
||
|
|
raise CameraContext.InvalidStateException()
|
||
|
|
mdl = CameraModel()
|
||
|
|
self.camera.available = False
|
||
|
|
self.camera.rtmp_handle = ""
|
||
|
|
mdl.update(self.camera)
|
||
|
|
self.camera = None
|
||
|
|
print("[CameraContext] : Close connection with {}:{}".format(self.connection.addr, self.connection.port))
|
||
|
|
self.connection.close()
|
||
|
|
self.connection = None
|
||
|
|
self.state = DeadState()
|