[python] 파일 타입 확인하기
파일 확장자로 확인하기
일반적으로 파일의 확장자를 확인하여 파일 타입을 알 수 있습니다. 예를 들어, 파일 이름이 “example.pdf”인 경우 “.pdf” 부분을 기반으로 파일이 PDF 파일임을 알 수 있습니다.
def get_file_type(filename):
file_extension = filename.split(".")[-1]
if file_extension == "pdf":
return "PDF 파일"
elif file_extension == "docx":
return "워드 파일"
else:
return "알 수 없는 파일"
파일 내용으로 확인하기
때로는 파일 내용을 확인하여 파일 타입을 결정해야 할 수도 있습니다. 예를 들어, 파일의 마법 숫자(magic number) 시그니처를 읽어서 파일 타입을 확인할 수 있습니다.
def get_file_type(file_path):
with open(file_path, 'rb') as f:
header = f.read(4) # 파일의 처음 4바이트 읽기
if header == b'%PDF':
return "PDF 파일"
elif header == b'\x50\x4B\x03\x04':
return "압축 파일"
else:
return "알 수 없는 파일"
위의 예제 코드는 파일의 내용을 읽어서 PDF 파일과 압축 파일을 확인하는 방법을 보여줍니다.
파이썬을 사용하여 파일 타입을 확인하는 방법에 대해 알아보았습니다. 파일 타입을 확인함으로써, 올바른 처리 로직을 수행할 수 있게 됩니다.