mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-22 16:40:28 +00:00
109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
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()
|