Spaces:
Sleeping
Sleeping
File size: 3,323 Bytes
c640bc9 066c36c c640bc9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import os
import time
from src.services.image_capture_service.capture_main import FrameSaver
class ImageCaptureService:
def __init__(self, flag_path, source, image_path_to_save, image_hash_threshold):
self.flag_path = flag_path
self.source = source
self.image_path_to_save = image_path_to_save
self.image_hash_threshold = image_hash_threshold
self.running = False
def create_stop_flag(self):
try:
with open(self.flag_path, 'w') as flag_file:
flag_file.write('False') # Set initial content as 'False'
print("Flag file created and set to 'False'.")
except Exception as e:
print(f"Error creating flag file: {e}")
def check_stop_flag(self):
try:
if os.path.exists(self.flag_path):
with open(self.flag_path, 'r') as flag_file:
content = flag_file.read().strip()
return content.lower() == "true"
else:
return False
except Exception as e:
print(f"Error checking stop flag: {e}")
return False
def update_stop_flag(self, value):
try:
with open(self.flag_path, 'w') as flag_file:
flag_file.write(str(value)) # Write the provided value to the flag file
except Exception as e:
print(f"Error updating stop flag: {e}")
def start_service(self):
try:
print("Image Capturing service starting.")
self.running = True
start_capture = FrameSaver(self.source, self.image_path_to_save, self.image_hash_threshold)
start_capture.start_stream()
while self.running:
time.sleep(1)
stop_flag = self.check_stop_flag()
if stop_flag:
print("Stop flag detected. Stopping all frame capturing services.")
self.running = False
stop_successful = start_capture.stop_stream()
if stop_successful:
print("Frame capturing services successfully stopped.")
else:
print("Issue encountered while stopping frame capturing services.")
except Exception as e:
print(f"Error in setting up FrameSaver: {e}")
print(f"Error in setting up FrameSaver: {e}")
finally:
if 'start_capture' in locals():
stop_successful = start_capture.stop_stream()
if stop_successful:
print("Frame capturing services successfully stopped in finally block.")
else:
print("Issue encountered while stopping frame capturing services in finally block.")
def stop_service(self):
self.update_stop_flag("True")
if __name__ == "__main__":
FLAG_PATH = "resources/image_capturing"
# SOURCE = 'rtsp://ishwor:subedi@192.168.1.106:5555/h264_opus.sdp'
SOURCE = 0
IMAGE_PATH_TO_SAVE = "images/cam_images"
IMAGE_HASH_THRESHOLD = 5
image_capture_service = ImageCaptureService(FLAG_PATH, SOURCE, IMAGE_PATH_TO_SAVE, IMAGE_HASH_THRESHOLD)
image_capture_service.create_stop_flag()
image_capture_service.start_service()
time.sleep(1)
|