Image sequence recorder linux (requires ffmpeg)

quam.py

# -*- coding: utf-8 -*-
#Requires ffmpeg, audio not working at the moment: silageman
import gtk.gdk
import time
import os
 
print '\033[1;33m  ________ _______    _____    \033[1;m'
print '\033[1;33m / ____/  |  \__  \  /     \   \033[1;m'
print '\033[1;33m< <_|  |  |  // __ \|  Y Y  \  \033[1;m'
print '\033[1;33m \__   |____/(____  /__|_|  /  \033[1;m'
print '\033[1;33m    |__|          \/      \/   \033[1;m'
print '\033[1;33m    |__|          \/      \/   \033[1;m'
 
def main():
    
    i = -1
    speed = 0.015
 
    restart = raw_input("Start Recording?: y/n ")
    if restart == "no" or restart == "n":
         exit(0)
    if restart == "yes" or restart == "y":
         print""

    try:
        while (True):
            i += 1
            w = gtk.gdk.get_default_root_window()
            sz = w.get_size()
            time.sleep(speed)
            print "FRAME %d OK!"%i
            pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,True,8,sz[0],sz[1])
            pb = pb.get_from_drawable(w,w.get_colormap(),0,0,0,0,sz[0],sz[1])
            pb.save("image%04d.jpeg"%i,"jpeg")
    except KeyboardInterrupt:
        os.system("ffmpeg -framerate 15 -i image%04d.jpeg -codec copy output.mkv")%i
        os.system("find . -type f -iname \*.jpeg -delete")
        print '\033[1;44m#########################################################\033[1;m'
        print '\033[1;44m##################### VIDEO READY #######################\033[1;m'
        print '\033[1;44m#########################################################\033[1;m'
 
if __name__ == "__main__":
    main()

try...finally around this?

See:

• https://docs.python.org/2/library/signal.html

CTRL-C is SIGINT.

Something like this:

import signal

def signal_handler(signal, frame):
    print('Exit!!!')
    sys.exit(0)

if __name__ == '__main__':
    signal.signal(signal.SIGINT, signal_handler)
    main()