From 4cd1edd59901c7f8f02d438ce64612ac5177ca4a Mon Sep 17 00:00:00 2001 From: checktheroads Date: Sun, 19 Apr 2020 00:23:56 -0700 Subject: [PATCH] cleanup --- hyperglass/external/_base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/hyperglass/external/_base.py b/hyperglass/external/_base.py index 9630b22..9a3f791 100644 --- a/hyperglass/external/_base.py +++ b/hyperglass/external/_base.py @@ -3,9 +3,9 @@ # Standard Library import re import json as _json +import socket import asyncio from json import JSONDecodeError -import socket from socket import gaierror # Third Party @@ -81,6 +81,7 @@ class BaseExternal: return True def __enter__(self): + """Test connection on entry.""" available = self._test() if available: @@ -90,6 +91,7 @@ class BaseExternal: raise self._exception(f"Unable to create session to {self.name}") def __exit__(self, exc_type=None, exc_value=None, traceback=None): + """Close connection on exit.""" if exc_type is not None: log.error(traceback) self._session.close() @@ -139,6 +141,7 @@ class BaseExternal: return False def build_request(self, **kwargs): + """Process requests parameters into structure usable by http library.""" from operator import itemgetter supported_methods = ("GET", "POST", "PUT", "DELETE", "HEAD", "PATCH")