import pygame
import pygame.camera
import time

import cv2

def capture(camera_stream = 0, camera_res = (1920, 1080)):
    cam = cv2.VideoCapture(camera_stream)
    s, img = cam.read()

    if s:
        cv2.imwrite('./capture.jpg', img)
    # pygame.camera.init()
    # pygame.camera.list_cameras()
    # cam = pygame.camera.Camera(camera_stream, camera_res)
    # cam.start()
    # time.sleep(5)
    # img = cam.get_image()
    # pygame.image.save(img, './capture.jpg')
    # cam.stop()
    return './capture.jpg'