asteroid-automator/main.py

33 lines
725 B
Python
Raw Normal View History

#!/usr/bin/env python3
"""Main asteroid-automator script."""
import os
import pyautogui
if os.name == "windows":
import d3dshot
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):
pass
class AutomatableGame__Windows(AutomatableGame):
def __init__(self):
super().__init__(self)
self.d = d3dshot.create()
def fetch_sshot(self):
return self.d.screenshot() # TODO: Cut this to self.loc(x, y, w, h)