client.py

import threading
import socket
from datetime import datetime
import os


FORMAT = 'utf-8'    
RESET = True
MAX_TICKETS = int()
MAX_VIP_TICKETS = int()
USERNAME = str()
CUR_PATH = os.path.dirname(__file__)

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 9898))

def reservation():
    global RESET
    while RESET:
        try:
            br_karata = int(input("Unesite broj karata koje zelite da rezervisete: "))
        except ValueError:
            print("Unos mora biti cijeli broj!")
        if br_karata > 0 and br_karata <= 4:
            return br_karata
            break
        if br_karata > 4:
            print("Maksimalan broj karata koje mozete da unesete je 4!")
            RESET = True
        if br_karata <= 0:
            print("Minimalan broj karata koje mozete da rezervisete je 1!")
            RESET = True
        else:
            print("GRESKA")
            client.close()
            break

def recieve():
    global RESET
    global MAX_TICKETS
    while RESET:
        try:
            msg = client.recv(1024).decode(FORMAT)
            if msg[:11] == 'MAX_TICKETS':
                MAX_TICKETS = int(msg[11:])
            elif msg[:3] == 'ANN':
                print(msg[3:])
            elif msg[:4] == 'FANN':
                print(msg[4:])
                date = datetime.now()
                try:
                    f = open(f'client_info\\{USERNAME}_info.txt', "a")
                    f.write(f'{msg[4:]} {date}\n')
                finally:
                    f.close()
        except:
            print("Veza sa serverom je prekinuta")
            client.close()
            input()
            break

def meni():
    global RESET
    client.send('MAX_TICKETS'.encode(FORMAT))
    while RESET:
        choice = input().upper()
        # choice = f'{input("Vas izbor: ")}'
        choice = choice.rstrip()
        if choice.upper() == 'LIST': 
            client.send(choice.encode(FORMAT))
        elif choice.upper() == 'RESERVE':
            client.send('MAX_TICKETS'.encode(FORMAT))
            print(f'Unesite broj karata koje zelite da rezervisete (mozete rezervisati {MAX_TICKETS}):')
            while True:
                try:
                    br = int(input())
                    break
                except:
                    print('Mozete unositi samo brojeve!')
            client.send((choice+str(br)).encode(FORMAT))
            client.send('MAX_TICKETS'.encode(FORMAT))
        elif choice.upper() == 'RESERVE VIP':
            client.send('MAX_TICKETS'.encode(FORMAT))
            print(f'Unesite broj karata koje zelite da rezervisete (mozete rezervisati {MAX_TICKETS}):')
            while True:
                try:
                    br = int(input())
                    break
                except:
                    print('Mozete unositi samo brojeve!')
            client.send((('VIP_RESERVE')+str(br)).encode(FORMAT))
            client.send('MAX_TICKETS'.encode(FORMAT))

        elif choice.upper() == 'CANCEL TICKETS':
            client.send('MAX_TICKETS'.encode(FORMAT))
            print(f'Unesite broj karata koje zelite da otkazete (mozete otkazati {(4-MAX_TICKETS)}):')
            while True:
                try:
                    br = int(input())
                    if(br>=0):
                        break
                    print("Pogresan unos!")
                except:
                    print('Mozete unositi samo brojeve!')
            client.send((('CANCEL_T')+str(br)).encode(FORMAT))
            client.send('MAX_TICKETS'.encode(FORMAT))

        elif choice.upper() == 'CANCEL VIP TICKETS':
            client.send('MAX_TICKETS'.encode(FORMAT))
            print(f'Unesite broj VIP karata koje zelite da otkazete (mozete otkazati {(4-MAX_TICKETS)}):')
            while True:
                try:
                    br = int(input())
                    if(br>=0):
                        break
                    print("Pogresan unos!")
                except:
                    print('Mozete unositi samo brojeve!')
            client.send((('CANCEL_V_T')+str(br)).encode(FORMAT))
            client.send('MAX_TICKETS'.encode(FORMAT))

        elif choice.upper() == 'IZLAZ':
            RESET = False
            input()
            break
        else:
            print("Pogresan unos!")

def register(client):
    global RESET
    global USERNAME
    auth_complete = False
    while True:
        try:
            message_rcvd = client.recv(1024).decode(FORMAT)
            if message_rcvd[:3] == 'ANN':
                print(message_rcvd[3:])
            if message_rcvd[:10] == 'REG_U_NAME':
                print(message_rcvd[10:])
                while True:
                    try:
                        message = input()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('U_NAME'+message).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo:')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()
            
            if message_rcvd[:12] == 'REG_PASSWORD':
                print(message_rcvd[12:])
                while True:
                    try:
                        message = input()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('PASSWORD'+message).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo:')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()
            
            if message_rcvd[:8] == 'REG_NAME':
                print(message_rcvd[8:])
                while True:
                    try:
                        message = input()
                        message = message.rstrip()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('NAME'+message).encode(FORMAT))
                            USERNAME = message
                            break
                        else:
                            print('Greska! Pokusajte ponovo:')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()

            if message_rcvd[:10] == 'REG_L_NAME':
                print(message_rcvd[10:])
                while True:
                    try:
                        message = input()
                        message = message.rstrip()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('L_NAME'+message).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo: ')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()

            if message_rcvd[:8] == 'REG_JMBG':
                print(message_rcvd[8:])
                while True:
                    try:
                        while True:
                            try:
                                message = int(input())
                                break
                            except:
                                print('Mozete unositi samo brojeve!')
                        
                        if (len(str(message)) == 13):
                            client.send(('JMBG'+(str(message))).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo: ')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()

            if message_rcvd[:9] == 'REG_EMAIL':
                print(message_rcvd[9:])
                while True:
                    try:
                        message = input()
                        message = message.rstrip()
                        if ((len(message) > 2) and (len(message) < 30)):
                            client.send(('EMAIL'+message).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo: ')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()

            if message_rcvd[:12] == 'AUTH_SUCCESS':
                auth_complete = True
                break
        except:
            print("Veza sa serverom je prekinuta")
            client.close()
            input()
            break
    if auth_complete:
        recieve_thread = threading.Thread(target=recieve)
        meni_thread = threading.Thread(target=meni)

        recieve_thread.start()
        meni_thread.start()

def login(client):
    global RESET
    global USERNAME
    complete = False
    while True:
        try:
            message_rcvd = client.recv(1024).decode(FORMAT)
            if message_rcvd[:3] == 'ANN':
                print(message_rcvd[3:])

            if message_rcvd[:12] == 'LOGIN_U_NAME':
                print(message_rcvd[12:])
                while True:
                    try:
                        message = input()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('U_NAME'+message).encode(FORMAT))
                            USERNAME = message
                            break
                        else:
                            print('Greska! Pokusajte ponovo:')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()
            
            if message_rcvd[:14] == 'LOGIN_PASSWORD':
                print(message_rcvd[14:])
                while True:
                    try:
                        message = input()
                        if ((len(message) > 2) and (len(message) < 20)):
                            client.send(('PASSWORD'+message).encode(FORMAT))
                            break
                        else:
                            print('Greska! Pokusajte ponovo:')
                    except:
                        print('KONEKCIJA PREKIINUTA')
                        client.close()
            if message_rcvd[:12] == 'AUTH_SUCCESS':
                complete = True
                break
        except:
            print("Veza sa serverom je prekinuta")
            client.close()
            input()
            break
    if complete:
        recieve_thread = threading.Thread(target=recieve)
        meni_thread = threading.Thread(target=meni)

        recieve_thread.start()
        meni_thread.start()

def init_recieve():
    try:
        message_rcvd = client.recv(1024).decode(FORMAT)
        print(message_rcvd)
        
        while True:
            # try:
                message = input()
                message = message.rstrip().lower()
                if (message == 'login'):
                    break
                elif (message == 'register'):
                    break
                elif (message == 'izlaz'):
                    print("Veza sa serverom je prekinuta")
                    client.close()
                    input()
                    return
                else:
                    print('Greska! Pokusajte ponovo: ')
        if message == 'register':
            client.send('REG'.encode(FORMAT))
            register_thread = threading.Thread(target=register, args=(client,))
            register_thread.start()
        elif message == 'login':
            client.send('LOGIN'.encode(FORMAT))
            login_thread = threading.Thread(target=login, args=(client,))
            login_thread.start()
    except:
        print("Veza sa serverom je prekinuta")
        client.close()
        input()
        return

init_recieve()