가져온 함수가 반복되지 않습니다. "TypeError : 'NoneType'object> iserable이 아닙니다. ... 어떻게이 작업을 수행합니까?

Question

이것은 내가 지금 작업하고있는 파이썬의 스 니펫입니다.이 코드를 모두 나열하지 않았으므로 필요한 것이 "누락되었습니다"라고 사과하는 경우 사과 드리겠습니다. 그것 ...

아래에는 기능이 있습니다 main() - 이것은 내 스크립트에 명시 적으로 정의되어 있지 않습니다 - 다른 사람이 만든 다른 스크립트에서 가져 왔습니다. 호출 될 때 파이썬이 호출 할 수있는 모든 사용 가능한 모듈의 매우 긴 목록을 출력합니다. 각 모듈에 줄 번호를 추가하려고합니다. 그래서 출력 할 때 매우 긴 모듈 이름 목록입니다. (하나의 모듈을 출력하고 새로운 줄을 그리고 나서 다른 모듈 이름을 출력하기 때문에 각 모듈 다음에 ""이 나옵니다.)라고 가정합니다. 내가하려고하는 것은 그 값을 취하여 각 모듈 이름 앞에 줄 번호를 추가하는 것입니다.

elif x == "list" or x == "1": 

      print "\n loading... please wait" 
      time.sleep(2) 
      counter=0 

      lnumber = 0 
      all_mods = (main()) 

      for x in all_mods: 
       print lnumber, x 
       lnumber = lnumber + 1 
       counter = counter + 1 

      print "-" * 30, "\nTotal number of modules detected: ", counter 

는 ** 나는 내가 다른 곳에서이 설정을 사용하지 않은 문서에있는 코드와 일치하기 때문에 내가 일부러 이런 짓을의 lnumber 및 카운터 같은 일을보고하고 이해합니다.

이 코드 조각은 (스크립트의 다른 부분과) 실행됩니다

이 다시보고 :

Traceback (most recent call last): 
    File "C:\Users\jc\Documents\Python Projects\Projects\myOwnfns\helpwiz.py", line 131, in <module> 
    main_loop() 
    File "C:\Users\jc\Documents\Python Projects\Projects\myOwnfns\helpwiz.py", line 90, in main_loop 
    for x in all_mods: #this variable comes from "list_all_mods" - an external script taken from another author. 
TypeError: 'NoneType' object is not iterable 

Answer 1

기능 main인쇄 표준 출력에 라인; 아무것도 반환하지 않습니다. 더 정확하게는 None 객체를 반환하므로 all_mods is None입니다. 그 이유는 "NoneType '객체가 반복 가능하지 않기 때문입니다. 왜냐하면 당신이 for x in all_mods으로 반복을 시도하기 때문입니다.

import sys, StringIO 
buffer = StringIO.StringIO() 
sys.stdout = buffer 
main() 
buffer.seek(0) 
all_mods = buffer.read().splitlines() 
sys.stdout = sys.__stdout__ 

Answer 2

내가 모든 가능한 경우, 리팩토링 list_all_mods()을 권하고 싶습니다 :

여기에 작동합니다 정말 hackish 솔루션입니다. 특히 값을 인쇄하지 말고 값을 반환하도록 변경하십시오. 또는 목록을 반환하는 새로운 함수 find_all_mods()으로 바꿀 및 list_all_mods()을 재정의 :

def list_all_mods(): 
    print '\n'.join(find_all_mods()) 

를 내가 당신의 코드가 아닙니다 알고, 그래서 이것은 옵션이 될하지 않을 수 있습니다. 그렇지 않다면 balpha의 해킹이 아마도 당신이 할 수있는 최선일 것입니다. 그 후, 행 번호와 모듈을 인쇄하려면, 당신이 할 수 있습니다

for (i, module_name) in enumerate(all_mods): 
    # n.b.: we use i+1 because we want numbering to start from 1. 
    print "%4d %s" % (i+1, module_name) 

추신, 난 당신을 "모든 모듈"로 계산 무엇인지 모르겠지만, 현재 수입하고있는 모듈의 경우, 당신은 할 수 그냥 sys.modules.keys()을보고 이걸 얻으세요.

Answer 3

nl 명령은이 작업을 수행합니다.

python the_existing_program.py | nl 

원하는대로 할 수 있습니다.

Windows에서 작업하는 경우 nl 버전을 매우 쉽게 작성할 수 있습니다.

import fileinput 
for n, line in enumerate(fileinput.input()): 
    print "%d %s" % (n, line) 

전화를 nl.py라고 가정 해 봅시다.

python the_existing_program.py | python nl.py 

원본 프로그램을 수정할 필요가 없습니다.