mirror of
https://github.com/ProtoThis/python-synology.git
synced 2026-01-19 09:28:42 +00:00
0622bcf560
* API discovery (SYNO.API.Info) + use Exceptions - remove dsm_version - use SynologyDSMRequestException for failed requests * _discover_apis any time * simplify requests - Add api.request(api, method, params=None) * Add connection failed tests * Finalized ? * Fix SynologyDSMRequestException not beeing SynologyDSMException * Fix SynologyDSMAPINotExistsException not beeing SynologyDSMException * Add Virtual Machine Manager API documentation * Can get and post request - Add error handling by API code * Pylint * Add get + post tests + SynologyDSMAPIErrorException
151 lines
4.6 KiB
Python
151 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Library tests."""
|
|
import six
|
|
from requests.exceptions import ConnectionError as ConnError, RequestException, SSLError
|
|
from simplejson.errors import JSONDecodeError
|
|
|
|
from synology_dsm import SynologyDSM
|
|
from synology_dsm.exceptions import SynologyDSMRequestException
|
|
from synology_dsm.api.core.utilization import SynoCoreUtilization
|
|
from synology_dsm.api.dsm.information import SynoDSMInformation
|
|
from synology_dsm.api.storage.storage import SynoStorage
|
|
|
|
from .const import (
|
|
ERROR_INSUFFICIENT_USER_PRIVILEGE,
|
|
ERROR_AUTH_INVALID_CREDENTIALS,
|
|
ERROR_AUTH_OTP_AUTHENTICATE_FAILED,
|
|
DEVICE_TOKEN,
|
|
)
|
|
from .api_data.dsm_6 import (
|
|
DSM_6_API_INFO,
|
|
DSM_6_AUTH_LOGIN,
|
|
DSM_6_AUTH_LOGIN_2SA,
|
|
DSM_6_AUTH_LOGIN_2SA_OTP,
|
|
DSM_6_DSM_INFORMATION,
|
|
DSM_6_CORE_UTILIZATION,
|
|
DSM_6_STORAGE_STORAGE,
|
|
)
|
|
|
|
if six.PY2:
|
|
from future.moves.urllib.parse import urlencode
|
|
else:
|
|
from urllib.parse import urlencode # pylint: disable=import-error,no-name-in-module
|
|
|
|
VALID_HOST = "nas.mywebsite.me"
|
|
VALID_PORT = "443"
|
|
VALID_SSL = True
|
|
VALID_USER = "valid_user"
|
|
VALID_USER_2SA = "valid_user_2sa"
|
|
VALID_PASSWORD = "valid_password"
|
|
VALID_OTP = "123456"
|
|
|
|
|
|
class SynologyDSMMock(SynologyDSM):
|
|
"""Mocked SynologyDSM."""
|
|
|
|
API_URI = "entry.cgi"
|
|
|
|
def __init__(
|
|
self,
|
|
dsm_ip,
|
|
dsm_port,
|
|
username,
|
|
password,
|
|
use_https=False,
|
|
device_token=None,
|
|
debugmode=False,
|
|
):
|
|
SynologyDSM.__init__(
|
|
self,
|
|
dsm_ip,
|
|
dsm_port,
|
|
username,
|
|
password,
|
|
use_https,
|
|
device_token,
|
|
debugmode,
|
|
)
|
|
|
|
def _execute_request(self, method, url, **kwargs):
|
|
url += urlencode(kwargs["params"])
|
|
if "no_internet" in url:
|
|
raise SynologyDSMRequestException(
|
|
ConnError(
|
|
"<urllib3.connection.VerifiedHTTPSConnection object at 0x106c1f250>: Failed to establish a new connection: [Errno 8] nodename nor servname provided, or not known"
|
|
)
|
|
)
|
|
|
|
if VALID_HOST not in url:
|
|
raise SynologyDSMRequestException(
|
|
ConnError(
|
|
"<urllib3.connection.HTTPConnection object at 0x10d6f8090>: Failed to establish a new connection: [Errno 8] nodename nor servname provided, or not known"
|
|
)
|
|
)
|
|
|
|
if VALID_PORT not in url and "https" not in url:
|
|
raise SynologyDSMRequestException(
|
|
JSONDecodeError("Expecting value", "<html>document</html>", 0, None)
|
|
)
|
|
|
|
if VALID_PORT not in url:
|
|
raise SynologyDSMRequestException(
|
|
SSLError(
|
|
"[SSL: WRONG_VERSION_NUMBER] wrong version number (_ssl.c:1076)"
|
|
)
|
|
)
|
|
|
|
if "https" not in url:
|
|
raise SynologyDSMRequestException(RequestException("Bad request"))
|
|
|
|
if self.API_INFO in url:
|
|
return DSM_6_API_INFO
|
|
|
|
if self.API_AUTH in url:
|
|
if VALID_USER_2SA in url and VALID_PASSWORD in url:
|
|
if "otp_code" not in url and "device_id" not in url:
|
|
return DSM_6_AUTH_LOGIN_2SA
|
|
|
|
if "device_id" in url and DEVICE_TOKEN in url:
|
|
return DSM_6_AUTH_LOGIN
|
|
|
|
if "otp_code" in url:
|
|
if VALID_OTP in url:
|
|
return DSM_6_AUTH_LOGIN_2SA_OTP
|
|
return ERROR_AUTH_OTP_AUTHENTICATE_FAILED
|
|
|
|
if VALID_USER in url and VALID_PASSWORD in url:
|
|
return DSM_6_AUTH_LOGIN
|
|
|
|
return ERROR_AUTH_INVALID_CREDENTIALS
|
|
|
|
if self.API_URI in url:
|
|
if not self._session_id or not self._syno_token:
|
|
return ERROR_INSUFFICIENT_USER_PRIVILEGE
|
|
|
|
if SynoDSMInformation.API_KEY in url:
|
|
return DSM_6_DSM_INFORMATION
|
|
|
|
if SynoCoreUtilization.API_KEY in url:
|
|
return DSM_6_CORE_UTILIZATION
|
|
|
|
if SynoStorage.API_KEY in url:
|
|
return DSM_6_STORAGE_STORAGE
|
|
|
|
if (
|
|
"SYNO.FileStation.Upload" in url
|
|
and "upload" in url
|
|
and "file_already_exists" in kwargs["files"]["file"]
|
|
):
|
|
return {"error": {"code": 1805}, "success": False}
|
|
|
|
if (
|
|
"SYNO.DownloadStation2.Task" in url
|
|
and "create" in url
|
|
and "test_not_exists" in url
|
|
):
|
|
return {"error": {"code": 408}, "success": False}
|
|
|
|
return {"success": False}
|
|
|
|
return None
|