필자는 파이썬 함수를 사용하여 행을 반복한다. 문자열이나 파일과 같은 객체를 받아 들인다.

Question

입력 인자의 라인을 반복 처리하지만 문자열이나 파일과 같은 객체를 받아 들일 수있는 파이썬 함수를 작성하고 싶다. 이것을 할 수있는 방법이 있습니까?

def myfunc(x): 
    for line in x: 
     doSomething(line) 

위의 코드는 문자열을 파일 - 류의 객체에 대한 아니라 작동하고, 내 진짜 작업이 위보다 훨씬 더 복잡하기 때문에 차라리 두 개가 아닌 하나의 함수를 작성할 것입니다.

Answer 1

만 변형에 대해 당신은 여기가 사용할 수 있습니다 try-except

def myfunc(x): 
    try: 
     x = x.splitlines(True) 
    except AttributeError: 
     pass #likely a file 

    for line in x: 
     doSomething(line) 

또 다른 버전 (약간 덜 허용 - 주석 참조)

try: 
    from cStringIO import StringIO 
except ImportError: #python 3 
    from io import StringIO 

def myfunc(x): 
    try: 
     x.seek(0) 
    except AttributeError: #likely a string 
     x = StringIO(x) 

    for line in x: 
     doSomething(line) 

Answer 2

문자열에서도 작동하지만 문자 단위로 문자열을 반복합니다. 사용자는 문자열 대신 목록을 사용하여 함수를 호출해야합니다.

명시 적으로 다음 라인으로 해석되도록, 하나의 문자열을 전달 지원하려는 경우 해당 테스트해야합니다 :

def myfunc(x): 
    if isinstance(x, basestring): 
     # Interpret argument as one line 
     x = [x] 
    for line in x: 
     doSomething(line) 

을 다른 방법으로, 줄 바꿈에 문자열을 분할 수 대신 :

def myfunc(x): 
    if hasattr(x, 'splitlines'): 
     # Interpret argument as multiple lines: 
     x = x.splitlines() 
    for line in x: 
     doSomething(line) 

Answer 3

사용 오리 입력 :

def myfunc(x): 
    try: 
     lines = x.split('\n') 
    except AttributeError: 
     lines = x 
    for line in lines: 
     doSomething(line) 

당신은 또한 유형을 확인하실 수 있습니다 당신이 원하는 경우 :

def myfunc(x): 
    if isinstance(x, basestring): # will work in Python >=2.3, <3.x 
     lines = x.split('\n') 
    else: 
     lines = x 
    for line in lines: 
     doSomething(line) 

Answer 4

방금 ​​파일 - 류의 객체를 받아 들인다면, 당신은 StringIO에 문자열 인수를 포장 발신자를 지시 할 수 있습니다 :

는

myfunc(StringIO(s)) 

Answer 5

isinstance (var, str) var가 str 유형인지 확인합니다.

파일 형식을 직접 기억하지 못했지만 파일 var을 사용하여 찾을 수 있습니다. 그것은

당신은 사용할 수 있습니다 ...> <에 표시됩니다 :

def myfunc(x): 
    for line in x: 
     if isinstance(line, str): 
      print line 
     else: 
      # Its a file so do file things 

Answer 6

한 가지 방법은 x가 문자열 인 경우 결정하기 위해 isinstance 같은 것을 사용하는 것입니다. 그럴 경우 StringIO (또는 cStringIO)에 전달하여 객체와 같은 파일로 만들고 원하는대로 작업을 진행할 수 있습니다. 나는 이것을 테스트하지는 않았지만 대략적으로 보일 것이다 :

def myfunc(x): 
    if isinstance(x, str): 
     x = StringIO.StringIO(x) 
    for line in x: 
     doSomething(line)