81 lines
2.3 KiB
Python
Executable File
81 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import argparse
|
|
|
|
from rewind.core import clip, mark, save, print_markers
|
|
|
|
def build_clip_parser(subparsers: argparse._SubParsersAction) -> None:
|
|
clip_parser = subparsers.add_parser("clip", help="Clips the last 'x' seconds")
|
|
clip_parser.add_argument(
|
|
"-s", "--seconds",
|
|
type=int,
|
|
default=30,
|
|
help="Number of seconds to include in the clip (default: 30)"
|
|
)
|
|
|
|
def build_save_parser(subparsers: argparse._SubParsersAction) -> None:
|
|
save_parser = subparsers.add_parser("save", help="Saves a segment between any two markers")
|
|
save_parser.add_argument(
|
|
"-s", "--start",
|
|
type=str,
|
|
required=True,
|
|
help="Marker to begin the recording from"
|
|
)
|
|
|
|
save_parser.add_argument(
|
|
"-e", "--end",
|
|
type=str,
|
|
required=True,
|
|
help="Marker to end the recording from"
|
|
)
|
|
|
|
def build_mark_parser(subparsers: argparse._SubParsersAction) -> None:
|
|
mark_parser = subparsers.add_parser("mark", help="Mark the current time in the recording for later reference")
|
|
mark_parser.add_argument(
|
|
"-n", "--name",
|
|
type=str,
|
|
required=True,
|
|
help="Name of the marker"
|
|
)
|
|
|
|
def build_list_parser(subparsers: argparse._SubParsersAction) -> None:
|
|
list_parser = subparsers.add_parser("list", help="List all markers in the recording")
|
|
|
|
def build_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser(
|
|
prog="rewind",
|
|
description="Control OBS recording and create instant clips",
|
|
)
|
|
|
|
sub = parser.add_subparsers(dest="command", required=True)
|
|
|
|
build_clip_parser(sub)
|
|
build_mark_parser(sub)
|
|
build_list_parser(sub)
|
|
build_save_parser(sub)
|
|
|
|
return parser
|
|
|
|
def main(argv=None) -> int:
|
|
parser = build_parser()
|
|
args = parser.parse_args(argv)
|
|
|
|
try:
|
|
if args.command == "clip":
|
|
clip(args.seconds)
|
|
elif args.command == "mark":
|
|
mark(args.name)
|
|
elif args.command == "list":
|
|
print_markers()
|
|
elif args.command == "save":
|
|
save(args.start, args.end)
|
|
else:
|
|
parser.error("Unknown command")
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
return 1
|
|
|
|
return 0
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main()) |