67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import pyautogui
|
|
import time
|
|
from ctypes import windll
|
|
import threading
|
|
|
|
|
|
class lumberjackBot():
|
|
|
|
__author__ = "EnriqueMoran"
|
|
|
|
def __init__(self, playX, playY, treeX, treeY, x, y):
|
|
self.playX = playX
|
|
self.playY = playY
|
|
self.treeX = treeX
|
|
self.treeY = treeY
|
|
# Those attributes has been placed here in order to save calcs:
|
|
self.x = x # Left side branch X location
|
|
self.y = y + 5 # Left side branch Y location
|
|
self.movement_buffer = ["right"] # First movement
|
|
|
|
|
|
def move(self):
|
|
speed = 0.14
|
|
if self.movement_buffer[0] == "left" and len(self.movement_buffer) == 2:
|
|
pyautogui.typewrite(['left', 'left'], speed)
|
|
elif self.movement_buffer[0] == "right" and len(self.movement_buffer) == 2:
|
|
pyautogui.typewrite(['right', 'right'], speed)
|
|
self.movement_buffer = self.movement_buffer[1:]
|
|
|
|
def get_color(self, rgb):
|
|
r = rgb & 0xff
|
|
g = (rgb >> 8) & 0xff
|
|
b = (rgb >> 16) & 0xff
|
|
return r,g,b
|
|
|
|
def get_pixel(self, x, y): # Modify class atribute
|
|
screen = windll.user32.GetDC(0)
|
|
rgb = windll.gdi32.GetPixel(screen, x, y)
|
|
return self.get_color(rgb)
|
|
|
|
def play(self):
|
|
while True:
|
|
pixel_color = self.get_pixel(self.x, self.y)
|
|
if pixel_color == (161, 116, 56):
|
|
self.movement_buffer.append("right")
|
|
self.move()
|
|
elif pixel_color == (211, 247, 255) or pixel_color == (241, 252, 255):
|
|
self.movement_buffer.append("left")
|
|
self.move()
|
|
#print(pixel_color, self.x, self.y, self.movement_buffer)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("Running in 3 seconds, minimize this windows. To stop the program drag the mouse to the top-left corner of your screen.")
|
|
time.sleep(3)
|
|
playX, playY = pyautogui.locateCenterOnScreen('playButton.png')
|
|
pyautogui.moveTo(playX, playY)
|
|
pyautogui.click() # Start the game by pressing play button
|
|
time.sleep(0.5) # Wait for screen refresh
|
|
x, y = pyautogui.locateCenterOnScreen('branch.png')
|
|
pyautogui.moveTo(x, y + 5)
|
|
treeX, treeY = playX - 6, playY - 177 # Tree position
|
|
time.sleep(0.3)
|
|
print("Im playing... To stop me click on IDLE and press CTRL+F6.")
|
|
lumberjack = lumberjackBot(playX, playY, treeX, treeY, x, y)
|
|
lumberjack.play() # Game start
|