Fleet Manager: paths cleanup (#216)

This commit is contained in:
Jason Wen
2023-09-23 10:33:33 -04:00
committed by GitHub
parent 3b776937fa
commit da58d5114b
2 changed files with 10 additions and 10 deletions
+4 -4
View File
@@ -3,8 +3,8 @@ import os
import random
import secrets
from flask import Flask, render_template, Response, request, send_from_directory, session, redirect, url_for
import system.fleetmanager.helpers as fleet
from system.loggerd.config import ROOT as REALDATA
import openpilot.system.fleetmanager.helpers as fleet
from openpilot.system.hardware.hw import Paths
app = Flask(__name__)
@@ -39,7 +39,7 @@ def login():
def full(cameratype, route):
chunk_size = 1024 * 512 # 5KiB
file_name = cameratype + (".ts" if cameratype == "qcamera" else ".hevc")
vidlist = "|".join(REALDATA + "/" + segment + "/" + file_name for segment in fleet.segments_in_route(route))
vidlist = "|".join(Paths.log_root() + "/" + segment + "/" + file_name for segment in fleet.segments_in_route(route))
def generate_buffered_stream():
with fleet.ffmpeg_mp4_concat_wrap_process_builder(vidlist, cameratype, chunk_size) as process:
@@ -53,7 +53,7 @@ def full(cameratype, route):
def fcamera(cameratype, segment):
if not fleet.is_valid_segment(segment):
return render_template("error.html", error="invalid segment")
file_name = REALDATA + "/" + segment + "/" + cameratype + (".ts" if cameratype == "qcamera" else ".hevc")
file_name = Paths.log_root() + "/" + segment + "/" + cameratype + (".ts" if cameratype == "qcamera" else ".hevc")
return Response(fleet.ffmpeg_mp4_wrap_process_builder(file_name).stdout.read(), status=200, mimetype='video/mp4')
+6 -6
View File
@@ -3,9 +3,9 @@ import subprocess
from flask import render_template, request, session
from functools import wraps
from pathlib import Path
from system.hardware import PC
from system.loggerd.config import ROOT as REALDATA
from system.loggerd.uploader import listdir_by_creation
from openpilot.system.hardware import PC
from openpilot.system.hardware.hw import Paths
from openpilot.system.loggerd.uploader import listdir_by_creation
from tools.lib.route import SegmentName
@@ -36,7 +36,7 @@ def list_files(path):
def is_valid_segment(segment):
try:
segment_to_segment_name(REALDATA, segment)
segment_to_segment_name(Paths.log_root(), segment)
return True
except AssertionError:
return False
@@ -49,9 +49,9 @@ def segment_to_segment_name(data_dir, segment):
def all_segment_names():
segments = []
for segment in listdir_by_creation(REALDATA):
for segment in listdir_by_creation(Paths.log_root()):
try:
segments.append(segment_to_segment_name(REALDATA, segment))
segments.append(segment_to_segment_name(Paths.log_root(), segment))
except AssertionError:
pass
return segments