picture_manager.py

from picamera import PiCamera
from time import sleep
from os.path import join


class PictureManager(object):

    def __init__(self):
        self.camera = PiCamera()
        self.set_max_resolution()
        
    def set_max_resolution(self):
        '''
        method setting camera to max resolution for pictures
        '''
        self.set_framerate()
        self.set_resolution()

    def set_resolution(self, length=3280, height=2464):
        '''
        method setting camera resolution
        :param length: length in pixel of the picture
        :param height: height in pixel of the picture
        '''
        self.camera.resolution = (length, height)
    
    def get_resolution(self):
        '''
        method returning camera resolution
        '''
        return [self.camera.resolution(0), self.camera.resolution(1)]
        
    def set_framerate(self, framerate=15):
        '''
        method setting camera framerate
        :param framerate: framerate of the camera
        '''
        self.camera.framerate = framerate
    
    def get_framerate(self):
        '''
        method returning camera framerate
        '''
        return self.camera.framerate
    
    def take_picture(self, path, sleeptime=5, action_func=lambda x: None):
        '''
        method taking a picture and saving it to given path
        :param path: full path of the picture, need to include name of the
                     picture and extension (.jpg)
        :param sleeptime: pausing time before taking a picture, advised to 
                          leave at least 2 second for the camera to get used 
                          to the light
        '''
        self.camera.start_preview(hflip=True)
        sleep(sleeptime)
        action_func()
        self.camera.capture(path)
        self.camera.stop_preview()
    
    def record_video(self, path, time):
        '''
        method taking a video of given time and saving it to given path
        :param path: full path of the picture, need to include name of the
                     picture and extension (.h264)
        :param time: time of the video in seconds
        '''
        self.camera.start_preview()
        self.camera.start_recording(path)
        sleep(time)
        self.camera.stop_recording()
        self.camera.stop_preview()
        
    def time_lapse(self, path, lapse_minute, interval_picture=10, file_name="lapse"):
        '''
        method taking a time lapse during a given time
        :param path: full path of the created pictures, no extension or filename needed
        :param lapse_minute: time in minutes of the time lapse
        :param interval: interval between two picture, must be more than 5 seconds
        :param file_name: name of the files generated, no need for extension
        '''
        if interval_picture <= 2:
            msg = "the interval between two picture must be more than 5 seconds"
            raise Exception(msg)
        
        number_picture = lapse_minute * 60 / interval_picture
        self.camera.start_preview()
        for picture in range(number_picture):
            sleep(interval_picture)
            self.camera.capture(join(path, file_name + str(picture) + ".jpg"))
        self.camera.stop_preview()