nedo-vision-worker 1.3.11__py3-none-any.whl → 1.3.12__py3-none-any.whl

nedo_vision_worker/__init__.py

@@ -6,5 +6,5 @@ A library for running worker agents in the Nedo Vision platform.
 
 from .worker_service import WorkerService
 
-__version__ = "1.3.11"
+__version__ = "1.3.12"
 __all__ = ["WorkerService"] 

nedo_vision_worker/bootstrap.py

@@ -2,6 +2,7 @@ import logging
 from nedo_vision_worker.config.ConfigurationManager import ConfigurationManager
 from nedo_vision_worker.database.DatabaseManager import set_storage_path, DatabaseManager
 from nedo_vision_worker.worker_service import WorkerService
+from nedo_vision_worker.util.RTMPUrl import RTMPUrl
 
 
 def start_worker(
@@ -13,6 +14,7 @@ def start_worker(
     rtmp_server: str,
     storage_path: str,
     log_level: str = "INFO",
+    rtmp_publish_query_strings: str = ""
 ) -> WorkerService:
     # Logging (only once, force allowed)
     logging.basicConfig(
@@ -36,6 +38,11 @@ def start_worker(
         logger=logging.getLogger("configuration_manager"),
     )
 
+    RTMPUrl.configure(
+        rtmp_server_url=rtmp_server,
+        rtmp_publish_query_strings=rtmp_publish_query_strings
+    )
+
     service = WorkerService(
         configuration_manager=configuration_manager,
         server_host=server_host,

nedo_vision_worker/cli.py

@@ -7,6 +7,11 @@ from typing import NoReturn
 from nedo_vision_worker.bootstrap import start_worker
 from nedo_vision_worker import __version__
 
+import faulthandler
+
+# Enable fault handler to get a traceback
+faulthandler.enable()
+
 class NedoWorkerCLI:
     def __init__(self):
         self.logger = logging.getLogger(__name__)
@@ -38,6 +43,7 @@ class NedoWorkerCLI:
         run.add_argument("--server-host", default="be.vision.sindika.co.id")
         run.add_argument("--server-port", type=int, default=50051)
         run.add_argument("--rtmp-server", default="rtmp://live.vision.sindika.co.id:1935/live")
+        run.add_argument("--rtmp-publish-query-strings", default="")
         run.add_argument("--storage-path", default="data")
         run.add_argument("--system-usage-interval", type=int, default=30)
         run.add_argument("--log-level", default="INFO")
@@ -61,6 +67,7 @@ class NedoWorkerCLI:
                 rtmp_server=args.rtmp_server,
                 storage_path=args.storage_path,
                 log_level=args.log_level,
+                rtmp_publish_query_strings=args.rtmp_publish_query_strings
             )
             signal.pause()  # wait for signal
 

nedo_vision_worker/config/ConfigurationManager.py

@@ -13,13 +13,22 @@ class ConfigurationManager(ConfigurationManagerInterface):
     A class to manage local and remote configuration stored in the 'config' database.
     """
 
-    def __init__(self, worker_token: str, server_host: str, server_port: int, rtmp_server_url: str, logger: logging.Logger):
+    def __init__(
+            self,
+            worker_token: str,
+            server_host: str,
+            server_port: int,
+            rtmp_server_url: str,
+            logger: logging.Logger,
+            rtmp_publish_query_strings: str = "",
+        ):
         try:
             self._logger = logger
             self._worker_token = worker_token
             self._server_host = server_host
             self._server_port = server_port
             self._rtmp_server_url = rtmp_server_url
+            self._rtmp_publish_query_strings = rtmp_publish_query_strings
 
             self._initialize_remote_configuration()
 

nedo_vision_worker/config/DummyConfigurationManager.py

@@ -15,6 +15,7 @@ class DummyConfigurationManager(ConfigurationManagerInterface):
         server_host: str,
         server_port: str,
         rtmp_server_url: str,
+        rtmp_publish_query_strings: str,
         rabbitmq_host: str,
         rabbitmq_port: str,
         rabbitmq_username: str,
@@ -27,6 +28,7 @@ class DummyConfigurationManager(ConfigurationManagerInterface):
         self._server_host = server_host
         self._server_port = server_port
         self._rtmp_server_url = rtmp_server_url
+        self._rtmp_publish_query_strings = rtmp_publish_query_strings
         self._rabbitmq_host = rabbitmq_host
         self._rabbitmq_port = rabbitmq_port
         self._rabbitmq_username = rabbitmq_username
@@ -38,6 +40,7 @@ class DummyConfigurationManager(ConfigurationManagerInterface):
             "server_host": self._server_host,
             "server_port": str(self._server_port),
             "rtmp_server": self._rtmp_server_url,
+            "rtmp_publish_query_strings": self._rtmp_publish_query_strings,
             "rabbitmq_host": self._rabbitmq_host,
             "rabbitmq_port": self._rabbitmq_port,
             "rabbitmq_username": self._rabbitmq_username,

nedo_vision_worker/services/ConnectionInfoClient.py

@@ -31,7 +31,7 @@ class ConnectionInfoClient(GrpcClientBase):
         try:
             if not self.stub:
                 raise Exception("Not connected to manager")
-            
+
             # Prepare the request
             request = GetWorkerConnectionInfoRequest(token=self.token)
 

nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py

@@ -9,10 +9,11 @@ import os
 from .VideoSharingDaemon import VideoSharingClient
 from ..database.DatabaseManager import get_storage_path
 from ..util.EncoderSelector import EncoderSelector
+from ..util.RTMPUrl import RTMPUrl
 
 
 class DirectDeviceToRTMPStreamer:
-    def __init__(self, device_url: str, rtmp_server: str, stream_key: str, stream_duration: int):
+    def __init__(self, device_url: str, stream_key: str, stream_duration: int):
         """
         Initialize the DirectDeviceToRTMPStreamer.
         
@@ -23,7 +24,6 @@ class DirectDeviceToRTMPStreamer:
             stream_duration: Duration in seconds
         """
         self.device_url = device_url
-        self.rtmp_server = rtmp_server
         self.stream_key = stream_key
         self.duration = stream_duration
         
@@ -32,7 +32,7 @@ class DirectDeviceToRTMPStreamer:
             raise ValueError(f"Invalid device URL: {device_url}")
         
         self.device_index = device_index
-        self.rtmp_url = f"{rtmp_server}/{stream_key}"
+        self.rtmp_url = RTMPUrl.get_publish_url(stream_key)
         
         self.ffmpeg_process = None
         self.active = False

nedo_vision_worker/services/FileToRTMPServer.py

@@ -2,9 +2,10 @@ import subprocess
 import logging
 import os
 from ..util.EncoderSelector import EncoderSelector
+from ..util.RTMPUrl import RTMPUrl
 
 class FileToRTMPStreamer:
-    def __init__(self, video_path, rtmp_url, stream_key, fps=30, resolution="1280x720", loop=False):
+    def __init__(self, video_path, stream_key, fps=30, resolution="1280x720", loop=False):
         """
         Initialize the file streamer.
 
@@ -17,7 +18,7 @@ class FileToRTMPStreamer:
             loop (bool): Loop the video until manually stopped.
         """
         self.video_path = video_path
-        self.rtmp_url = f"{rtmp_url}/{stream_key}"
+        self.rtmp_url = RTMPUrl.get_publish_url(stream_key)
         self.fps = fps
         self.resolution = resolution
         self.loop = loop

nedo_vision_worker/services/RTSPtoRTMPStreamer.py

@@ -4,9 +4,9 @@ import time
 import os
 from urllib.parse import urlparse
 from ..util.EncoderSelector import EncoderSelector
-
+from ..util.RTMPUrl import RTMPUrl
 class RTSPtoRTMPStreamer:
-    def __init__(self, rtsp_url, rtmp_url, stream_key, fps=30, resolution="1280x720", duration=120):
+    def __init__(self, rtsp_url, stream_key, fps=30, resolution="1280x720", duration=120):
         """
         Initialize the streamer.
 
@@ -19,7 +19,7 @@ class RTSPtoRTMPStreamer:
             duration (int): Duration in seconds to stream.
         """
         self.rtsp_url = rtsp_url
-        self.rtmp_url = f"{rtmp_url}/{stream_key}"
+        self.rtmp_url = RTMPUrl.get_publish_url(stream_key)
         self.fps = fps
         self.resolution = resolution
         self.duration = duration

nedo_vision_worker/util/RTMPUrl.py

@@ -0,0 +1,24 @@
+class RTMPUrl:
+    """
+    Call `configure()` static method first.
+    """
+    _rtmp_server_url = ""
+    _rtmp_publish_query_strings = ""
+
+    @staticmethod
+    def configure(rtmp_server_url: str, rtmp_publish_query_strings: str = ""):
+        """
+        Example:
+        - `rtmp_server_url` = rtmp://live.sindika.co.id:8554/live
+        - `rtmp_publish_query_strings` = username=my-username&pass=mantap-jiwa
+        """
+        RTMPUrl._rtmp_server_url = rtmp_server_url
+        RTMPUrl._rtmp_publish_query_strings = rtmp_publish_query_strings
+        
+    @staticmethod
+    def get_publish_url(stream_key: str) -> str:
+        url = f"{RTMPUrl._rtmp_server_url}/{stream_key}"
+        if RTMPUrl._rtmp_publish_query_strings:
+            url = f"{RTMPUrl._rtmp_server_url}/{stream_key}?{RTMPUrl._rtmp_publish_query_strings}"
+
+        return url

nedo_vision_worker/worker/VideoStreamWorker.py

@@ -109,6 +109,25 @@ class VideoStreamWorker:
 
     def _process_video_preview_message(self, message):
         """Process messages related to video preview streaming."""
+
+        print("masuk broooo anjayyy")
+
+        # # Try segmentation fault
+        # import ctypes
+
+        # print("Attempting to access a NULL pointer...")
+
+        # # Method 1: Dereference a null pointer
+        # ctypes.string_at(0)
+
+        # # Method 2: Write to a protected memory address
+        # ctypes.memset(0, 0, 1)
+
+        # # Method 3: Access invalid pointer
+        # invalid_ptr = ctypes.cast(0, ctypes.POINTER(ctypes.c_int))
+        # print(invalid_ptr.contents)
+
+
         try:
             data = json.loads(message)
             worker_id = data.get("workerId")
@@ -157,12 +176,12 @@ class VideoStreamWorker:
             logger.info(f"🎥 [APP] Starting RTMP stream (Worker: {worker_id})")
             
             if url.startswith("worker-source/"):
-                streamer = FileToRTMPStreamer(self.source_file_path / os.path.basename(url), rtmp_server, stream_key, stream_duration)
+                streamer = FileToRTMPStreamer(self.source_file_path / os.path.basename(url), stream_key, stream_duration)
             elif self._is_direct_device(url):
-                streamer = DirectDeviceToRTMPStreamer(url, rtmp_server, stream_key, stream_duration)
+                streamer = DirectDeviceToRTMPStreamer(url, stream_key, stream_duration)
             else:
                 # Assume RTSP or other supported protocols
-                streamer = RTSPtoRTMPStreamer(url, rtmp_server, stream_key, stream_duration)
+                streamer = RTSPtoRTMPStreamer(url, stream_key, stream_duration)
 
             streamer.start_stream()
 

{nedo_vision_worker-1.3.11.dist-info → nedo_vision_worker-1.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nedo-vision-worker
-Version: 1.3.11
+Version: 1.3.12
 Summary: Nedo Vision Worker Service Library for AI Vision Processing
 Author-email: Willy Achmat Fauzi <willy.achmat@gmail.com>
 Maintainer-email: Willy Achmat Fauzi <willy.achmat@gmail.com>

{nedo_vision_worker-1.3.11.dist-info → nedo_vision_worker-1.3.12.dist-info}/RECORD

@@ -1,11 +1,11 @@
-nedo_vision_worker/__init__.py,sha256=MaxhB8y9jx14hNcZeQE4eYWCNeAu9Esl615RHv95dRk,204
-nedo_vision_worker/bootstrap.py,sha256=CUwNtb_mf1VGwO1-ujai7ZNN0ojiO3XQzV50JEwmS3U,1430
-nedo_vision_worker/cli.py,sha256=L0hrjSdU4yfoC-KuxkYJ3orpwH_dWzdmGMTHzjOj5MM,2539
+nedo_vision_worker/__init__.py,sha256=RoVtZhuxCdarwYu83xdXmrvTl26wSon6_uw47nooOHQ,204
+nedo_vision_worker/bootstrap.py,sha256=pZ_2sLTQw0PO6hyDTgYoZgdpGJ7W6F7rwHWqnNAuUEs,1652
+nedo_vision_worker/cli.py,sha256=Yov2_YqZzOEPn6Ce-zZu5exCRCxg93l7xPgISL79Jf8,2769
 nedo_vision_worker/doctor.py,sha256=wNkpe8gLVd76Y_ViyK2h1ZFdqeSl37MnzZN5frWKu30,48410
 nedo_vision_worker/worker_service.py,sha256=4SoWqjuz7lQKByU48-yBLV0CeTmX9LIJ5uIc-vmn6ck,7318
-nedo_vision_worker/config/ConfigurationManager.py,sha256=NQD6kXmBrahuMIEhvUr1zrcB60HlQxTmWvBY1dD28PI,8392
+nedo_vision_worker/config/ConfigurationManager.py,sha256=gsNPjaK65sOQBGta99PEGxtejpxso82jtGq77lUElbc,8599
 nedo_vision_worker/config/ConfigurationManagerInterface.py,sha256=doV1Qv6e2edL5Llu-JOF_vhmB87iG-yTXL_GYmlvv9M,270
-nedo_vision_worker/config/DummyConfigurationManager.py,sha256=J1Ur5AD4YKl6CmdJ78SufSw5gdTQURUQtww61FxIoks,1685
+nedo_vision_worker/config/DummyConfigurationManager.py,sha256=-ugVGKa8NNDXe3WChrVGAZ5Y4CsdvyTKC1D4WBGAhLI,1872
 nedo_vision_worker/config/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/database/DatabaseManager.py,sha256=j2koXo1fnMmAyQnY4sv4txfZR8qIzrPyev-sQ4HBaOQ,9478
 nedo_vision_worker/database/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
@@ -49,16 +49,16 @@ nedo_vision_worker/repositories/WorkerSourcePipelineRepository.py,sha256=xfmEvgn
 nedo_vision_worker/repositories/WorkerSourceRepository.py,sha256=AhAJLAacMFdsOgtQNiu7Pahl1DAGI0T1THHeUlKwQJc,2385
 nedo_vision_worker/repositories/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 nedo_vision_worker/services/AIModelClient.py,sha256=lxRNax6FR-pV0G1NpJnlaqjbQeu3kRolIUNSw1RkoZA,15406
-nedo_vision_worker/services/ConnectionInfoClient.py,sha256=udvcKVjZaWfmMbC3HM1cybU5fz9ZidnJGnHj6OSr0uQ,2226
+nedo_vision_worker/services/ConnectionInfoClient.py,sha256=toC9zuY2Hrx1Cwq8Gycy_iFlaG1DvFT4qewlLlitpEQ,2214
 nedo_vision_worker/services/DatasetSourceClient.py,sha256=O5a7onxFl0z47zXaMXWxHAMPuuc-i_vzkd2w5fwrukc,3319
-nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py,sha256=Ypcc0fh1WiUMkICN_KRRAvGmeGkgF8jQ06gfiRYReW4,24747
-nedo_vision_worker/services/FileToRTMPServer.py,sha256=0hY5pmeAzLw_d3uPR2Qp6gSAYb4rJHiAunuNe08OvkM,2870
+nedo_vision_worker/services/DirectDeviceToRTMPStreamer.py,sha256=YO_z1DPQgVoR2hakmd5vP07D8_DYkWYaWT68mOT4rUk,24731
+nedo_vision_worker/services/FileToRTMPServer.py,sha256=ZxcV1sjvvTqfeIuRuwa8uxKbKATYnxP7xwxX5cWdlTc,2904
 nedo_vision_worker/services/GrpcClientBase.py,sha256=RkJDGRsXu5HalMDR8cOsIaoFf5tA_cLTkh5euBPyo2M,6852
 nedo_vision_worker/services/GrpcClientManager.py,sha256=DLXekmxlQogLo8V9-TNDXtyHT_UG-BaggqwsIups55k,5568
 nedo_vision_worker/services/GrpcConnection.py,sha256=UNjaUC4ZcXuteHQx8AAAL5ymYkT1OpoIvyCYPUc3tCI,4915
 nedo_vision_worker/services/ImageUploadClient.py,sha256=T353YsRfm74G7Mh-eWr5nvdQHXTfpKwHJFmNW8HyjT8,3019
 nedo_vision_worker/services/PPEDetectionClient.py,sha256=3oE_Y0Avw_kcWKgTA7qmChq32woW7JerWefrCxllUYU,5172
-nedo_vision_worker/services/RTSPtoRTMPStreamer.py,sha256=LtfrWDHNcm-Ky6nZLnFCF8xgqIm7VQmsWIenK2yKNfo,3804
+nedo_vision_worker/services/RTSPtoRTMPStreamer.py,sha256=UoSMRlp9r7QCQRNkPjrVO9OmDeGIgJRxD1la6YFOPyk,3837
 nedo_vision_worker/services/RestrictedAreaClient.py,sha256=TD2Y5UJ0NjXgNFNgwS5ze4P-5jWwgrP0j0Nlyyd7-6Q,4844
 nedo_vision_worker/services/SharedDirectDeviceClient.py,sha256=dylMhqpMsfK_UKLWIVL-ApJRP4g-NCP_55xvlGYBiwo,10760
 nedo_vision_worker/services/SharedVideoStreamServer.py,sha256=WMKVxkzMoyfbgYiJ0fQOT-Ujz9btz6FLlaDP738yfoY,11601
@@ -78,6 +78,7 @@ nedo_vision_worker/util/HardwareID.py,sha256=rSW8-6stm7rjXEdkYGqXMUn56gyw62YiWnS
 nedo_vision_worker/util/ImageUploader.py,sha256=2xipN3fwpKgFmbvoGIdElpGn5ARJyrgR4dXtbRf73hw,3764
 nedo_vision_worker/util/Networking.py,sha256=uOtL8HkKZXJp02ZZIHWYMAvAsaYb7BsAPTncfdvJx2c,3241
 nedo_vision_worker/util/PlatformDetector.py,sha256=-iLPrKs7hp_oltkCI3hESJQkC2uRyu1-8mAbZrvgWx0,1501
+nedo_vision_worker/util/RTMPUrl.py,sha256=BKB_FyFMFbRFhqvu_PftbiUCe_DhleJCx6FGg8z-htw,845
 nedo_vision_worker/util/SystemMonitor.py,sha256=2kkqj9mOlywAS2fHdN1TaIXSXvCApcIHj0IO2T9k3Yw,11471
 nedo_vision_worker/util/VideoProbeUtil.py,sha256=cF-vJ7hIDlXfEJby2a0s9tqwkPGVz_6B3Vv4D5pMmIw,12876
 nedo_vision_worker/util/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
@@ -93,11 +94,11 @@ nedo_vision_worker/worker/PipelinePreviewWorker.py,sha256=owFiBbktcOZkdImQeykZSe
 nedo_vision_worker/worker/RabbitMQListener.py,sha256=Gwn7VpRg0fMZ0fva98eOnTzZ4HPFf2i_ZCUgwfTdyYQ,6768
 nedo_vision_worker/worker/RestrictedAreaManager.py,sha256=Pz2M9KPSMa1QPXZeYhyN9ih4eX6wmkNS_ZPu5KrvNxY,5927
 nedo_vision_worker/worker/SystemUsageManager.py,sha256=mkh4sT-HkIEY1CJHMEG6LP9ATu39YXvLRLyf995OkoQ,5315
-nedo_vision_worker/worker/VideoStreamWorker.py,sha256=5n6v1PNO7IB-jj_McALLkUP-cBjJoIEw4UiSAs3vTb0,7606
+nedo_vision_worker/worker/VideoStreamWorker.py,sha256=e6J_9en03JHkWLBqHR5QDRQHWdhcD3Nz3pN8GgkPTgk,8052
 nedo_vision_worker/worker/WorkerManager.py,sha256=2bxXi19fp3p1qjYBStYRdVVgko8dnevXx1_M_sqH5og,5521
 nedo_vision_worker/worker/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
-nedo_vision_worker-1.3.11.dist-info/METADATA,sha256=aA9ZFmzYLp9PLOHV3LN_vx1Y4XWLo6_kKk9AxDH5he4,14654
-nedo_vision_worker-1.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nedo_vision_worker-1.3.11.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
-nedo_vision_worker-1.3.11.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
-nedo_vision_worker-1.3.11.dist-info/RECORD,,
+nedo_vision_worker-1.3.12.dist-info/METADATA,sha256=9ZtnikwQ8W5CZk726SpCC1eQi5TUY_6qvaQ2lucqexM,14654
+nedo_vision_worker-1.3.12.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+nedo_vision_worker-1.3.12.dist-info/entry_points.txt,sha256=LrglS-8nCi8C_PL_pa6uxdgCe879hBETHDVXAckvs-8,60
+nedo_vision_worker-1.3.12.dist-info/top_level.txt,sha256=vgilhlkyD34YsEKkaBabmhIpcKSvF3XpzD2By68L-XI,19
+nedo_vision_worker-1.3.12.dist-info/RECORD,,

{nedo_vision_worker-1.3.11.dist-info → nedo_vision_worker-1.3.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any