make screenshots

apple-farm.py

@@ -1,18 +1,30 @@
 from datetime import datetime
+import shutil
 
 __author__ = "Alexander Popov <iiiypuk@fastmail.fm>"
 __version__ = "1.0.0"
 __license__ = "Unlicense"
 
 
-def date_string():
+def date_string(symbols=True):
     """Return current date."""
     date_now = datetime.now()
-    date_str = date_now.strftime("%Y-%m-%d %H:%M:%S")
+
+    if symbols:
+        date_str = date_now.strftime("%Y-%m-%d %H:%M:%S")
+    else:
+        date_str = date_now.strftime("%Y-%m-%d_%H_%M_%S")
 
     return(date_str)
 
 
+def make_screenshot():
+    """..."""
+
+    image = capture(App.focusedWindow())
+    shutil.move(image, "{0}.png".format(date_string(False)))
+
+
 def run_level():
     """Running level and farming apples"""
 
@@ -22,6 +34,8 @@ def run_level():
     wait("images/level_title.png", 3)
     click("images/level_start_button.png")
 
+    make_screenshot()
+
     wait("images/equip_window.png", 3)
     click("images/button_start.png")