bp_utils.py

Go to the documentation of this file.

import os
import sys

import bp_config as config             #/// User Configuration

def progressBar(value, endvalue, docdb_id, bar_length=40):
    percent = float(value) / endvalue
    arrow = '=' * int(round(percent * bar_length)-1) + '>'
    spaces = ' ' * (bar_length - len(arrow))

    sys.stdout.write("\rProgress: [\033[01;32m{0}\033[00m] {1}/{2} ({3}%) -- DocDB {4}".format(arrow + spaces, value, endvalue, int(round(percent * 100)), docdb_id))
    sys.stdout.flush()
        

def ListBlessedFilesInDir(dir):
    ret = []
    for root, dirs, files in os.walk(dir):
        # ignore weird system directories
        if root.find('__MACOSX') > 0: continue
        
        # loop over files in dir and all subdirectories
        for file in files:
            # ignore dotfilfes
            if file.startswith('.'): continue
            
            # ignore files that don't have an expected .txt extension
            if os.path.splitext(os.path.basename(file))[1][1:] not in config.CAPTION_EXTS: continue
            ret.append('{}/{}'.format(root, file)[2:])
            
            # find all files with images extensions that are associated with this caption file
            for file2 in files:
                if os.path.splitext(os.path.basename(file2))[0] != os.path.splitext(os.path.basename(file))[0] or os.path.splitext(os.path.basename(file2))[1][1:] not in config.PLOT_EXTS: continue
                
                ret.append('{}/{}'.format(root, file2)[2:])

    return ret


def ListCaptionFilesInDir(dir):
    ret = []
    for root, dirs, files in os.walk(dir):
        # ignore weird system directories
        if root.find('__MACOSX') > 0: continue
        
        # loop over files in dir and all subdirectories
        for file in files:
            # ignore dotfilfes
            if file.startswith('.'): continue
            
            # ignore files that don't have an expected .txt extension
            if os.path.splitext(os.path.basename(file))[1][1:] not in config.CAPTION_EXTS: continue
            ret.append('{}/{}'.format(root, file)[2:])

    return ret


def ListPlotsInDirForCaption(dir, cap):
    ret = []    
    for root, dirs, files in os.walk(dir):
        # ignore weird system directories
        if root.find('__MACOSX') > 0: continue
        
        # loop over files in dir and all subdirectories
        for file in files:
            # ignore dotfilfes
            if file.startswith('.'): continue
            
            if os.path.splitext(os.path.basename(file))[0] != os.path.splitext(os.path.basename(cap))[0] or os.path.splitext(os.path.basename(file))[1][1:] not in config.PLOT_EXTS: continue
            
            ret.append('{}/{}'.format(root, file)[2:])

    return ret