odrling/fansub-utils
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Add "check_exists", minor refactoring

Browse source 
generate_keyframes:
* "check_exists" checks if a keyframe file already exists. Default is "True". You can set it to "False" by setting -C.

* Minor refactoring. Deleting indexing files is now a one-liner (but still works if if-else, basically).

* Remove the "Done." when a keyframe file is done being generated. It will just move on to the next file now. Instead it will now say it's done when all the keyframe files are done being generated.
This commit is contained in:
LightArrowsEXE 2019-11-21 10:37:19 +01:00
parent da47452b28
commit c166491775
1 changed files with 14 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
generate_keyframes.py
View file

@ -49,6 +49,10 @@ def main():
for f in files:
if f.endswith(ext_in):
if args.check_exists:
if os.path.exists(f"{f[:-len(ext_in)]}_keyframes.txt"):
print(f"\nKeyframes already exist for {f}. Skipping.")
continue
print(f"\nGenerating keyframes for {f}:")
src = core.lsmas.LWLibavSource(f) if f.endswith("m2ts") else core.ffms2.Source(f)
@ -71,18 +75,12 @@ def main():
generate_keyframes(src, os.path.abspath(f"{f[:-len(ext_in)]}_keyframes.txt"), args.noheader)
print(f"Progress: {src.num_frames}/{src.num_frames} frames")
if f.endswith(".m2ts"):
try:
os.remove(f"{f}.lwi")
except FileNotFoundError:
pass
else:
try:
os.remove(f"{f}.ffindex")
except FileNotFoundError:
pass
try:
os.remove(f"{f}.lwi") if f.endswith("m2ts") else os.remove(f"{f}.ffindex")
except FileNotFoundError:
pass
print(f"Output: {args.outfile}\nDone.") if args.outfile and args.file else print(f"Output: {f[:-len(ext_in)]}_keyframes.txt\nDone.")
print(f"Output: {args.outfile}") if args.outfile and args.file else print(f"Output: {f[:-len(ext_in)]}_keyframes.txt")
if __name__ == "__main__":
@ -110,7 +108,12 @@ if __name__ == "__main__":
help="string of trims to source file. " \
"format: \"[inclusive,exclusive],[inclusive,exclusive],[None,exclusive],[inclusive,None]\"",
action="store")
parser.add_argument("-C", "--check_exists",
default=True,
help="Check if keyframe file already exists (default: %(default)s)",
action="store_false")
# TO-DO: Add a SCXvid mode for those where the WWXD format
# doesn't appear to work properly.
args = parser.parse_args()
main()
print(f"\nDone generating keyframes.")