121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
|
|
||
|
import json
|
||
|
import logging
|
||
|
import requests
|
||
|
from flask import Flask, request
|
||
|
from .settings import *
|
||
|
|
||
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
||
|
class EagleEyev3():
|
||
|
|
||
|
def __init__(self):
|
||
|
self.client_id = None
|
||
|
self.client_secret = None
|
||
|
self.access_token = None
|
||
|
self.refresh_token = None
|
||
|
self.redirect_uri = None
|
||
|
|
||
|
self._load_vars_from_settings()
|
||
|
|
||
|
self.user_base_url = None
|
||
|
self.current_user = None
|
||
|
self.users = []
|
||
|
self.bridges = []
|
||
|
self.cameras = []
|
||
|
self.users = []
|
||
|
self.accounts = []
|
||
|
|
||
|
self.base_url = None
|
||
|
|
||
|
|
||
|
def _load_vars_from_settings(self):
|
||
|
self.client_id = settings.client_id
|
||
|
self.client_secret = settings.client_secret
|
||
|
self.server_host = settings.server_host
|
||
|
self.server_port = settings.server_port
|
||
|
|
||
|
# combine server_protocol, server_host, and server_port to make the redirect_uri
|
||
|
# please see note in settings.py about trailing slashes and modify this line if needed
|
||
|
self.redirect_uri = f"{settings.server_protocol}://{settings.server_host}:{settings.server_port}"
|
||
|
|
||
|
|
||
|
|
||
|
def login_tokens(self, code=None, cascade=True):
|
||
|
baseUrl = "https://auth.eagleeyenetworks.com/oauth2/token"
|
||
|
pathUrl = f"?grant_type=authorization_code&scope=vms.all&code={code}&redirect_uri={self.redirect_uri}" # note the trailing slash, make sure it matches the whitelist
|
||
|
url = baseUrl + pathUrl
|
||
|
|
||
|
response = requests.post(url, auth=(self.client_id, self.client_secret))
|
||
|
response_json = response.json()
|
||
|
|
||
|
logging.info(f"{response.status_code} in login_tokens")
|
||
|
|
||
|
if response.status_code == 200:
|
||
|
success = True
|
||
|
self.access_token = response_json['access_token']
|
||
|
self.refresh_token = response_json['refresh_token']
|
||
|
if cascade:
|
||
|
self.get_base_url()
|
||
|
else:
|
||
|
success = False
|
||
|
|
||
|
return {
|
||
|
"success": success,
|
||
|
"response_http_status": response.status_code,
|
||
|
"data": response_json,
|
||
|
'current_user': self.current_user
|
||
|
}
|
||
|
|
||
|
def get_base_url(self, cascade=True):
|
||
|
url = "https://api.eagleeyenetworks.com/api/v3.0/clientSettings"
|
||
|
headers = {
|
||
|
"Authorization": f"Bearer {self.access_token}",
|
||
|
"Accept": "application/json"
|
||
|
}
|
||
|
|
||
|
response = requests.get(url, headers=headers)
|
||
|
response_json = response.json()
|
||
|
|
||
|
logging.info(f"{response.status_code} in get_base_url")
|
||
|
|
||
|
if response.status_code == 200:
|
||
|
success = True
|
||
|
if 'httpsBaseUrl' in response_json and 'hostname' in response_json['httpsBaseUrl']:
|
||
|
self.user_base_url = response_json['httpsBaseUrl']['hostname']
|
||
|
if cascade:
|
||
|
self.get_current_user()
|
||
|
else:
|
||
|
success = False
|
||
|
|
||
|
return {
|
||
|
"success": success,
|
||
|
"response_http_status": response.status_code,
|
||
|
"data": response_json
|
||
|
}
|
||
|
|
||
|
def get_current_user(self):
|
||
|
url = f"https://{self.user_base_url}/api/v3.0/users/self"
|
||
|
headers = {
|
||
|
"Authorization": f"Bearer {self.access_token}",
|
||
|
"Accept": "application/json"
|
||
|
}
|
||
|
|
||
|
response = requests.get(url, headers=headers)
|
||
|
response_json = response.json()
|
||
|
|
||
|
logging.info(f"{response.status_code} in get_current_user")
|
||
|
|
||
|
if response.status_code == 200:
|
||
|
success = True
|
||
|
self.current_user = response_json
|
||
|
else:
|
||
|
success = False
|
||
|
|
||
|
return {
|
||
|
"success": success,
|
||
|
"response_http_status": response.status_code,
|
||
|
"data": response_json
|
||
|
}
|
||
|
|