Added simple documentation and a shebang

Added simple documentation and a shebang and fixed a simple error at line 32.
This commit is contained in:
Martysh12 2021-12-19 20:26:08 +02:00
parent 555ecfa7a3
commit b5c4bbcdd9
1 changed files with 7 additions and 1 deletions

View File

@ -1,3 +1,6 @@
#!/usr/bin/env python3
"""Main asteroid-automator script."""
import os import os
import pyautogui import pyautogui
@ -7,10 +10,13 @@ if os.name == "windows":
class AutomatableGame: class AutomatableGame:
"""Base class for each platform."""
def __init__(self): def __init__(self):
self.loc = pyautogui.locateOnScreen("images/app.png") self.loc = pyautogui.locateOnScreen("images/app.png")
def fetch_sshot(self): def fetch_sshot(self):
"""Creates a screenshot, and returns it in Pillow format."""
return pyautogui.screenshot(region=self.loc) return pyautogui.screenshot(region=self.loc)
def send_key(self, key): def send_key(self, key):
@ -23,4 +29,4 @@ class AutomatableGame__Windows(AutomatableGame):
self.d = d3dshot.create() self.d = d3dshot.create()
def fetch_sshot(self): def fetch_sshot(self):
return d.screenshot() # Cut this to self.loc(x, y, w, h) return self.d.screenshot() # TODO: Cut this to self.loc(x, y, w, h)