def __init__(self, path=os.getcwd(), extensions=[], title="Select File",
width=540, height=300, window=None, batch=None, group=None,
anchor=ANCHOR_CENTER, offset=(0, 0),
theme=None, movable=True, on_select=None, on_escape=None):
self.path = path
self.extensions = extensions
self.title = title
self.on_select = on_select
self.selected_file = None
self._set_files()
def on_parent_menu_select(choice):
self._select_file(self.parents_dict[choice])
def on_menu_select(choice):
self._select_file(self.files_dict[choice])
self.dropdown = Dropdown(options=self.parents,
selected=self.parents[-1],
align=VALIGN_BOTTOM,
on_select=on_parent_menu_select)
self.menu = Menu(options=self.files, align=HALIGN_LEFT,
on_select=on_menu_select)
self.scrollable = Scrollable(
VerticalLayout([self.dropdown, self.menu], align=HALIGN_LEFT),
width=width, height=height)
content = self._get_content()
Dialog.__init__(self, content, window=window, batch=batch, group=group,
anchor=anchor, offset=offset, theme=theme,
movable=movable, on_escape=on_escape)
def __init__(self, path=os.getcwd(), extensions=[], title="Select File",
width=540, height=300, window=None, batch=None, group=None,
anchor=ANCHOR_CENTER, offset=(0, 0),
theme=None, movable=True, on_select=None, on_escape=None):
self.path = path
self.extensions = extensions
self.title = title
self.on_select = on_select
self.selected_file = None
self._set_files()
def on_parent_menu_select(choice):
self._select_file(self.parents_dict[choice])
def on_menu_select(choice):
self._select_file(self.files_dict[choice])
self.dropdown = Dropdown(options=self.parents,
selected=self.parents[-1],
align=VALIGN_BOTTOM,
on_select=on_parent_menu_select)
self.menu = Menu(options=self.files, align=HALIGN_LEFT,
on_select=on_menu_select)
self.scrollable = Scrollable(
VerticalLayout([self.dropdown, self.menu], align=HALIGN_LEFT),
width=width, height=height)
content = self._get_content()
Dialog.__init__(self, content, window=window, batch=batch, group=group,
anchor=anchor, offset=offset, theme=theme,
movable=movable, on_escape=on_escape)
class FileLoadDialog(Dialog):
def __init__(self, path=os.getcwd(), extensions=[], title="Select File",
width=540, height=300, window=None, batch=None, group=None,
anchor=ANCHOR_CENTER, offset=(0, 0),
theme=None, movable=True, on_select=None, on_escape=None):
self.path = path
self.extensions = extensions
self.title = title
self.on_select = on_select
self.selected_file = None
self._set_files()
def on_parent_menu_select(choice):
self._select_file(self.parents_dict[choice])
def on_menu_select(choice):
self._select_file(self.files_dict[choice])
self.dropdown = Dropdown(options=self.parents,
selected=self.parents[-1],
align=VALIGN_BOTTOM,
on_select=on_parent_menu_select)
self.menu = Menu(options=self.files, align=HALIGN_LEFT,
on_select=on_menu_select)
self.scrollable = Scrollable(
VerticalLayout([self.dropdown, self.menu], align=HALIGN_LEFT),
width=width, height=height)
content = self._get_content()
Dialog.__init__(self, content, window=window, batch=batch, group=group,
anchor=anchor, offset=offset, theme=theme,
movable=movable, on_escape=on_escape)
def _get_content(self):
return Frame(
VerticalLayout([
SectionHeader(self.title),
self.scrollable,
], align=HALIGN_LEFT)
)
def _select_file(self, filename):
if os.path.isdir(filename):
self.path = filename
self._set_files()
self.dropdown.set_options(self.parents,
selected=self.parents[-1])
self.menu.set_options(self.files)
else:
self.selected_file = filename
if self.on_select is not None:
self.on_select(filename)
def _set_files(self):
# Once we have a new path, update our files
filenames = glob.glob(os.path.join(self.path, '*'))
# First, a list of directories
self.parents = []
self.parents_dict = {}
path = self.path
index = 1
while 1:
name = "%d %s" % (index, os.path.basename(path) or path)
self.parents_dict[name] = path
self.parents.append(name)
index += 1
path, child = os.path.split(path)
if not child:
break
self.parents.reverse()
files = [("%s (dir)" % os.path.basename(x), x) for x in filenames
if os.path.isdir(x)]
# Now add the files that match the extensions
if self.extensions:
for filename in filenames:
if os.path.isfile(filename):
ext = os.path.splitext(filename)[1]
if ext in self.extensions:
files.append((os.path.basename(filename), filename))
else:
files.extend([(os.path.basename(x), x) for x in filenames
if os.path.isfile(x)])
self.selected_file = None
self.files_dict = dict(files)
self.files = self.files_dict.keys()
def dir_sort(x, y):
if x.endswith(' (dir)') and y.endswith(' (dir)'):
return cmp(x, y)
elif x.endswith(' (dir)'):
return -1
elif y.endswith(' (dir)'):
return 1
else:
return cmp(x, y)
self.files.sort(dir_sort)
def get(self):
return self.selected_file
def size(self, dialog):
Dialog.size(self, dialog)
def teardown(self):
self.on_select = None
Dialog.teardown(self)
class FileLoadDialog(Dialog):
def __init__(self, path=os.getcwd(), extensions=[], title="Select File",
width=540, height=300, window=None, batch=None, group=None,
anchor=ANCHOR_CENTER, offset=(0, 0),
theme=None, movable=True, on_select=None, on_escape=None):
self.path = path
self.extensions = extensions
self.title = title
self.on_select = on_select
self.selected_file = None
self._set_files()
def on_parent_menu_select(choice):
self._select_file(self.parents_dict[choice])
def on_menu_select(choice):
self._select_file(self.files_dict[choice])
self.dropdown = Dropdown(options=self.parents,
selected=self.parents[-1],
align=VALIGN_BOTTOM,
on_select=on_parent_menu_select)
self.menu = Menu(options=self.files, align=HALIGN_LEFT,
on_select=on_menu_select)
self.scrollable = Scrollable(
VerticalLayout([self.dropdown, self.menu], align=HALIGN_LEFT),
width=width, height=height)
content = self._get_content()
Dialog.__init__(self, content, window=window, batch=batch, group=group,
anchor=anchor, offset=offset, theme=theme,
movable=movable, on_escape=on_escape)
def _get_content(self):
return Frame(
VerticalLayout([
SectionHeader(self.title),
self.scrollable,
], align=HALIGN_LEFT)
)
def _select_file(self, filename):
if os.path.isdir(filename):
self.path = filename
self._set_files()
self.dropdown.set_options(self.parents,
selected=self.parents[-1])
self.menu.set_options(self.files)
else:
self.selected_file = filename
if self.on_select is not None:
self.on_select(filename)
def _set_files(self):
# Once we have a new path, update our files
filenames = glob.glob(os.path.join(self.path, '*'))
# First, a list of directories
self.parents = []
self.parents_dict = {}
path = self.path
index = 1
while 1:
name = "%d %s" % (index, os.path.basename(path) or path)
self.parents_dict[name] = path
self.parents.append(name)
index += 1
path, child = os.path.split(path)
if not child:
break
self.parents.reverse()
files = [("%s (dir)" % os.path.basename(x), x) for x in filenames
if os.path.isdir(x)]
# Now add the files that match the extensions
if self.extensions:
for filename in filenames:
if os.path.isfile(filename):
ext = os.path.splitext(filename)[1]
if ext in self.extensions:
files.append((os.path.basename(filename), filename))
else:
files.extend([(os.path.basename(x), x) for x in filenames
if os.path.isfile(x)])
self.selected_file = None
self.files_dict = dict(files)
self.files = self.files_dict.keys()
def dir_sort(x, y):
if x.endswith(' (dir)') and y.endswith(' (dir)'):
return cmp(x, y)
elif x.endswith(' (dir)'):
return -1
elif y.endswith(' (dir)'):
return 1
else:
return cmp(x, y)
self.files.sort(dir_sort)
def get(self):
return self.selected_file
def size(self, dialog):
Dialog.size(self, dialog)
def teardown(self):
self.on_select = None
Dialog.teardown(self)
