send database fix

Author: Koldim2001

configs/app_config.yaml

@@ -13,7 +13,7 @@ hydra:
 
 pipeline:
   save_video : False  # Сохранение итогового видео обработки
-  sent_info_db: False  # Сохраняем ли инфо в бд (требуется заранее запустить микросервисы если ставите True)
+  send_info_db: False  # Сохраняем ли инфо в бд (требуется заранее запустить микросервисы если ставите True)
 
 #--------------------------------------------------GENERAL--------------------------------------------------
 
@@ -69,7 +69,7 @@ video_saver_node:
   out_folder : test_videos/videos_out  # в какую папку сохранить результат
 
 
-sent_info_db_node:
+send_info_db_node:
   how_often_add_info: 5  # как часто добавлять новую информацию (раз в столько секунд)
   table_name: traffic_info  # имя таблицы в БД в которую ведем запись
   connection_info:  # данные о подключении (должны совпадать со значениями из docker-compose файла)

main.py

@@ -5,7 +5,7 @@
 from nodes.DetectionTrackingNodes import DetectionTrackingNodes
 from nodes.TrackerInfoUpdateNode import TrackerInfoUpdateNode
 from nodes.CalcStatisticsNode import CalcStatisticsNode
-from nodes.SentInfoDBNode import SentInfoDBNode
+from nodes.SendInfoDBNode import SendInfoDBNode
 
 
 @hydra.main(version_base=None, config_path="configs", config_name="app_config")
@@ -18,19 +18,19 @@ def main(config) -> None:
     video_saver_node = VideoSaverNode(config["video_saver_node"])
 
     save_video = config["pipeline"]["save_video"]
-    sent_info_db = config["pipeline"]["sent_info_db"]
+    send_info_db = config["pipeline"]["send_info_db"]
 
-    if sent_info_db:
-        sent_info_db_node = SentInfoDBNode(config)
+    if send_info_db:
+        send_info_db_node = SendInfoDBNode(config)
 
     for frame_element in video_reader.process():
 
         frame_element = detection_node.process(frame_element)
         frame_element = tracker_info_update_node.process(frame_element)
         frame_element = calc_statistics_node.process(frame_element)
 
-        if sent_info_db:
-            frame_element = sent_info_db_node.process(frame_element)
+        if send_info_db:
+            frame_element = send_info_db_node.process(frame_element)
 
         frame_element = show_node.process(frame_element)
 

main_optimized.py

@@ -9,7 +9,7 @@
 from nodes.DetectionTrackingNodes import DetectionTrackingNodes
 from nodes.TrackerInfoUpdateNode import TrackerInfoUpdateNode
 from nodes.CalcStatisticsNode import CalcStatisticsNode
-from nodes.SentInfoDBNode import SentInfoDBNode
+from nodes.SendInfoDBNode import SendInfoDBNode
 
 from elements.VideoEndBreakElement import VideoEndBreakElement
 
@@ -40,17 +40,17 @@ def proc_frame_reader_and_detection(queue_out: Queue, config: dict, time_sleep_s
 def proc_tracker_update_and_calc(queue_in: Queue, queue_out: Queue, config: dict):
     tracker_info_update_node = TrackerInfoUpdateNode(config)
     calc_statistics_node = CalcStatisticsNode(config)
-    sent_info_db = config["pipeline"]["sent_info_db"]
-    if sent_info_db:
-        sent_info_db_node = SentInfoDBNode(config)
+    send_info_db = config["pipeline"]["send_info_db"]
+    if send_info_db:
+        send_info_db_node = SendInfoDBNode(config)
     while True:
         ts0 = time()
         frame_element = queue_in.get()
         ts1 = time()
         frame_element = tracker_info_update_node.process(frame_element)
         frame_element = calc_statistics_node.process(frame_element)
-        if sent_info_db:
-            frame_element = sent_info_db_node.process(frame_element)
+        if send_info_db:
+            frame_element = send_info_db_node.process(frame_element)
         ts2 = time()
         queue_out.put(frame_element)
         if PRINT_PROFILE_INFO:

nodes/SentInfoDBNode.py renamed to nodes/SendInfoDBNode.py

@@ -8,11 +8,11 @@
 
 logger = logging.getLogger(__name__)
 
-class SentInfoDBNode:
+class SendInfoDBNode:
     """Модуль для отправки актуальной информации о трафике в базу данных"""
 
     def __init__(self, config: dict) -> None:
-        config_db = config["sent_info_db_node"]
+        config_db = config["send_info_db_node"]
         self.how_often_add_info = config_db["how_often_add_info"]
         self.table_name = config_db["table_name"]
         self.last_db_update = time.time()
@@ -89,7 +89,7 @@ def process(self, frame_element: FrameElement) -> FrameElement:
             return frame_element
         assert isinstance(
             frame_element, FrameElement
-        ), f"SentInfoDBNode | Неправильный формат входного элемента {type(frame_element)}"
+        ), f"SendInfoDBNode | Неправильный формат входного элемента {type(frame_element)}"
 
         # Получение значений для записи в бд новой строки:
         info_dictionary = frame_element.info