From 28f01510389ee951c5f3bedc237f25a4a6cec089 Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Mon, 11 Apr 2022 15:13:06 +0300
Subject: [PATCH] wallpaper-screenshot

---
 daun.py                         | 19 +++++++++++++++++++
 modules/wallpaper-screenshot.py |  4 ++++
 2 files changed, 23 insertions(+)
 create mode 100644 modules/wallpaper-screenshot.py

diff --git a/daun.py b/daun.py
index 2bb5e9d..1038407 100644
--- a/daun.py
+++ b/daun.py
@@ -23,6 +23,10 @@ parser.add_argument('--wp-control', help='control command for wallpaper engine',
 # - Wallpaper ///
 parser.add_argument('--set-wallpaper', help='link (web image url or file path) to wallpaper to be set',
                     metavar='P:/ath/To/wallpaper.jpg', dest='set_wallpaper')
+# - Wallpaper Screenshot ///
+parser.add_argument('--wallpaper-screenshot', help='make a screenshot and set it to wallpaper (prank),'
+                                                   'specify time in seconds to wait before screenshot',
+                    type=float, metavar='0.1')
 # - Download ///
 parser.add_argument('-d', '--download', help='download file from url to specified',
                     nargs=2, metavar=('https://sample.url/to/file.ext', 'P:/ath/To/Folder/With/file.ext'),
@@ -41,6 +45,8 @@ args = parser.parse_args()
 # -- Base ///
 """
 Base daun library, this module does not contain any code and will be added anyway
+Sizes of modules are specified if you add only one module, but if you add more modules, some of them may be 
+double-used, so size of build will be much smaller
 
 7.61 Mb
 """
@@ -102,6 +108,19 @@ if args.set_wallpaper:
 
     set_wallpaper(args.set_wallpaper)
 
+# -- Wallpaper Screenshot ///
+"""
+Make a screenshot and set it to wallpaper (prank)
+
+2.9 Mb
+"""
+if args.wallpaper_screenshot:
+    import time
+    from modules.wallpaper import set_wallpaper
+    from modules.screenshot import save_screenshot
+    time.sleep(args.wallpaper_screenshot)
+    set_wallpaper(save_screenshot())
+
 # -- Download ///
 """
 Download file from url to specified
diff --git a/modules/wallpaper-screenshot.py b/modules/wallpaper-screenshot.py
new file mode 100644
index 0000000..6bc8179
--- /dev/null
+++ b/modules/wallpaper-screenshot.py
@@ -0,0 +1,4 @@
+from modules.wallpaper import set_wallpaper
+from modules.screenshot import save_screenshot
+
+