def add_user(config_uri, username, password_plain, title="default", desc="",
flags=0, cellphone="", email="", user_type=USER_TYPE_NORMAL):
config = configparser.ConfigParser()
config.read(config_uri)
settings = dict(config.items("alembic"))
engine = engine_from_config(settings, 'sqlalchemy.')
if is_str(password_plain):
password_plain = password_plain.encode()
password_hex = STRING(binascii.hexlify(pbkdf2(bytes(password_plain),
PASSWORD_PBKDF2_HMAC_SHA256_SALT,
10000,
keylen=32,
prf=str('hmac-sha256'))))
dao_context_mngr = AlchemyDaoContextMngr(engine)
user_dao = SAUserDao(dao_context_mngr)
user_mngr = UserManager(user_dao=user_dao,
project_dao=None,
dao_context_mngr=dao_context_mngr)
user_mngr.add_user(username=username,
password=password_hex,
title=title,
desc=desc,
flags=flags,
cellphone=cellphone,
email=email,
user_type=user_type)
def remove_user(config_uri, username):
config = configparser.ConfigParser()
config.read(config_uri)
settings = dict(config.items("alembic"))
engine = engine_from_config(settings, 'sqlalchemy.')
dao_context_mngr = AlchemyDaoContextMngr(engine)
user_dao = SAUserDao(dao_context_mngr)
user_mngr = UserManager(user_dao=user_dao,
project_dao=None,
dao_context_mngr=dao_context_mngr)
user_mngr.delete_user_by_name(username)
def remove_user(config_uri, username):
config = configparser.ConfigParser()
config.read(config_uri)
settings = dict(config.items("alembic"))
engine = engine_from_config(settings, 'sqlalchemy.')
dao_context_mngr = AlchemyDaoContextMngr(engine)
user_dao = SAUserDao(dao_context_mngr)
user_mngr = UserManager(user_dao=user_dao,
project_dao=None,
dao_context_mngr=dao_context_mngr)
user_mngr.delete_user_by_name(username)
def add_user(config_uri,
username,
password_plain,
title="default",
desc="",
flags=0,
cellphone="",
email="",
user_type=User.USER_TYPE_NORMAL):
config = configparser.ConfigParser()
config.read(config_uri)
settings = dict(config.items("alembic"))
engine = engine_from_config(settings, 'sqlalchemy.')
if is_str(password_plain):
password_plain = password_plain.encode()
password_hex = STRING(
binascii.hexlify(
pbkdf2(password_plain,
PASSWORD_PBKDF2_HMAC_SHA256_SALT,
100000,
keylen=32,
prf=str('hmac-sha256'))))
dao_context_mngr = AlchemyDaoContextMngr(engine)
user_dao = SAUserDao(dao_context_mngr)
user_mngr = UserManager(user_dao=user_dao,
project_dao=None,
dao_context_mngr=dao_context_mngr)
user_mngr.add_user(username=username,
password=password_hex,
title=title,
desc=desc,
flags=flags,
cellphone=cellphone,
email=email,
user_type=user_type)
def main():
default_log_config(debug=True)
import logging
log = logging.getLogger(__name__)
try:
if len(sys.argv) == 2:
config = parse_conf(sys.argv[1])
else:
config = parse_conf('ivc.yml')
config = config_schema.validate(config)
# prepare data backend
if config.get('sqlalchemy'):
from sqlalchemy import engine_from_config
from ivr.ivc.daos.sa_dao_context_mngr import AlchemyDaoContextMngr
from ivr.ivc.daos.sa_camera_dao import SACameraDao
from ivr.ivc.daos.sa_device_dao import SADeviceDao
from ivr.ivc.daos.sa_project_dao import SAProjectDao
from ivr.ivc.daos.sa_user_dao import SAUserDao
from ivr.ivc.daos.sa_user_session_log_dao import SAUserSessionLogDao
from ivr.ivc.daos.sa_access_key_dao import SAAccessKeyDao
engine = engine_from_config(config['sqlalchemy'], prefix='')
dao_context_mngr = AlchemyDaoContextMngr(engine)
project_dao = SAProjectDao(dao_context_mngr)
camera_dao = SACameraDao(dao_context_mngr)
device_dao = SADeviceDao(dao_context_mngr)
user_dao = SAUserDao(dao_context_mngr)
access_key_dao = SAAccessKeyDao(dao_context_mngr)
from ivr.ivc.dummy_daos import UserSessionDAO, StreamDAO
user_session_dao = UserSessionDAO()
user_session_log_dao = SAUserSessionLogDao(dao_context_mngr)
stream_dao = StreamDAO()
else:
from ivr.ivc.dummy_daos import CameraDAO, UserSessionDAO, StreamDAO, DeviceDAO
camera_dao = CameraDAO()
stream_dao = StreamDAO()
user_session_dao = UserSessionDAO()
device_dao = DeviceDAO()
user_dao = None
access_key_dao = None
project_mngr = ProjectManager(project_dao)
if not config['ws_listen']:
from ivr.ivc.manager.device import DummyDeviceManager
device_mngr = DummyDeviceManager(device_dao)
else:
device_mngr = DeviceWSConnectionManager(device_dao,
config['device_ttl'])
camera_mngr = CameraManager(camera_dao, device_mngr)
device_mngr.set_camera_mngr(camera_mngr)
stream_mngr = StreamManager(stream_dao,
camera_mngr,
config['rtmp_publish_url_prefix'],
config['rtmp_url_prefix'],
config['hls_url_prefix'],
stream_ttl=config['stream_ttl'])
user_session_log_mngr = UserSessionLogManager(user_session_log_dao)
user_session_mngr = UserSessionManager(user_session_dao, stream_mngr,
user_session_log_mngr,
config['user_session_ttl'])
user_mngr = UserManager(user_dao, project_dao, dao_context_mngr)
access_key_mngr = AccessKeyManager(access_key_dao, user_dao,
dao_context_mngr)
# prepare REST API
from pyramid.config import Configurator
from pyramid.renderers import JSON
from pyramid.authorization import ACLAuthorizationPolicy
from ivr.common.utils import CustomJSONEncoder
from ivr.ivc.rest.security import IvcJwtAuthenticationPolicy
authn_policy = IvcJwtAuthenticationPolicy()
authz_policy = ACLAuthorizationPolicy()
pyramid_config = Configurator()
pyramid_config.add_renderer(
None, JSON(indent=4, check_circular=True, cls=CustomJSONEncoder))
# pyramid_config.set_authentication_policy(authn_policy)
# pyramid_config.set_authorization_policy(authz_policy)
pyramid_config.include('ivr.ivc.rest', route_prefix='api/ivc/v1')
pyramid_config.registry.project_mngr = project_mngr
pyramid_config.registry.device_mngr = device_mngr
pyramid_config.registry.camera_mngr = camera_mngr
pyramid_config.registry.stream_mngr = stream_mngr
pyramid_config.registry.user_session_mngr = user_session_mngr
pyramid_config.registry.user_session_log_mngr = user_session_log_mngr
pyramid_config.registry.user_mngr = user_mngr
pyramid_config.registry.access_key_mngr = access_key_mngr
if config.get('debug'):
pyramid_config.add_settings({
'debugtoolbar.hosts': ['0.0.0.0/0', '::1'],
'debugtoolbar.enabled':
True,
'pyramid.debug_authorization':
False,
})
pyramid_config.include('pyramid_debugtoolbar')
pyramid_config.include('pyramid_chameleon')
rest_server = WSGIServer(config['rest_listen'],
pyramid_config.make_wsgi_app())
# start server and wait
if config['ws_listen']:
ws_server = WSServer(config['ws_listen'],
device_mngr.device_online)
gevent.joinall(
map(gevent.spawn,
(ws_server.server_forever, rest_server.serve_forever)))
else:
gevent.joinall(map(gevent.spawn, (rest_server.serve_forever, )))
log.info("Quit")
except Exception:
log.exception("Failed to start IVC")
