refactor dig

utils.py

@@ -2,6 +2,16 @@ import numpy as np
 import os
 import cv2 as cv
 from PIL import Image
+import pydirectinput
+import random
+import random
+from time import time
+
+import cv2 as cv
+
+import pydirectinput
+import keyboard
+from tresh_util import super_tresh_main, super_tresh_needle
 
 
 def mse(imageA, imageB):
@@ -144,3 +154,22 @@ def update_screenshot_with_short_pic_list(short_pic_lst, lst_bl_items):
         # save_line_item_npy_jpg(r_short_pic_lst)
 
     return screenshot
+
+
+def dig_point(point1, point2, dig_time):
+    pydirectinput.moveTo(point1, point2)
+    pydirectinput.mouseDown()
+    w = random.randint(1, 50)
+    cv.waitKey(dig_time + w)
+    pydirectinput.mouseUp()
+
+
+def check_for_ok_button(cap_win, vis, conf):
+    screenshot = cap_win.get_screenshot()
+    rectangles = vis.find(screenshot, cv.imread("dig/ok_button.jpg", cv.IMREAD_UNCHANGED), 0.5, 1)
+    offset_left = conf.returnDiggingWindowPos2()[2]
+    offset_down = conf.returnDiggingWindowPos2()[3]
+    if len(rectangles) == 1:
+        pointis = vis.get_click_points(rectangles)
+        for pointi in pointis:
+            dig_point(pointi[0] + offset_left, pointi[1] + offset_down, 150)