使用curses和panel在Python中运行系统命令,然后返回上一级菜单

时间:2013-11-07 10:19:09

标签: python subprocess panel curses

我使用顶级的几个commanline工具编写python脚本,所以我需要一个适当的视觉反馈。现在是时候给它一个菜单了,所以问题出现了。

I found here a great approach我需要什么,但每次尝试在回到上一个菜单之前显示反馈都是徒劳的。

我只需要菜单,子菜单,启动命令,终止它,然后返回上一级菜单。一个伟大的奖金将是在一个分裂的期限运行它们。

是否有任何模式/骨架/东西/用作模板以显示具有可预测输出的几种小部件?

这里是一个代码示例,其中有两个要运行的函数示例:

#!/usr/bin/env python2                                                       

import curses                                                                
from curses import panel                                                     

class Menu(object):                                                          

    def __init__(self, items, stdscreen):                                    
        self.window = stdscreen.subwin(0,0)                                  
        self.window.keypad(1)                                                
        self.panel = panel.new_panel(self.window)                            
        self.panel.hide()                                                    
        panel.update_panels()                                                

        self.position = 0                                                    
        self.items = items                                                   
        self.items.append(('exit','exit'))                                   

    def navigate(self, n):                                                   
        self.position += n                                                   
        if self.position < 0:                                                
            self.position = 0                                                
        elif self.position >= len(self.items):                               
            self.position = len(self.items)-1                                

    def display(self):                                                       
        self.panel.top()                                                     
        self.panel.show()                                                    
        self.window.clear()                                                  

        while True:                                                          
            self.window.refresh()                                            
            curses.doupdate()                                                
            for index, item in enumerate(self.items):                        
                if index == self.position:                                   
                    mode = curses.A_REVERSE                                  
                else:                                                        
                    mode = curses.A_NORMAL                                   

                msg = '%d. %s' % (index, item[0])                            
                self.window.addstr(1+index, 1, msg, mode)                    

            key = self.window.getch()                                        

            if key in [curses.KEY_ENTER, ord('\n')]:                         
                if self.position == len(self.items)-1:                       
                    break                                                    
                else:                                                        
                    self.items[self.position][1]()                           

            elif key == curses.KEY_UP:                                       
                self.navigate(-1)                                            

            elif key == curses.KEY_DOWN:                                     
                self.navigate(1)                                             

        self.window.clear()                                                  
        self.panel.hide()                                                    
        panel.update_panels()                                                
        curses.doupdate()

######################################################### !# 
######################################################### !#    
############# HERE MY FUNCTIONS examples
############  Everithing works OK, but displays it awfully

def GetPid(name):
    import subprocess
    command= str(("""pgrep %s""") % name )
    p = subprocess.Popen(command, shell = True, stdout = subprocess.PIPE)
    procs = []
    salida = p.stdout
    for line in salida:
        procs.append(str.strip(line))

    return procs


def top():
    os.system("top")    

def menuGP():
    print GetPid("top")  


######################################################### !# 

class MyApp(object):                                                         

    def __init__(self, stdscreen):                                           
        self.screen = stdscreen                                              
        curses.curs_set(0)                                                   

        submenu_items = [                                                    
                ('beep', curses.beep),                                       
                ('top', top)                                      
                ]                                                            
        submenu = Menu(submenu_items, self.screen)                           

        main_menu_items = [                                                  
                ('get PID', GetPid),                                       
                ('submenu', submenu.display)                                 
                ]                                                            
        main_menu = Menu(main_menu_items, self.screen)                       
        main_menu.display()                                                  

if __name__ == '__main__':                                                       
    curses.wrapper(MyApp)

谢谢你的建议(对不起我粗略的英语)

1 个答案:

答案 0 :(得分:6)

你真的有两个选择。一个你可以离开curses模式,执行你的程序,然后恢复curses。二,您可以异步执行程序,解析其输出并将其写入屏幕。

第一个选项的好消息是你实际上不需要为ui编写任何花哨的save_state / load_state方法。诅咒为你做这件事。这是一个显示我的观点的简单例子

import curses, time, subprocess

class suspend_curses():
    """Context Manager to temporarily leave curses mode"""

    def __enter__(self):
        curses.endwin()

    def __exit__(self, exc_type, exc_val, tb):
        newscr = curses.initscr()
        newscr.addstr('Newscreen is %s\n' % newscr)
        newscr.refresh()
        curses.doupdate()

def main(stdscr):
    stdscr.addstr('Stdscreen is %s\n' % stdscr)
    stdscr.refresh()
    time.sleep(1)

    with suspend_curses():
        subprocess.call(['ls'])
        time.sleep(1)

    stdscr.refresh()
    time.sleep(5)

curses.wrapper(main)

如果您运行该示例,您会注意到curses.wrapper创建的屏幕和恢复时在curses.initscr中创建的屏幕是同一个对象。也就是说,curses.initscr返回的窗口是单例。这使我们可以像上面一样退出curses和恢复,而不必每次都更新每个小部件的self.screen引用。

第二种选择涉及更多,但也更灵活。以下只是代表基本思想。

class procWidget():
    def __init__(self, stdscr):
        # make subwindow / panel
        self.proc = subprocess.Popen(my_args, stdout=subprocess.PIPE)

    def update(self):
        data = self.proc.stdout.readline()
        # parse data as necessary
        # call addstr() and refresh()

然后在程序的某个地方,您需要在计时器上的所有procWidgets上调用update。这使您可以选择使您的子窗口任意大小/位置,以便您可以拥有尽可能多的procWidgets。当然,您必须为流程终止时添加一些处理以及其他类似事件。