From 8d3bc2c9499cc24bbd3720d760a4f45e98e38231 Mon Sep 17 00:00:00 2001 From: Sourcery AI Date: Mon, 15 Aug 2022 05:19:35 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- mss.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/mss.py b/mss.py index 66f5e798..5f862253 100644 --- a/mss.py +++ b/mss.py @@ -172,7 +172,7 @@ def save(self, # Monitors screen shots! for i, monitor in enumerate(self.enum_display_monitors(screen)): - if screen <= 0 or (screen > 0 and i + 1 == screen): + if screen <= 0 or i + 1 == screen: fname = output if '%d' in output: fname = output.replace('%d', str(i + 1)) @@ -579,8 +579,7 @@ def _callback(monitor, dc, rect, data): monitors = [] callback = self.MONITORENUMPROC(_callback) windll.user32.EnumDisplayMonitors(0, 0, callback, 0) - for mon in monitors: - yield mon + yield from monitors def get_pixels(self, monitor): ''' Retrieve all pixels from a monitor. Pixels have to be RGB. @@ -628,8 +627,11 @@ def get_pixels(self, monitor): windll.gdi32.DeleteObject(bmp) # Replace pixels values: BGR to RGB - self.image[2:buffer_len:3], self.image[0:buffer_len:3] = \ - self.image[0:buffer_len:3], self.image[2:buffer_len:3] + self.image[2:buffer_len:3], self.image[:buffer_len:3] = ( + self.image[:buffer_len:3], + self.image[2:buffer_len:3], + ) + return self.image @@ -663,7 +665,7 @@ def on_exists(fname): from os import rename from os.path import isfile if isfile(fname): - newfile = fname + '.old' + newfile = f'{fname}.old' print('{0} -> {1}'.format(fname, newfile)) rename(fname, newfile) return True