Exemple #1
0
def sendsMessageToRedis(userid, jobid, source_type, socketid, complete_output,
                        result_path=None, result_url=None, result_text=None, dropbox_token=None):
    #logger.write('P', 'Inside send message to redis')
    try:

        r.hset(jobid, 'output', str(complete_output))
        r.hset(jobid, 'socketid', str(socketid))

        if source_type != 'dropbox':
            if result_url is not None:
                r.hset(jobid, 'result_path', str(jobid) + '_resultdir')
                for file_name in os.listdir(result_path):
                    if os.path.isfile(os.path.join(result_path,file_name)):
                        r.lpush(str(jobid) + '_resultdir', os.path.join(result_url, str(file_name)))
            elif result_text is not None:
                r.hset(jobid, 'output', result_text)

        elif source_type == 'dropbox':
            f = open(result_path.rstrip('/') + '/output.txt', 'w')
            if complete_output is not None or complete_output is not '':
                f.write(str(complete_output))
            if result_text is not None:
                f.write(str(result_text))
            f.close()

            response, url = dbu.upload_files_to_dropbox(userid, jobid, result_path, dropbox_token)
            r.hset(jobid, 'output', str(response) + '\n' + str(url))
    except Exception as e:
        raise e
def sendsMessageToRedis(userid, jobid, source_type, socketid, complete_output,
                        result_path=None, result_url=None, result_text=None, dropbox_token=None):
    #logger.write('P', 'Inside send message to redis')
    try:

        r.hset(jobid, 'output', str(complete_output))
        r.hset(jobid, 'socketid', str(socketid))

        if source_type != 'dropbox':
            if result_url is not None:
                r.hset(jobid, 'result_path', str(jobid) + '_resultdir')
                for file_name in os.listdir(result_path):
                    if os.path.isfile(os.path.join(result_path,file_name)):
                        r.lpush(str(jobid) + '_resultdir', result_url + str(file_name))
            elif result_text is not None:
                r.hset(jobid, 'output', result_text)

        elif source_type == 'dropbox':
            f = open(result_path.rstrip('/') + '/output.txt', 'w')
            if complete_output is not None:
                f.write(str(complete_output))
            if result_text is not None:
                f.write(str(result_text))
            f.close()

            response, url = dbu.upload_files_to_dropbox(userid, jobid, result_path, dropbox_token)
            r.hset(jobid, 'output', str(response) + '\n' + str(url))
    except Exception as e:
        r.publish('chat', json.dumps({'error': str(traceback.format_exc()), 'socketid': str(socketid)}))
        raise e
Exemple #3
0
def sendsMessageToRedis(userid,
                        jobid,
                        source_type,
                        socketid,
                        complete_output,
                        result_path=None,
                        result_url=None,
                        result_text=None,
                        dropbox_token=None):
    # logger.write('P', 'Inside send message to redis')
    from app.thirdparty import dropbox_upload as dbu
    try:

        r.hset(jobid, 'output', str(complete_output))
        r.hset(jobid, 'socketid', str(socketid))

        if source_type != 'dropbox':
            if result_url is not None:
                r.hset(jobid, 'result_path', str(jobid) + '_resultdir')
                for file_name in os.listdir(result_path):
                    if os.path.isfile(os.path.join(result_path, file_name)):
                        r.lpush(
                            str(jobid) + '_resultdir',
                            os.path.join(result_url, str(file_name)))
            elif result_text is not None:
                r.hset(jobid, 'output', result_text)

        elif source_type == 'dropbox':
            f = open(result_path.rstrip('/') + '/output.txt', 'w')
            if complete_output is not None or complete_output is not '':
                f.write(str(complete_output))
            if result_text is not None:
                f.write(str(result_text))
            f.close()

            response, url = dbu.upload_files_to_dropbox(
                userid, jobid, result_path, dropbox_token)
            r.hset(jobid, 'output', str(response) + '\n' + str(url))
    except Exception as e:
        raise e