menu python command line

Question

У меня есть это меню. Он динамически загружает класс и вызывает метод. Все в порядке, но метод печатает привет, только когда я пишу 'exit'. Любое предложение приветствуется?

class ClassA(object): 
    def main(self): 
     libs = self.directory() 
     print '***************************************************' 
     print '* Menu' 
     for item in libs : 
      print '** ' + item 
     print '***************************************************' 

     while True: 
      module_name = raw_input(">>> ") 

      if module_name != "": 
       if module_name == 'exit': 
        break 
       elif module_name in libs: 
        lib = __import__(module_name) 
        class_ = getattr(lib, module_name) 
        theclass = class_() 
        theclass.myfunction() 
        break 
       else: 
        print 'Module does not exist' 

Класс B

class ClassB(object): 
    def myfunction(self): 
     print 'hello' 

Answer 1

После некоторых основных изменения в код, это может быть то, что вы хотите:

import glob 

class ClassB(): 
    def myfunction(self): 
     print 'hello' 

if __name__ == '__main__': 
    libs = [item[:-3] for item in glob.glob("*.py")] 
    print '***************************************************' 
    print '* Menu' 
    for item in libs : 
     print '** ' + item 
    print '***************************************************' 

    while True: 
     module_name = raw_input(">>> ") 
     if module_name != "": 
      if module_name == 'exit': 
       break 
      if module_name in libs: 
       lib = __import__(module_name) 
       class_ = getattr(lib, 'ClassB') 
       theclass = class_() 
       theclass.myfunction() 
       break 
      else: 
       print 'Module does not exist'