connection.py

import socket
import log
import manager
import time
dns = {}


class Client():

    def __init__(self, host, port):
        global dns
        self.host = host
        self.ip = socket.gethostbyname(host)
        dns[self.ip] = host
        self.port = int(port)
        self.conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.conn.setblocking(1)

    def connect(self):
        self.conn.connect((self.ip, self.port))
        return self.conn

    def send(self, msg):
        self.conn.sendall(msg.encode())
        data = self.conn.recv(1024).decode()
        return data

    def close(self):
        self.conn.close()


class Server():

    def __init__(self, host, port):
        self.host = host
        self.port = int(port)
        self.conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.conn.setblocking(1)
        self.conn.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.log = log.Log('listener')
        self.conn.bind((self.host, self.port))

    def listen(self):
        self.log.info('waiting for new connections on %s:%s' %
                      (self.host, self.port))
        self.conn.listen(100)
        try:
            current_conn, addr = self.conn.accept()
        except InterruptedError:
            return False
        current_conn.setblocking(1)
        return current_conn

    def recive(self):
        data = self.current_conn.recv(1024).decode()
        return data

    def send(self, msg):
        self.current_conn.sendall(msg.encode())
        data = self.current_conn.recv(1024).decode()
        return data

    def close(self):
        self.conn.shutdown(0)
        self.conn.close()