main.py

main.py

# coding: utf-8

# original code : https://github.com/jsbain/uicomponents/blob/master/uidir.py

import ui
import os
import console

class MyTableViewDataSource (object):
   # sel = [None]
    
    def __init__(self,setter, base_dir = '.'):
        self.dir = base_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.join(self.dir, self.data[section][row])
            if os.path.exists(dir):
                self.dir=dir
            newv = FileViewer(self.setter,self.dir)
            nav = tableview.superview.navigation_view
            nav.push_view(newv)
        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.table.name = 'FileTable'
        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)
    nv.height=800
    nv.name = 'File Selector'
    nv.present('popover')
    #ui.in_background(nv.wait_modal)
    nv.wait_modal()
    return fv.selection
    
if __name__=='__main__':
    class myc:
        def __init__(self):
            self.content = ''
        def __call__(self,x):
            self.content = x
    q=myc()
    getFile(setter=q)
    #unprinted content under xcode project 1.6 beta
    print "Selected File =",q.content #'log_high.csv'
    #