MineEVECraft/main.py

import time
import random
import logging

import numpy as np
import cv2
import pygetwindow as gw
import pyautogui as pg
from pyclick import HumanClicker
import easyocr
import pandas as pd

debug = True

win_name = "EVE - Nostrom Stone"

logging.basicConfig(
    level=logging.DEBUG,
    format="%(asctime)s - %(message)s",
    datefmt="%d-%b-%y %H:%M:%S",
    handlers=[logging.StreamHandler()],
)

logging.info("Бот: запущен")

reader = easyocr.Reader(["en"], gpu=True)
logging.info("Бот: модели загружены")

hc = HumanClicker()


def move_mouse(x, y):
    hc.move(
        (x, y - 3),
        random.uniform(0.1, 0.4),
    )


def click_mouse(x, y, right=False):
    move_mouse(x, y)
    if right:
        pg.click(button="right")
    else:
        pg.click()
        time.sleep(random.uniform(0.2, 0.8))
        move_mouse(
            x - random.randrange(500, 700), y - random.randrange(50, 150)
        )


def get_screenshot():
    eve_window = gw.getWindowsWithTitle(win_name)[0]

    was_minimized = False
    if eve_window.isMinimized:
        was_minimized = True

    eve_window.maximize()
    eve_window.activate()

    if debug:
        filepath = "images/screenshot.png"
    else:
        filepath = None

    time.sleep(0.5)
    screenshot = pg.screenshot(
        filepath,
        region=(
            eve_window.box.left + 10,
            eve_window.box.top,
            eve_window.box.width - 20,
            eve_window.box.height - 10,
        ),
    )
    logging.info("Скриншот: получен")
    if debug:
        logging.debug("Скриншот: сохранен")

    if was_minimized:
        eve_window.minimize()

    return cv2.cvtColor(np.array(screenshot), cv2.COLOR_BGR2RGB)


def highlite_boxes(boxes, module_name, file_name):
    canvas = screenshot.copy()
    for result in boxes.itertuples(index=False):
        tl = (result.tl_x, result.tl_y)
        br = (result.br_x, result.br_y)
        cv2.rectangle(canvas, tl, br, (0, 255, 0), 1)

        cent = (result.cent_x, result.cent_y)
        cv2.circle(canvas, cent, 0, (0, 0, 255), 3)  # отрисовка центра бокса

        cv2.putText(
            canvas,
            f"{result.text} {cent}",
            (tl[0], tl[1] - 10),
            cv2.FONT_HERSHEY_SIMPLEX,
            0.4,
            (0, 255, 0),
            1,
        )
    cv2.imwrite(f"images/{file_name}.png", canvas)
    logging.debug(f"{module_name}: изображение сохранено")


def get_boxes(screenshot):
    results = reader.readtext(
        cv2.bitwise_not(cv2.cvtColor(screenshot, cv2.COLOR_BGR2GRAY)),
        low_text=0.4,
        width_ths=1.5,
    )
    logging.debug("Боксы: получены")

    results_frame = pd.DataFrame(
        [
            [
                int(result[0][0][0]),  # tl_x
                int(result[0][0][1]),  # tl_y
                int(result[0][1][0]),  # tr_x
                int(result[0][1][1]),  # tr_y
                int(result[0][2][0]),  # br_x
                int(result[0][2][1]),  # br_y
                int(result[0][3][0]),  # bl_x
                int(result[0][3][1]),  # bl_y
                int((result[0][0][0] + result[0][1][0]) / 2),  # cent_x
                int((result[0][0][1] + result[0][3][1]) / 2),  # cent_y
                result[1].lower(),  # text
            ]
            for result in results
        ],
        columns=[
            "tl_x",
            "tl_y",
            "tr_x",
            "tr_y",
            "br_x",
            "br_y",
            "bl_x",
            "bl_y",
            "cent_x",
            "cent_y",
            "text",
        ],
    )

    results_frame = results_frame.loc[results_frame["text"].str.len() > 2]
    logging.debug("Боксы: переведены во фрейм")
    logging.info("Боксы: готовы")

    if debug:
        results_frame.to_excel("xlsx/boxes.xlsx", index=False)
        logging.debug("Боксы: документ сохранен")

        highlite_boxes(results_frame, "Боксы", "boxes")

    return results_frame


def get_targets(boxes_frame, text=False):
    anchor_top_x, anchor_top_y = boxes_frame.loc[
        boxes_frame["text"] == "name", ["cent_x", "cent_y"]
    ].values[0]
    anchor_bot_y = boxes_frame.loc[
        boxes_frame["text"] == "drones in", "cent_y"
    ].values[0]

    targets = boxes_frame.loc[
        (boxes_frame["cent_y"].between(anchor_top_y + 10, anchor_bot_y - 10))
        & (
            boxes_frame["cent_x"].between(
                anchor_top_x - 100, anchor_top_x + 100
            )
        )
    ]
    if text:
        targets = targets.loc[targets["text"].str.contains(text)]
    logging.info("Цели: получены")

    if debug:
        targets.to_excel("xlsx/targets.xlsx", index=False)
        logging.debug("Цели: документ сохранен")

        highlite_boxes(targets, "Цели", "targets")

    return targets


def get_cors_by_unique_name(boxes_frame, name):
    sub_ftame = boxes_frame.loc[
        boxes_frame["text"] == name, ["cent_x", "cent_y"]
    ]
    if not sub_ftame.empty:
        x, y = sub_ftame.values[0]
        return (x, y)
    else:
        return None


def go_to_minefield():
    global screenshot
    screenshot = get_screenshot()
    boxes_frame = get_boxes(screenshot)
    targets = get_targets(boxes_frame, "asteroid belt")

    click_mouse(targets.iloc[0].cent_x, targets.iloc[0].cent_y, True)

    screenshot = get_screenshot()
    boxes_frame = get_boxes(screenshot)

    x, y = boxes_frame.loc[
        boxes_frame["text"].str.contains("warp"), ["cent_x", "cent_y"]
    ].values[0]
    click_mouse(x, y)


def start_mine():
    while True:
        global screenshot
        screenshot = get_screenshot()
        boxes_frame = get_boxes(screenshot)
        target = get_targets(boxes_frame, "\\(veldspar\\)").iloc[0]

        click_mouse(target.cent_x, target.cent_y, True)

        screenshot = get_screenshot()
        boxes_frame = get_boxes(screenshot)

        target_lock_cor = get_cors_by_unique_name(boxes_frame, "lock target")
        if target_lock_cor:
            click_mouse(target_lock_cor[0], target_lock_cor[1])
            time.sleep(random.uniform(4.4, 5.8))
            pg.press("f1")
            pg.press("f2")
            return
        else:
            approach_cor = get_cors_by_unique_name(boxes_frame, "approach")
            click_mouse(approach_cor[0], approach_cor[1])


"""
while True:
    screenshot = get_screenshot()
    boxes_frame = get_boxes(screenshot)
    targets = get_targets(boxes_frame, "\\(veldspar\\)")

    # move_mouse(targets.iloc[0].cent_x, targets.iloc[0].cent_y)
    # pg.click()
    # pg.click(button="right")
    # move_mouse(30, 30)

    input("Следущий скриншот - enter")
"""


def main(current_state="EMPTY"):
    if current_state == "UNDOCKED":
        go_to_minefield()
        current_state = "ON_MINEFILD"
        time.sleep(random.uniform(15, 20))
    if current_state == "ON_MINEFILD":
        start_mine()
        current_state = "MINING"


main("UNDOCKED")