Python : 가끔 압축 된 파일에서 데이터를 읽지 못함 가끔

Question

일반적으로 gzip으로 압축 된 일부 파일에서 이진 데이터를 읽어야합니다. 나는 GZIP 모듈을 사용하여 데이터를 읽고 처리했다 :

def decode(self, filename): 
    with gzip.open(filename, 'rb') as f: 
     # ReadData 

그러나 때로는 파일은이 경우 내가 얻을 압축되지 IOError가 (파일이 GZIP 헤더를 가지고 있지 않기 때문에).

내가 좋아하는 뭔가를 할 수 :

try: 
    f = gzip.open(filename, 'rb') 
    f._read_gzip_header() 
    f.rewind() 
except IOError: 
    f.close() 
    f = open(filename, 'rb') 

with f as gz: 
    #ReadData 

을하지만 난 그것을 고칠 수있는 좋은 방법입니다 생각하지 않습니다.

이 문제를 해결하는 우아한 해결책을 찾고 있습니다. 필자는 여러 파일 유형에 대해 여러 "디코드"함수를 작성합니다. 내가 고려하는 솔루션은 GzipFile의 하위 클래스를 작성하여 처리하는 것이지만 더 나은 방법이있을 수 있다고 생각합니다.

저는 파이썬 2.7을 사용하고 있습니다.

미리 제안 해 주셔서 감사합니다!

Answer 1

처음 두 바이트 \x1f 및 \x8b을 경우 RFC 1952에 따라 당신은 확인할 수 있습니다 :이 고정이

회원 헤더와 트레일러

ID1 (식별 1) ID2 (식별 2) 파일을 gzip 형식으로 식별하려면 ID1 = 31 (0x1f, \ 037), ID2 = 139 (0x8b, \ 213) 값을 사용하십시오.

그래서 예를 들어

,

with open('test.gz','rb') as f: 
    print(f.read(2)) 

b'\x1f\x8b' #well, that was gzipped 

with open('test','rb') as f: 
    print(f.read(2)) 

b'he' #must not be gzipped 

은 아마도 당신은 그 f.seek(0), 두 바이트에 따라 약간의 제어 흐름을 할 것이라고하고 그에 따라 진행합니다.

솔직히? 귀하의 솔루션은 괜찮습니다 (불필요한 닫기/다시 열 부분을 모듈로). try/except은 pythonic입니다.