reflect/client-python
1
0
Fork 0
mirror of https://codeberg.org/reflect/reflect-client-python.git synced 2025-09-14 09:33:41 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: support for self-signed ssl and bugfixes

Browse source
This commit is contained in:
Victor Westerlund 2023-06-05 14:15:02 +02:00
parent 13e7e5301a
commit 112eac5ffe
2 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -15,6 +15,7 @@ Homestead.yaml
npm-debug.log npm-debug.log
yarn-error.log yarn-error.log
public/robots.txt public/robots.txt
__pycache__
# OS generated files # # OS generated files #
###################### ######################

10
reflect_client/Client.py
View file

@ -47,7 +47,9 @@ class Client:
# Close socket connection when class instance removed # Close socket connection when class instance removed
def __del__(self): def __del__(self):
self._socket.close() # Close socket if connected
if (self._con == Connection.AF_UNIX):
self._socket.close()
# Resolve connection type from endpoint string. # Resolve connection type from endpoint string.
# If the string is a valid URL we will treat it as HTTP otherwise we will assume it's a path on disk to a UNIX socket file. # If the string is a valid URL we will treat it as HTTP otherwise we will assume it's a path on disk to a UNIX socket file.
@ -75,9 +77,9 @@ class Client:
# Remove leading "/" if present, as it's already present in self._endpoint # Remove leading "/" if present, as it's already present in self._endpoint
endpoint = endpoint if endpoint[-1] != "/" else endpoint[:-1] endpoint = endpoint if endpoint[-1] != "/" else endpoint[:-1]
resp = requests.request(method.name, self._endpoint + endpoint, headers=self.http_headers(), data=json.dumps(payload)) resp = requests.request(method.name, self._endpoint + endpoint, verify=self._https_peer_verify, headers=self.http_headers(), data=json.dumps(payload))
# Return response as tuple of response code and response body as plain text # Return response as tuple of response code and response body as decoded JSON (mixed)
return (resp.status_code, resp.text) return (resp.status_code, json.loads(resp.text))
def socket_txn(self, endpoint: str, method: Union[Method, str] = None, payload: list = None) -> tuple: def socket_txn(self, endpoint: str, method: Union[Method, str] = None, payload: list = None) -> tuple:
data = f'["{endpoint}","{method.name}","{json.dumps(payload)}"]' data = f'["{endpoint}","{method.name}","{json.dumps(payload)}"]'