StarRailCopilot/dev_tools/screenshot.py
2024-02-29 23:01:29 +08:00

109 lines
2.8 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import os
from datetime import datetime
from PIL import Image
from pynput import keyboard
from module.config.config import AzurLaneConfig
from module.config.utils import alas_instance
from module.device.connection import Connection, ConnectionAttr
from module.device.device import Device
from module.logger import logger
"""
A tool to take screenshots on device
Usage:
python -m dev_tools.screenshot
"""
class EmptyConnection(Connection):
def __init__(self):
ConnectionAttr.__init__(self, AzurLaneConfig('template'))
logger.hr('Detect device')
print()
print('这里是你本机可用的模拟器serial:')
devices = self.list_device()
# Show available devices
available = devices.select(status='device')
for device in available:
print(device.serial)
if not len(available):
print('No available devices')
# Show unavailable devices if having any
unavailable = devices.delete(available)
if len(unavailable):
print('Here are the devices detected but unavailable')
for device in unavailable:
print(f'{device.serial} ({device.status})')
def handle_sensitive_info(image):
# Paint UID to black
image[680:720, 0:180, :] = 0
return image
_ = EmptyConnection()
name = input(
'输入src配置文件名称或者模拟器serial或者模拟器端口号: (默认输入 "src"):\n'
'例如:"src", "127.0.0.1:16384", "7555"\n'
)
name = name.strip().strip('"').strip()
if not name:
name = 'src'
if name.isdigit():
name = f'127.0.0.1:{name}'
if name in alas_instance():
print(f'{name} is an existing config file')
device = Device(name)
else:
print(f'{name} is a device serial')
config = AzurLaneConfig('template')
config.override(
Emulator_Serial=name,
Emulator_PackageName='com.miHoYo.hkrpg',
Emulator_ScreenshotMethod='adb_nc',
)
device = Device(config)
output = './screenshots/dev_screenshots'
os.makedirs(output, exist_ok=True)
device.disable_stuck_detection()
device.screenshot_interval_set(0.)
print('')
print(f'截图将保存到: {output}')
def screenshot():
print(f'截图中...')
image = device.screenshot()
now = datetime.strftime(datetime.now(), '%Y-%m-%d_%H-%M-%S-%f')
file = f'{output}/{now}.png'
image = handle_sensitive_info(image)
print(f'截图中...')
Image.fromarray(image).save(file)
print(f'截图已保存到: {file}')
# Bind global shortcut
GLOBAL_KEY = 'F3'
def on_press(key):
if str(key) == f'Key.{GLOBAL_KEY.lower()}':
screenshot()
listener = keyboard.Listener(on_press=on_press)
listener.start()
while 1:
print()
_ = input(f'按 <回车键> 或者按快捷键 <{GLOBAL_KEY}> 截一张图(快捷键全局生效):')
screenshot()