o /.]@sTdZddlmZddlmZddlmZddlmZddlZd dd Zd d d ZdS) z0Define the misc handler for InfluxDBClient test.)absolute_import)division)print_function)unicode_literalsN 127.0.0.1c Csg}t}z>t|D]'}t}|dg}|||tjtjd||df|d|d<q W|D] \}}| | |q5n|D] \}}| | |qFw|t |ks]J|S)aDetermine free ports on provided interface. Get `num_ports` free/available ports on the interface linked to the `ip` :param int num_ports: The number of free ports to get :param str ip: The ip on which the ports have to be taken :return: a set of ports number r) setrangesocketappendZ setsockoptZ SOL_SOCKETZ SO_REUSEADDRZbindZ getsocknamecloseaddlen)Z num_portsipZ sock_portsZports_sockZcurportr5/usr/lib/python3/dist-packages/influxdb/tests/misc.pyget_free_ports s(     rcCsLttjtj}z|||f}|s|tj|dkW|S|w)z/Check if given TCP port is open for connection.r)r ZAF_INETZ SOCK_STREAMZ connect_exZshutdownZ SHUT_RDWRr )rrrresultrrr is_port_open)s r)r) __doc__Z __future__rrrrr rrrrrrs