server.py

#!/usr/bin/env python3

import socket
import threading
import ports
import sys
import psycopg2

"""Implements the messaging servers. 

    The server first connects with the auth_server. Then it connects with the msg_storage and the groupdb database.

    Then it waits for connections redirectied from the auth_server and creates separate threads for each connection. 
"""

if len(sys.argv) != 3:
    print("Usage: python3 server.py <server_id> <port>")
    exit(-1)

from client_handler import *

ServerSideSocket = socket.socket()
host = ports.server_host
id = int(sys.argv[1])
port = int(sys.argv[2])
ThreadCount = 0
auth_host = ports.auth_server_host
auth_port = ports.auth_server_port

try:
    ServerSideSocket.bind((host, port))
except socket.error as e:
    print(f'Fatal Error: {e}')
    exit(-1)
print('Socket is listening...')
ServerSideSocket.listen(1)
authInterface = threading.Thread(target=client_handler.authServerInterface, args=(auth_host, auth_port, id, port))
authInterface.start()
# print('AB')
# client_handler.authServerInterface(auth_host, auth_port, id, port)
# print('CD')
# message_distributor = threading.Thread(target = client_handler.distribute_messages)
# message_distributor.start()

sql_msg_conn = psycopg2.connect(database="msg_storage", user="ananth", password="ananth", host="127.0.0.1", port =  "5432")

sql_grp_conn = psycopg2.connect(database="groupdb", user="ananth", password="ananth", host="127.0.0.1", port =  "5432")

try:
    while True:
        #we cannot have a recv statement inside this loop
        #else the server stops accepting connections till the previous connection receives data 
        # print('HELLO')
        Client, address = ServerSideSocket.accept()
        # print('Hello')
        t = threading.Thread(target = client_handler.getClientName, args = (Client, sql_msg_conn, sql_grp_conn))
        t.start()
        # name = bytes.decode(Client.recv(1024))
        # print('Connected to: ' + address[0] + ':' + str(address[1]))
        # obj = client_handler(name)
        # t = Thread(target = client_handler.multi_threaded_client, args = (obj, Client))
        # t.start()
        # t1 = Thread(target = client_handler.send_messages, args = (obj, ))
        # t1.start()
        # client_handler.active_threads[name] = (obj, t, t1, Client)
        # print(client_handler.active_threads)
        ThreadCount += 1
        # print('Thread Number: ' + str(ThreadCount))
finally:
    ServerSideSocket.close()