import cv2

class VideoStream:
    def __init__(self, rtsp_url):
        self.rtsp_url = rtsp_url
        self.stream = cv2.VideoCapture(rtsp_url)

    def get_frame(self):
        ret, frame = self.stream.read()
        if not ret:
            return None
        return frame