#!/usr/bin/env python  
import os

mini_image="/opt/ltsp/mini-light-client"
pxelinux_create="/usr/share/lliurex-ltsp/llx-create-pxelinux.sh"

def extract_file():
        if os.path.exists(mini_image+'.tar.lzma'):
		print "Change Directory to /opt/ltsp/"
                os.chdir('/opt/ltsp/')
                print "Decompress minimal light client image"
                os.system('tar --lzma -xf '+mini_image+'.tar.lzma')

def create_image_ltsp():
	print "Update sshkeys update"
        os.system('ltsp-update-sshkeys')
	print "Update Image ltsp"
        os.system('ltsp-update-image '+mini_image)
        print "Finished regenerate minimal light image"
	# Restart ndb-server
	os.system('service nbd-server restart')


if __name__=="__main__":
        try:
                if os.path.exists(mini_image):
                        create_image_ltsp()
                else:
                        extract_file()
                        create_image_ltsp()

        except Exception as e:
                print e