uidir.py

# coding: utf-8

import ui
import os
import console

class MyTableViewDataSource (object):
   # sel = [None]
    
    def __init__(self,setter, base_dir = '.'):
        self.dir = os.path.abspath(base_dir)
        #ffself.rel_dir=os.path.abspath(rel_dir)
        self.setter=setter
        _, folders, files = next(os.walk(base_dir))
        folders.insert(0,'..')
        self.data = (folders,files)
        self.sel = [None]
    def tableview_number_of_sections(self, tableview):
        return 2

    def tableview_number_of_rows(self, tableview, section):
        return len(self.data[section])

    def tableview_cell_for_row(self, tableview, section, row):

        cell = ui.TableViewCell()
        cell.accessory_type = ('disclosure_indicator', 'detail_button')[section]
        cell.text_label.text = self.data[section][row]
        if section==0:
            cell.background_color='#eeffee'
            
        return cell

    def tableview_title_for_header(self, tableview, section):
        return ('Folders','Files')[section]

    def tableview_did_select(self, tableview, section, row):
        '@type tableview: ui.TableView'
        if section == 0:
            dir = os.path.normpath(os.path.join(self.dir, self.data[section][row]))
            if os.path.exists(dir):
               try:
                  def setme(value):
                     # set and bubble up setters
                     self.sel[0]=value
                     if self.setter is not None:
                        self.setter(value)
                     
                  newv = FileViewer(setter=setme,base_dir=dir)
                  nav = tableview.superview.navigation_view
                  nav.push_view(newv)
               except StopIteration:
                  console.hud_alert('Folder not accessible','error')

        else:
           # print self.dir, self.data[section][row]
            self.sel[0] = os.path.join(self.dir, self.data[section][row])
            tableview.superview.navigation_view.close()
            if self.setter is not None:
                self.setter(self.sel[0])
       


    def tableview_accessory_button_tapped(self, tableview, section, row):
        full = os.path.join(self.dir,self.data[section][row])
        stats =  os.stat(full)
        console.hud_alert('Size: {0} KB'.format(stats.st_size//1024))


class FileViewer(ui.View):
    def __init__(self,setter, base_dir = '.', *args, **kargs):
        self.table = ui.TableView(*args, **kargs)
        self.name=os.path.split(os.path.abspath(base_dir))[1]
        self.src = MyTableViewDataSource(setter, base_dir)
        self.table.data_source = self.src
        self.table.delegate = self.src
        self.table.flex = 'WHTBLR'
        self.setter=setter
        #self.view = ui.View(name = base_dir)
        self.background_color = 'white'
        self.add_subview(self.table)

    @property
    def selection(self):
        return self.src.sel[0]

def getFile(setter=None,base_dir='.'):
    fv = FileViewer(setter,base_dir)
    fv.height=700
    nv = ui.NavigationView(fv)
    
    def openDocuments(sender,path):
       def setme(fv,value):
       # set and bubble up setters
           fv.src.sel[0]=value
           if fv.src.setter is not None:
              fv.src.setter(value)
       newfv = FileViewer(setter=lambda value:setme(fv,value),base_dir=path)
       nv.push_view(newfv)
       
       
    nv.right_button_items=[
        ui.ButtonItem(title='Documents',
         action=lambda sender:openDocuments(sender,os.path.expanduser('~/Documents'))), 
        ui.ButtonItem(title='Library',
         action=lambda sender:openDocuments(sender,os.path.split(os.__file__)[0]))]
    nv.height=800
    nv.width=500
    nv.name = 'File Selector'
    nv.present('popover')
    ui.in_background(nv.wait_modal)
    nv.wait_modal()
    return fv.src.sel[0]
if __name__=='__main__':
   def setter(s):
      print 'setter recieved' + s
   f=getFile(setter=setter)
   print 'getFile returned',f