evan/KISS
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Greyscale

Browse Source
This commit is contained in:
Evan Reichard 2022-11-17 09:13:35 -05:00
parent f286b23f6a
commit 77a4727cc0
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
kiss/ssmanager.py
View File

@ -6,7 +6,7 @@ from pathlib import Path
from threading import Thread
import yaml
from PIL import Image
from PIL import Image, ImageOps
class ScreenshotManager:
@ -29,10 +29,14 @@ class ScreenshotManager:
self.__loop_thread = Thread(target=self.__loop)
self.__loop_thread.start()
def __rotate_file(self, filename):
def __process_image(self, ss):
filename = ss.get("output")
fp = Path(self.data_path, filename)
img = Image.open(fp)
img = ImageOps.grayscale(Image.open(fp))
if ss.get("rotate", False) is True:
img = img.rotate(90, expand=True)
img.save(fp)
def __loop(self):
@ -43,8 +47,7 @@ class ScreenshotManager:
["shot-scraper", "multi", self.__config_path], cwd=self.data_path
)
for ss in self.__config_yaml:
if ss.get("rotate", False) is True:
self.__rotate_file(ss.get("output"))
self.__process_image(ss)
time.sleep(1)
counter += 1