From b5c4bbcdd9ef46af7dfa19ebb09fb29f4e5b4782 Mon Sep 17 00:00:00 2001 From: Martysh12 <49569238+Martysh12@users.noreply.github.com> Date: Sun, 19 Dec 2021 20:26:08 +0200 Subject: [PATCH] Added simple documentation and a shebang Added simple documentation and a shebang and fixed a simple error at line 32. --- main.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index f3b3cbe..cff6b9c 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,6 @@ +#!/usr/bin/env python3 +"""Main asteroid-automator script.""" + import os import pyautogui @@ -7,10 +10,13 @@ if os.name == "windows": class AutomatableGame: + """Base class for each platform.""" + def __init__(self): self.loc = pyautogui.locateOnScreen("images/app.png") def fetch_sshot(self): + """Creates a screenshot, and returns it in Pillow format.""" return pyautogui.screenshot(region=self.loc) def send_key(self, key): @@ -23,4 +29,4 @@ class AutomatableGame__Windows(AutomatableGame): self.d = d3dshot.create() 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)