Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Improve ffprobe executions

Browse Source 
When running ffprobe, use `subprocess.run` rather than
`subprocess.Popen`.  This simplifies the handling that is needed to run
and process the outputs.  Here, filename parsing is also simplified by
explicitly removing the file extension with `os.path.splitext` and
forcing a single split into the camera name and the formatted date.
This commit is contained in:
Sean Vig 2021-05-21 23:35:25 -04:00 committed by Blake Blackshear
parent ebb6d348a3
commit a4e6d9ed9a
3 changed files with 48 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
frigate/events.py
View File

@ -72,28 +72,23 @@ class EventProcessor(threading.Thread):
if f in files_in_use or f in self.cached_clips: if f in files_in_use or f in self.cached_clips:
continue continue
camera = "-".join(f.split("-")[:-1]) basename = os.path.splitext(f)[0]
start_time = datetime.datetime.strptime( camera, date = basename.rsplit("-", maxsplit=1)
f.split("-")[-1].split(".")[0], "%Y%m%d%H%M%S" start_time = datetime.datetime.strptime(date, "%Y%m%d%H%M%S")
)
ffprobe_cmd = " ".join( ffprobe_cmd = [
[ "ffprobe",
"ffprobe", "-v",
"-v", "error",
"error", "-show_entries",
"-show_entries", "format=duration",
"format=duration", "-of",
"-of", "default=noprint_wrappers=1:nokey=1",
"default=noprint_wrappers=1:nokey=1", f"{os.path.join(CACHE_DIR, f)}",
f"{os.path.join(CACHE_DIR,f)}", ]
] p = sp.run(ffprobe_cmd, capture_output=True)
) if p.returncode == 0:
p = sp.Popen(ffprobe_cmd, stdout=sp.PIPE, shell=True) duration = float(p.stdout.decode().strip())
(output, err) = p.communicate()
p_status = p.wait()
if p_status == 0:
duration = float(output.decode("utf-8").strip())
else: else:
logger.info(f"bad file: {f}") logger.info(f"bad file: {f}")
os.remove(os.path.join(CACHE_DIR, f)) os.remove(os.path.join(CACHE_DIR, f))
@ -113,10 +108,10 @@ class EventProcessor(threading.Thread):
else: else:
earliest_event = datetime.datetime.now().timestamp() earliest_event = datetime.datetime.now().timestamp()
# if the earliest event exceeds the max seconds, cap it # if the earliest event is more tha max seconds ago, cap it
max_seconds = self.config.clips.max_seconds earliest_event = max(
if datetime.datetime.now().timestamp() - earliest_event > max_seconds: earliest_event, datetime.datetime.now().timestamp() - max_seconds
earliest_event = datetime.datetime.now().timestamp() - max_seconds )
for f, data in list(self.cached_clips.items()): for f, data in list(self.cached_clips.items()):
if earliest_event - 90 > data["start_time"] + data["duration"]: if earliest_event - 90 > data["start_time"] + data["duration"]:

28
frigate/process_clip.py
View File

@ -31,22 +31,18 @@ logger = logging.getLogger(__name__)
def get_frame_shape(source): def get_frame_shape(source):
ffprobe_cmd = " ".join( ffprobe_cmd = [
[ "ffprobe",
"ffprobe", "-v",
"-v", "panic",
"panic", "-show_error",
"-show_error", "-show_streams",
"-show_streams", "-of",
"-of", "json",
"json", source,
'"' + source + '"', ]
] p = sp.run(ffprobe_cmd, capture_output=True)
) info = json.loads(p.stdout)
p = sp.Popen(ffprobe_cmd, stdout=sp.PIPE, shell=True)
(output, err) = p.communicate()
p_status = p.wait()
info = json.loads(output)
video_info = [s for s in info["streams"] if s["codec_type"] == "video"][0] video_info = [s for s in info["streams"] if s["codec_type"] == "video"][0]

37
frigate/record.py
View File

@ -66,28 +66,23 @@ class RecordingMaintainer(threading.Thread):
if f in files_in_use: if f in files_in_use:
continue continue
camera = "-".join(f.split("-")[:-1]) basename = os.path.splitext(f)[0]
start_time = datetime.datetime.strptime( camera, date = basename.rsplit("-", maxsplit=1)
f.split("-")[-1].split(".")[0], "%Y%m%d%H%M%S" start_time = datetime.datetime.strptime(date, "%Y%m%d%H%M%S")
)
ffprobe_cmd = " ".join( ffprobe_cmd = [
[ "ffprobe",
"ffprobe", "-v",
"-v", "error",
"error", "-show_entries",
"-show_entries", "format=duration",
"format=duration", "-of",
"-of", "default=noprint_wrappers=1:nokey=1",
"default=noprint_wrappers=1:nokey=1", f"{os.path.join(RECORD_DIR, f)}",
f"{os.path.join(RECORD_DIR,f)}", ]
] p = sp.run(ffprobe_cmd, capture_output=True)
) if p.returncode == 0:
p = sp.Popen(ffprobe_cmd, stdout=sp.PIPE, shell=True) duration = float(p.stdout.decode().strip())
(output, err) = p.communicate()
p_status = p.wait()
if p_status == 0:
duration = float(output.decode("utf-8").strip())
else: else:
logger.info(f"bad file: {f}") logger.info(f"bad file: {f}")
os.remove(os.path.join(RECORD_DIR, f)) os.remove(os.path.join(RECORD_DIR, f))