1
0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-12 03:58:22 +08:00

Add: Delete sensitive info in log images

This commit is contained in:
LmeSzinc
2020-06-30 02:42:36 +08:00
parent 0550ea3cd1
commit 5982e44dd7
6 changed files with 38 additions and 2 deletions

View File

@@ -0,0 +1,34 @@
from PIL import Image
from module.ui.page import *
def put_image_mask(image, mask):
"""
Args:
image (PIL.Image.Image):
mask (str): Filename
Returns:
PIL.Image.Image:
"""
mask = Image.open(f'./assets/mask/{mask}.png').convert('L')
new = Image.new('RGB', image.size, (0, 0, 0))
new.paste(image, box=(0, 0, image.size[0], image.size[1]), mask=mask)
return new
def handle_sensitive_image(image):
"""
Args:
image (PIL.Image.Image):
Returns:
PIL.Image.Image:
"""
if PLAYER_CHECK.match(image, offset=(30, 30)):
return put_image_mask(image, mask='MASK_PLAYER')
if MAIN_CHECK.match(image, offset=(30, 30)):
return put_image_mask(image, mask='MASK_MAIN')
return image