from picamera2 import Picamera2, Preview
from gpiozero import DistanceSensor
import time
import os

user = os.getlogin()
user_home = os.path.expanduser(f'~{user}')

camera = Picamera2()
camera.start()

sensor = DistanceSensor(echo=24, trigger=23)

try:
    i = 1
    while (1):
        dis = sensor.distance * 100
        print('Distance: {:.2f} cm'.format(dis))
        if dis < 100:
            camera.capture_file(f'{user_home}/Assignments/10/rev_capture%s.jpg' % i)
            print('The number is %s' % i)
            time.sleep(3)
            i += 1
        else:
            print('waiting')
            time.sleep(0.5)
except KeyboardInterrupt:
    camera.stop_preview
    camera.stop
    pass