PasswordHackerStage4

import sys
import socket
import itertools
import json

def encode_json(msg):
    return json.dumps(msg)

def decode_json(json_msg):
    return json.loads(json_msg)

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())
        response = client_socket.recv(1024)
        return response.decode()

def get_name():
    lis = get_name_variants()
    for i in lis:
        x = get_server_response(encode_json(f'"login": {i},"password": " "'))
        x = decode_json(x)
        if x == '"result": "Wrong password!"':
            return i

def get_pw():
    pass
    # return pw
login = get_name()
print(encode_json(login))