StatusBarOverlay.py

StatusBarOverlay.py

# coding: utf-8
''' Statusbarmenu
A menu that lives in the status bar
A set of tools to add or delete custom buttons from the toolbar.  This may not be super robust, but seems to work ok.  Button objects and actions are saved so they survive global clearing, but thid has not been tested extensively.  If a function relies on imports that occured outside of the function, these might dissappear -- user must make sure those modules are added to a module that is kept by pythonista, such as anything in site-packages, or name starting with __   '''

from objc_util import *
import ui,console,editor
import weakref
from functools import partial
MARGIN=10
HEIGHT=32
CORNERWIDTH=80
try:
   __s.teardown()
   del __s
except NameError:
   pass

class StatusBarOverlay(object):
   def __init__(self):
      app=UIApplication.sharedApplication()
      self.bar=app.statusBar()
      b=self.bar.bounds()

      self.overlay=ui.View(frame=      (b.origin.x,b.origin.y,20,b.size.height+12))
      self.overlay.flex='lwh'
      self.overlay.y=-self.overlay.height
      #self.overlay.border_color='#ff0000'
      self.overlay.border_width=1
      self.overlay.corner_radius=5
      self.bar.addSubview_(self.overlay)
      btn=ui.Button(frame=(0,0,CORNERWIDTH,20))
      btn.height=20
      btn.width=CORNERWIDTH
      btn.flex='l'
      btn.alpha=.6
      btn.x=b.size.width-CORNERWIDTH
      btn.action=self.toggleBar
      with ui.ImageContext(CORNERWIDTH,20) as ctx:
         p=ui.Path()
         p.move_to(0,0)
         p.move_to(CORNERWIDTH-30,0)
         p.line_to(CORNERWIDTH,20)
         p.line_to(CORNERWIDTH,0)
         #p.line_to(CORNERWIDTH,20)
         #p.line_to(0,20)
         p.line_width=4
         p.close()
         ui.set_color((1,0,0,1))
         p.stroke()
         p.fill()
         btn.image=ctx.get_image()
      self.btn=btn
      self.overlay.bg_color='white'
      self.expanded=False
      self.bar.addSubview_(btn)
      self.overlay.x=self.btn.x+CORNERWIDTH
      self.overlay.x0=MARGIN
      self.overlay.bring_to_front
      self.btn.bring_to_front
   def toggleBar(self,sender):
      ''' display self.overlay'''
      if self.expanded:
         #hide
         def ani():
            self.overlay.x=self.btn.x-self.overlay.x0
            #self.overlay.alpha=0
            self.overlay.y=-self.overlay.height
      else: #show
         def ani():
            self.overlay.x=self.btn.x-self.overlay.x0
            #self.overlay.alpha=1
            self.overlay.y=0
      self.expanded=not (self.expanded)
      ui.animate(ani,.25)
   def add_action(self,action,title=None,image=None):
      '''adds a button to the overlay, at next position'''
      b=ui.Button(title=title,image=image,action=action)
      b.size_to_fit()
      self.overlay.add_subview(b)
      b.x=self.overlay.x0
      self.overlay.x0+=b.width+MARGIN
      self.overlay.width=self.overlay.x0
      self.overlay.x=self.btn.x-self.overlay.x0-MARGIN
   @on_main_thread
   def teardown(self):
      '''cleanup...'''
      self.overlay.objc_instance.removeFromSuperview()
      self.btn.objc_instance.removeFromSuperview()
   def __del__(self):
      self.teardown()




if __name__=='__main__':
   #for debug only:
   app=UIApplication.sharedApplication()
   bar=app.statusBar()
   
   #dummy action
   def a(sender):
      '''generic action, to print x position of button'''
      print(sender.x)
   
   __s=StatusBarOverlay()   #dunder retains in globals...
   __s.add_action(a,'Title') #title only
   __s.add_action(a,image=ui.Image.named('iow:archive_32')) #image only
   __s.add_action(a,'Blah')
   __s.add_action(a,'Find',ui.Image.named('iow:search_32')) #title and image
   __s.add_action(a,'Copy',ui.Image.named('iow:clipboard_32')) #title and image