Change start time priority to make database take priority

trunk
HeNine 3 years ago
parent 50c4d8a096
commit cc49374096

@ -44,20 +44,22 @@ def main(channel, database="", base_dir=".",
logging.debug("Got database cursor.") logging.debug("Got database cursor.")
logging.info("Figuring out starting time...") logging.info("Figuring out starting time...")
if start_time is not None: db_start_time = get_end_of_transcript(db_cursor)
# Database start time takes priority
if db_start_time is not None:
start_time = db_start_time
elif start_time is not None:
start_time = dateutil.parse(start_time) start_time = dateutil.parse(start_time)
else: else:
start_time = get_end_of_transcript(db_cursor)
if end_time is not None:
end_time = dateutil.parse(end_time)
# No start time argument AND no end of transcript (empty database) # No start time argument AND no end of transcript (empty database)
if start_time is None:
logging.error("Couldn't figure out start time!") logging.error("Couldn't figure out start time!")
db_conn.close() db_conn.close()
exit(1) exit(1)
if end_time is not None:
end_time = dateutil.parse(end_time)
logging.info("Loading models...") logging.info("Loading models...")
recognizer = BuscribeRecognizer(SAMPLE_RATE, model, spk_model) recognizer = BuscribeRecognizer(SAMPLE_RATE, model, spk_model)
logging.info("Models loaded.") logging.info("Models loaded.")
@ -77,7 +79,7 @@ def main(channel, database="", base_dir=".",
gevent.signal_handler(signal.SIGTERM, stop) gevent.signal_handler(signal.SIGTERM, stop)
while True: while start_time < end_time:
# If end time isn't given, use current time (plus fudge) to get a "live" segment list # If end time isn't given, use current time (plus fudge) to get a "live" segment list
segments = common.get_best_segments(segments_dir, segments = common.get_best_segments(segments_dir,
start_time, start_time,

Loading…
Cancel
Save