paint.py 文件源码

def save(filename):
    try:
        os.mkdir('saves/')
    except OSError:
        pass
    try:
        data = unicorn.get_pixels()
        print(filename, data)
        file = open('saves/' + filename + '.py', 'w')
        file.write('#!/usr/bin/env python\n')
        file.write('import unicornhat, signal\n')
        file.write('pixels = ' + str(unicorn.get_pixels()) + '\n')
        file.write('unicornhat.set_pixels(pixels)\n')
        file.write('unicornhat.show()\n')
        file.write('signal.pause()')
        file.close()
        os.chmod('saves/' + filename + '.py', 0o777 | stat.S_IEXEC)

        return("ok" + str(unicorn.get_pixels()))
    except AttributeError:
        print("Unable to save, please update")
        print("unicornhat library!")
        return("fail")