lumberjackBot/lumberjackBot.py
2018-05-18 20:44:59 +02:00

64 lines
2.3 KiB
Python

import pyautogui
import time
from ctypes import windll
import threading
class lumberjackBot():
__author__ = "EnriqueMoran"
def __init__(self, playX, playY, treeX, treeY):
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 = treeX - 22 # Left side branch X location
self.y = treeY - 244 # Left side branch Y location
self.movement_buffer = ["right"] # First movement
def move(self):
speed = 0.17
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):
pyautogui.moveTo(self.x, self.y)
while True:
pixel_color = self.get_pixel(self.x, self.y)
if pixel_color == (161, 116, 56):
self.movement_buffer.append("right")
else:
self.movement_buffer.append("left")
#print(pixel_color, self.x, self.y, self.movement_buffer)
self.move()
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
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)
lumberjack.play() # Game start