server.py

import socket
import time
import _thread
from utils import *
from dataqueries import *
import mysql.connector
from mysql.connector import Error
from datetime import datetime
import re


# server configurations
server_ip = socket.gethostname()
server_port = 12345
MAX_USERS =1024


# Databases functions

def create_db_connection(host_name, user_name, user_password, db_name):
    connection = None
    try:
        connection = mysql.connector.connect(
            host=host_name,
            user=user_name,
            passwd=user_password,
            database=db_name
        )
        print("MySQL Database connection successful")
    except Error as err:
        print(f"Error: '{err}'")

    return connection


# create database connection object
MySQL_hostname = 'localhost'
MySQL_username = 'root'
MySQL_password = 'mysqlpwd'
db_conn = create_db_connection(MySQL_hostname, MySQL_username, MySQL_password, "Tweeter")


#CLI
def home_page(db_conn, user, client_socket):
    while(True):
        client_socket.send(
        """
        Hi!!!!!
        Options: (Reply with)
        1: News feed (Recent Tweets/Updates)
        2: Your Pinned Tweets
        3: Followers List
        4: Following List
        5: Search a registered user 
        6: View Tweets by hashtags (Top 5 trending will be shown)
        7: Post a Tweet (max 256 characters)
        8: Your Profile
        9: Pin a Tweet
        10: Chat Box
        0: Logout
        """.encode())
        # time.sleep(0.5)
        option = client_socket.recv(1024).decode()
        print(option)
        if (option == "1"):
            news_feed(db_conn, client_socket, user) 
        # Display another menu (Enter the tweet ID of the tweet to Retweet) to user when he chooses option 1
        
        elif (option == "2"):
            pinned_tweets_list(db_conn, client_socket, user) 
        
        elif (option == "3"):
            followers_list(db_conn, client_socket, user)
            # show active/online followers
            # give user a chance to delete a follower

        elif (option == "4"):
            following_list(db_conn, client_socket, user)
            # give him option to unfollow

        elif (option == "5"):
            search_registered_users(db_conn, client_socket, user)
            # allow him to follow/unfollow that user

        elif (option == '6'):
            hashtags_and_tweets(db_conn, client_socket)
        
        elif (option == "7"):
            post_tweet(db_conn, client_socket, user)

        elif (option == "8"):
            user_profile(db_conn, client_socket, user)
        
        elif (option == "9"):
            pin_tweet(db_conn, client_socket, user)
            
        elif (option == "10"):
            chat_box(db_conn, client_socket, user)
        
        elif (option == "0"):
            return
        
        else:
            client_socket.send("Invalid Option!".encode())
    return
    

#client thread

def client_thread(db_conn, client_socket, address):
    user = authenticate(db_conn, client_socket) 
    # mark_user_online(db_conn, user)
    home_page(db_conn, user, client_socket) 
    logout(db_conn, user)
    print(f"Closing client thread: {address}")
    client_socket.send("Thanks for using Tweeter. See you soon!".encode())
    client_socket.shutdown(2)
    client_socket.close()
    return

#create TCP socket
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((server_ip, server_port))
server_socket.listen(MAX_USERS)

print("Server started and listening...")

while True:
    client_socket, address = server_socket.accept()
    print(f"New client connected: {address}")

    _thread.start_new_thread(client_thread, (db_conn, client_socket, address))