From e4f1cd277246c2b39f6592cd2da63c2bc5589314 Mon Sep 17 00:00:00 2001 From: march3 Date: Sat, 30 Dec 2023 20:06:25 +0800 Subject: [PATCH] =?UTF-8?q?Python=E8=B6=85=E4=BA=BA-=E5=AE=87=E5=AE=99?= =?UTF-8?q?=E6=A8=A1=E6=8B=9F=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sim_scenes/featured/jupiter_moon_protects_earth.py | 2 +- tools/sim_video_cap.py | 13 ++++++++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/sim_scenes/featured/jupiter_moon_protects_earth.py b/sim_scenes/featured/jupiter_moon_protects_earth.py index 3a3a406..e4345c7 100644 --- a/sim_scenes/featured/jupiter_moon_protects_earth.py +++ b/sim_scenes/featured/jupiter_moon_protects_earth.py @@ -228,7 +228,7 @@ class JupiterMoonProtectsEarthSim(UniverseSimScenes): print("满足条件退出") exit() - if ctime - self.run_begin_time > 600: + if ctime - self.run_begin_time > 700: print("没有满足条件退出") exit() diff --git a/tools/sim_video_cap.py b/tools/sim_video_cap.py index 672d6ce..0dc85a3 100644 --- a/tools/sim_video_cap.py +++ b/tools/sim_video_cap.py @@ -20,18 +20,29 @@ def get_window_handle(window_name="宇宙模拟器(universe sim)"): handle = win32gui.FindWindow(None, window_name) return handle +def rename_exists(save_name): + file_no = 0 + rename_save_name = save_name + while os.path.exists(os.path.abspath(rename_save_name)): + rename_save_name = f"{save_name[:-4]}_{file_no}{save_name[-4:]}" + file_no += 1 + + return rename_save_name def get_args(): parser = argparse.ArgumentParser() parser.add_argument('--fps', type=int, default=30, help='frame per second') parser.add_argument('--total_time', type=int, default=10000000, help='video total time') - parser.add_argument('--save_name', type=str, default='video.mp4', help='save file name') + parser.add_argument('--save_name', type=str, default='jupiter_moon_protects_earth7.mp4', help='save file name') parser.add_argument('--wait', type=int, default=0, help='wait for recoding') # parser.add_argument('--screen_type', default=0, type=int, choices=[0, 1], help='1: full screen, 0: region screen') args = parser.parse_args() print("total_time:", args.total_time) print("fps:", args.fps) + + args.save_name = rename_exists(args.save_name) + print("save_name:", args.save_name) print("wait:", args.wait) if args.save_name == '': -- GitLab