jZddlmZddlmZmZmZddlmZddlm Z GddeZ dS))Any) BaseSolutionSolutionAnnotatorSolutionResults)LOGGER)colorscbeZdZdZdeddffd Zdedededdfd Zdd eddfd Z de fd Z xZ S) SecurityAlarmaA class to manage security alarm functionalities for real-time monitoring. This class extends the BaseSolution class and provides features to monitor objects in a frame, send email notifications when specific thresholds are exceeded for total detections, and annotate the output frame for visualization. Attributes: email_sent (bool): Flag to track if an email has already been sent for the current event. records (int): Threshold for the number of detected objects to trigger an alert. server (smtplib.SMTP): SMTP server connection for sending email alerts. to_email (str): Recipient's email address for alerts. from_email (str): Sender's email address for alerts. Methods: authenticate: Set up email server authentication for sending alerts. send_email: Send an email notification with details and an image attachment. process: Monitor the frame, process detections, and trigger alerts if thresholds are crossed. Examples: >>> security = SecurityAlarm() >>> security.authenticate("abc@gmail.com", "1111222233334444", "xyz@gmail.com") >>> frame = cv2.imread("frame.jpg") >>> results = security.process(frame) kwargsreturnNc tjdi|d|_|jd|_d|_d|_d|_dS)zInitialize the SecurityAlarm class with parameters for real-time object monitoring. Args: **kwargs (Any): Additional keyword arguments passed to the parent class. FrecordsN)super__init__ email_sentCFGrserverto_email from_email)selfr __class__s i/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/solutions/security_alarm.pyrzSecurityAlarm.__init__$sP ""6"""x *   rpasswordrcddl}|dd|_|j|j||||_||_dS)aAuthenticate the email server for sending alert notifications. This method initializes a secure connection with the SMTP server and logs in using the provided credentials. Args: from_email (str): Sender's email address. password (str): Password for the sender's email account. to_email (str): Recipient's email address. Examples: >>> alarm = SecurityAlarm() >>> alarm.authenticate("sender@example.com", "password123", "recipient@example.com") rNzsmtp.gmail.comiK)smtplibSMTPrstarttlsloginrr)rrrrrs r authenticatezSecurityAlarm.authenticate1sa ll#3S99   *h///  $rrcddlm}ddlm}ddlm}ddl}|d|d}|}|j |d<|j |d <d |d <d |d } | || ||d} | |  |j |tjddS#t $r"} tjd| Yd} ~ dSd} ~ wwxYw)avSend an email notification with an image attachment indicating the number of objects detected. This method encodes the input image, composes the email message with details about the detection, and sends it to the specified recipient. Args: im0 (np.ndarray): The input image or frame to be attached to the email. records (int, optional): The number of detected objects to be included in the email message. Examples: >>> alarm = SecurityAlarm() >>> frame = cv2.imread("path/to/image.jpg") >>> alarm.send_email(frame, records=10) r) MIMEImage) MIMEMultipart)MIMETextNz.jpgFromTozSecurity AlertSubjectzUltralytics alert: z object(s) detected.zultralytics.jpg)namezEmail sent successfully!zFailed to send email: )email.mime.imager%email.mime.multipartr&email.mime.textr'cv2imencodetobytesrrattachr send_messagerinfo Exceptionerror) rim0rr%r&r'r0 img_bytesmessage message_bodyimage_attachmentes r send_emailzSecurityAlarm.send_emailGsj /.....666666,,,,,, LL--a088::  -///  - KWJJJ xx --...%9Y5FGGG'((( 7 K $ $W - - - K2 3 3 3 3 3 7 7 7 L5!55 6 6 6 6 6 6 6 6 6 7s/.C D )DD c $||t||j}t|j|jD]6\}}|||j|t|d7t|j}||j kr$|j s| ||d|_ | }||t|t|j|j S)aGMonitor the frame, process object detections, and trigger alerts if thresholds are met. This method processes the input frame, extracts detections, annotates the frame with bounding boxes, and sends an email notification if the number of detected objects meets or exceeds the specified threshold and an alert has not already been sent. Args: im0 (np.ndarray): The input image or frame to be processed and annotated. Returns: (SolutionResults): Contains processed image `plot_im`, 'total_tracks' (total number of tracked objects) and 'email_sent' (whether an email alert was triggered). Examples: >>> alarm = SecurityAlarm() >>> frame = cv2.imread("path/to/image.jpg") >>> results = alarm.process(frame) ) line_widthT)labelcolor)plot_im total_tracksr)extract_tracksrr@zipboxesclss box_labelnamesrlenrrr>resultdisplay_outputr track_ids)rr8 annotatorboxcls total_detrCs rprocesszSecurityAlarm.processss& C   %cdoFFF DJ 22 U UHC   4:c?&dBSBS  T T T T NN  $ $T_ $ OOC + + +"DO""$$ G$$$wS=P=P]a]lmmmmr)r#) __name__ __module__ __qualname____doc__rrstrr"intr>rrS __classcell__)rs@rr r s2        %s%c%S%T%%%%,*7*7s*74*7*7*7*7X$no$n$n$n$n$n$n$n$nrr N) typingrultralytics.solutions.solutionsrrrultralytics.utilsrultralytics.utils.plottingrr rrrr_s\\\\\\\\\\$$$$$$------MnMnMnMnMnLMnMnMnMnMnr