file_select_gui.py

# -*- coding: utf-8 -*-
"""
Created on Thu Jan 12 15:10:28 2017

Small functions to select file or directory paths via a GUI

@author: Robin Amsters
@email: robin.amsters@gmail.com
"""
import numpy as np
import os

from fnmatch import fnmatch
from Tkinter import Tk
from tkFileDialog import askopenfile , askdirectory

def get_file_path(msg):
    # Selecting file trough GUI
    Tk().withdraw() # we don't want a full GUI, so keep the root window from appearing
    filePath = askopenfile(title = msg) # show an "Open" dialog box and return the path to the selected file
    return filePath

def get_directory_path(msg):
    # Selecting directory trough GUI
    Tk().withdraw() # we don't want a full GUI, so keep the root window from appearing
    filePath = askdirectory(title = msg) # show an "Open" dialog box and return the path
    return filePath

def get_all_files_directory(directory, pattern):
    """
        Function that returns all the path to all files in a directory
        and its subdirectories that end in a certain pattern.
        
    """
    
    root = directory
    allFiles = np.array([])
    
    for path, subdirs, files in os.walk(root):
        for name in files:
            if fnmatch(name, pattern):
                allFiles = np.append(allFiles, os.path.join(path, name))
    
    return allFiles

def get_folder_names(directory):
    
    folders = os.walk(directory).next()[1]
    
    return folders


if __name__=='__main__':
    # list all rosbag files
    directory = get_directory_path('select directory')
    print(get_all_files_directory(directory, '*.bag'))