Add Node creation with a URL string

Author: CodeSandwich

typesense/configuration.py

@@ -1,8 +1,18 @@
 from .exceptions import ConfigError
 from .logger import logger
-
+from urllib.parse import urlparse
 
 class Node(object):
+    def __init__(self, url):
+        parsed = urlparse(url);
+        if not parsed.hostname:
+            raise ConfigError('Node URL does not contain the host name.')
+        if not parsed.port:
+            raise ConfigError('Node URL does not contain the port.')
+        if not parsed.scheme:
+            raise ConfigError('Node URL does not contain the protocol.')
+        self.__init__(parsed.hostname, parsed.port, parsed.path, parsed.scheme)
+
     def __init__(self, host, port, path, protocol):
         self.host = host
         self.port = port
@@ -21,19 +31,21 @@ def __init__(self, config_dict):
         Configuration.show_deprecation_warnings(config_dict)
         Configuration.validate_config_dict(config_dict)
 
-        node_dicts = config_dict.get('nodes', [])
-
         self.nodes = []
-        for node_dict in node_dicts:
-            self.nodes.append(
-                Node(node_dict['host'], node_dict['port'], node_dict.get('path', ''), node_dict['protocol'])
-            )
+        for node_config in config_dict.get('nodes', []):
+            if isinstance(node_config, str):
+                node = Node(node_config)
+            else:
+                node = Node(node_config['host'], node_config['port'], node_config.get('path', ''), node_config['protocol'])
+            self.nodes.append(node)
 
         nearest_node = config_dict.get('nearest_node', None)
-        if nearest_node:
-            self.nearest_node = Node(nearest_node['host'], nearest_node['port'], nearest_node.get('path', ''), nearest_node['protocol'])
-        else:
+        if not nearest_node:
             self.nearest_node = None
+        else if isinstance(nearest_node, str):
+            self.nearest_node = Node(nearest_node)
+        else:
+            self.nearest_node = Node(nearest_node['host'], nearest_node['port'], nearest_node.get('path', ''), nearest_node['protocol'])
 
         self.api_key = config_dict.get('api_key', '')
         self.connection_timeout_seconds = config_dict.get('connection_timeout_seconds', 3.0)
@@ -53,16 +65,18 @@ def validate_config_dict(config_dict):
 
         for node in nodes:
             if not Configuration.validate_node_fields(node):
-                raise ConfigError('`node` entry must be a dictionary with the following required keys: '
+                raise ConfigError('`node` entry must be a URL string or a dictionary with the following required keys: '
                                   'host, port, protocol')
 
         nearest_node = config_dict.get('nearest_node', None)
         if nearest_node and not Configuration.validate_node_fields(nearest_node):
-            raise ConfigError('`nearest_node` entry must be a dictionary with the following required keys: '
+            raise ConfigError('`nearest_node` entry must be a URL string or a dictionary with the following required keys: '
                                   'host, port, protocol')
 
     @staticmethod
     def validate_node_fields(node):
+        if isinstance(node, str):
+            return true
         expected_fields = {'host', 'port', 'protocol'}
         return expected_fields.issubset(node)
 
@@ -76,4 +90,3 @@ def show_deprecation_warnings(config_dict):
 
         if config_dict.get('read_replica_nodes'):
             logger.warn('Deprecation warning: read_replica_nodes is now consolidated to nodes, starting with Typesense Server v0.12')
-