hackedteam/test-av2

import os
import sys
from AVCommon.logger import logging
from AVCommon import logger

import time


def execute(vm, protocol, img_path):
    """ server side """
    # TODO: add Proc name
    from AVMaster import vm_manager

    #logging.debug("    CS Execute")
    assert vm, "null vm"

    basedir = "%s/screenshots" % logger.logdir

    if not img_path:
        if not os.path.exists(basedir):
            os.makedirs(basedir)
        img_path = "%s/%s.%s.png" % (basedir, vm, int(time.time()))

    ret = vm_manager.execute(vm, "takeScreenshot", img_path)
    if ret is True:
        #blob = open(img_path).read()
        return ret, img_path
    else:
        return ret, "Screenshot not saved"