async def _validate_and_create_auth(self, data):
"""Try to login to ezviz cloud account and create entry if successful."""
await self.async_set_unique_id(data[CONF_USERNAME])
self._abort_if_unique_id_configured()
client = EzvizClient(
data[CONF_USERNAME],
data[CONF_PASSWORD],
data.get(CONF_REGION, DEFAULT_REGION),
data.get(CONF_TIMEOUT, DEFAULT_TIMEOUT),
)
# Attempts a login request.
try:
await self.hass.async_add_executor_job(client.login)
except (PyEzvizError, HTTPError) as err:
raise InvalidAuth from err
auth_data = {
CONF_USERNAME: data[CONF_USERNAME],
CONF_PASSWORD: data[CONF_PASSWORD],
CONF_REGION: data.get(CONF_REGION, DEFAULT_REGION),
CONF_TYPE: ATTR_TYPE_CLOUD,
}
return self.async_create_entry(title=data[CONF_USERNAME],
data=auth_data)
async def _validate_and_create_auth(self, data):
"""Validate the user input allows us to connect.
Data has the keys from data_schema with values provided by the user.
"""
await self.async_set_unique_id(data[CONF_USERNAME])
self._abort_if_unique_id_configured()
# constructor does login call
client = EzvizClient(
data[CONF_USERNAME],
data[CONF_PASSWORD],
data.get(CONF_REGION, DEFAULT_REGION),
data.get(CONF_TIMEOUT, DEFAULT_TIMEOUT),
)
# Validate data by sending a login request to ezviz
try:
await self.hass.async_add_executor_job(client.login)
except PyEzvizError as err:
raise InvalidAuth from err
auth_data = {
"username": data[CONF_USERNAME],
"password": data[CONF_PASSWORD],
"region": data.get(CONF_REGION, DEFAULT_REGION),
}
return self.async_create_entry(title=data[CONF_USERNAME], data=auth_data)
def ezviz_config_flow(hass):
"""Mock the EzvizAPI for easier config flow testing."""
with patch.object(EzvizClient, "login", return_value=True), patch(
"homeassistant.components.ezviz.config_flow.EzvizClient"
) as mock_ezviz:
instance = mock_ezviz.return_value = EzvizClient(
"test-username",
"test-password",
"test-region",
"test-timeout",
)
instance.login = MagicMock(return_value=True)
yield mock_ezviz
def ezviz_config_flow(hass):
"""Mock the EzvizAPI for easier config flow testing."""
with patch.object(EzvizClient, "login", return_value=True), patch(
"homeassistant.components.ezviz.config_flow.EzvizClient"
) as mock_ezviz:
instance = mock_ezviz.return_value = EzvizClient(
"test-username",
"test-password",
"local.host",
"1",
)
instance.login = MagicMock(return_value=ezviz_login_token_return)
instance.get_detection_sensibility = MagicMock(return_value=True)
yield mock_ezviz
def main():
"""Main function"""
parser = argparse.ArgumentParser(prog='pyezviz')
parser.add_argument('-u',
'--username',
required=True,
help='Ezviz username')
parser.add_argument('-p',
'--password',
required=True,
help='Ezviz Password')
parser.add_argument('--debug',
'-d',
action='store_true',
help='Print debug messages to stderr')
subparsers = parser.add_subparsers(dest='action')
parser_device = subparsers.add_parser('devices',
help='Play with all devices at once')
parser_device.add_argument(
'device_action',
type=str,
default='status',
help='Device action to perform',
choices=['device', 'status', 'switch', 'connection', 'switch-all'])
parser_camera = subparsers.add_parser('camera', help='Camera actions')
parser_camera.add_argument('--serial', required=True, help='camera SERIAL')
subparsers_camera = parser_camera.add_subparsers(dest='camera_action')
parser_camera_status = subparsers_camera.add_parser(
'status', help='Get the status of the camera')
# parser_camera_status.add_argument('--status', required=True, help='Status to status', choices=['device','camera','switch','connection','wifi','status'])
parser_camera_move = subparsers_camera.add_parser('move',
help='Move the camera')
parser_camera_move.add_argument('--direction',
required=True,
help='Direction to move the camera to',
choices=['up', 'down', 'right', 'left'])
parser_camera_move.add_argument('--speed',
required=False,
help='Speed of the movement',
default=5,
type=int,
choices=range(1, 10))
parser_camera_switch = subparsers_camera.add_parser(
'switch', help='Change the status of a switch')
parser_camera_switch.add_argument(
'--switch',
required=True,
help='Switch to switch',
choices=['audio', 'ir', 'state', 'privacy', 'follow_move'])
parser_camera_switch.add_argument('--enable',
required=False,
help='Enable (or not)',
default=1,
type=int,
choices=[0, 1])
parser_camera_alarm = subparsers_camera.add_parser(
'alarm', help='Configure the camera alarm')
parser_camera_alarm.add_argument('--notify',
required=False,
help='Enable (or not)',
default=0,
type=int,
choices=[0, 1])
parser_camera_alarm.add_argument(
'--sound',
required=False,
help='Sound level (2 is disabled, 1 intensive, 0 software)',
type=int,
choices=[0, 1, 2])
parser_camera_alarm.add_argument('--sensibility',
required=False,
help='Sensibility level (form 1 to 6)',
default=3,
type=int,
choices=[0, 1, 2, 3, 4, 5, 6])
args = parser.parse_args()
# print("--------------args")
# print("--------------args: %s",args)
# print("--------------args")
client = EzvizClient(args.username, args.password)
if args.debug:
import http.client
http.client.HTTPConnection.debuglevel = 5
# You must initialize logging, otherwise you'll not see debug output.
logging.basicConfig()
logging.getLogger().setLevel(logging.DEBUG)
requests_log = logging.getLogger("requests.packages.urllib3")
requests_log.setLevel(logging.DEBUG)
requests_log.propagate = True
if args.action == 'devices':
if args.device_action == 'device':
try:
client.login()
print(json.dumps(client.get_DEVICE(), indent=2))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
if args.device_action == 'status':
try:
client.login()
# print(json.dumps(client.load_cameras(), indent=2))
print(pandas.DataFrame(client.load_cameras()))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
if args.device_action == 'switch':
try:
client.login()
print(json.dumps(client.get_SWITCH_STATUS(), indent=2))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.device_action == 'connection':
try:
client.login()
print(json.dumps(client.get_CONNECTION(), indent=2))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.device_action == 'switch-all':
try:
client.login()
print(json.dumps(client.switch_devices(args.enable), indent=2))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.action == 'camera':
# load camera object
try:
client.login()
camera = EzvizCamera(client, args.serial)
logging.debug("Camera loaded")
except BaseException as exp:
print(exp)
return 1
# if args.camera_action == 'list':
# try:
# pagelist = client.get_PAGE_LIST()
# df = pandas.DataFrame(pagelist['statusInfos'])
# df
# except BaseException as exp:
# print(exp)
# return 1
# finally:
# client.close_session()
if args.camera_action == 'move':
try:
camera.move(args.direction, args.speed)
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.camera_action == 'status':
try:
# camera.load()
# if args.status == 'device':
# print(camera._device)
# elif args.status == 'status':
# print(camera._status)
# elif args.status == 'switch':
# # print(json.dumps(camera._switch, indent=2))
# print(camera._switch)
# elif args.status == 'connection':
# # print(json.dumps(camera._switch, indent=2))
# print(camera._connection)
# elif args.status == 'wifi':
# # print(json.dumps(camera._switch, indent=2))
# print(camera._wifi)
# print(camera.status())
print(json.dumps(camera.status(), indent=2))
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.camera_action == 'switch':
try:
if args.switch == 'ir':
camera.switch_device_ir_led(args.enable)
elif args.switch == 'state':
camera.switch_device_state_led(args.enable)
elif args.switch == 'audio':
camera.switch_device_audio(args.enable)
elif args.switch == 'privacy':
camera.switch_privacy_mode(args.enable)
elif args.switch == 'follow_move':
camera.switch_follow_move(args.enable)
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
elif args.camera_action == 'alarm':
try:
if args.sound:
camera.alarm_sound(args.sound)
if args.notify:
camera.alarm_notify(args.notify)
if args.sensibility:
camera.alarm_detection_sensibility(args.sensibility)
except BaseException as exp:
print(exp)
return 1
finally:
client.close_session()
else:
print("Action not implemented: %s", args.action)