파이썬 프로그래밍 기초(3)

 

 

파이썬 프로그래밍 기초(3)

 

 

탭을 4개의 공백으로 바꾸기

 

이번에는 문서 파일을 읽어서 그 문서 파일 안에 있는 탭 문자(Tab)를 공백 문자(Space) 4개로 바꾸어 주는

스크립트를 작성해 보자.

 

필요한 기능은 문서 파일 읽어 들이기, 문자열 변경하기, 입력은 탭을 포함한 문서 파일

출력은 탭이 공백으로 수정된 문서 파일

 

다음과 같은 형식으로 프로그램이 수행되도록 만들 것이다. 'python tabto4.py src dst'

tabto4.py는 우리가 작성해야 할 파이썬 프로그램 이름, src는 탭을 포함하고 있는 원본 파일 이름

dst는 파일 안의 탭을 공백 4개로 변환한 결과를 저장할 파일 이름이다.

 

예를 들어 a.txt에 있는 탭을 4개의 공백으로 바꾸어 b.txt에 저장하고 싶다면 다음과 같이 수행해야 한다.

'python tabto4.py a.txt b.txt'

 

 

1. 다음과 같이 tabto4.py 파일을 작성해 보자. 파일은 tabto4.py는 ipmes1 디렉터리에 저장하였다.

## tabto4.py 코드 작성
import sys
src = sys.argv[1]
dst = sys.argv[2]
print(src)
print(dst)

 

 

 

2. 다음과 같이 수행했을 때 입력값이 정상적으로 출력되는지 확인해 보자.

입력으로 전달한 a.txt와 b.txt가 정상적으로 출력되는 것을 확인할 수 있다.

## tabto4.py 코드 수행
C:\ipmes1> python tabto4.py a.txt b.txt
→ a.txt
→ b.txt

 

 

 

3. 테스트를 위한 원본 파일인 a.txt를 다음과 같이 작성한다. 각 단어는 탭(\t) 문자로 분리되도록 입력해야 한다.

## a.txt 파일 내용
C:\ipmes1> python tabto4.py a.txt b.txt
→ Life 	is 	too 	short
→ you 	need	python

 

 

 

4. 이제 탭 문자를 포함한 a.txt를 읽어 탭을 공백 4개로 변환할 수 있도록 코드를 수정해 보자.

## tabto4.py 코드 수정
import sys

src = sys.argv[1]
dst = sys.argv[2]

f = open(src)
tab_content = f.read()
f.close()

space_content = tab_content.replace("\t", " "*4)
print(space_content)

 

위 코드는 src에 해당되는 입력 파일을 읽어서 그 내용을 tab_content라는 변수에 저장한 후

문자열의 replace 함수함수를 사용해 탭 (\t)을 4개의 공백으로 변경하도록 코드를 수정했다.

 

 

5. 다음과 같은 명령을 수행해 보자.

 

탭 문자가 공백 4개로 변경되어 출력될 것이다.

하지만 탭과 공백의 차이점을 눈으로 알 수는 없으므로 탭이 정상적으로 공백으로 변경되었는지 확인하기 어렵다.

## a.txt 실행
python tabto4.py a.txt b.txt
→ Life     is     too     short
→ you     need        python

 

 

 

6. 이제 변경된 내용을 b.txt에 저장할 수 있도록 다음과 같이 프로그램을 변경해 보자.

 

탭이 공백으로 변경된 space_content를 출력 파일인 dst에 쓰도록 코드를 수정했다.

## a.txt의 변경된 내용 b.txt에 저장
import sys

src = sys.argv[1]
dst = sys.argv[2]

f = open(src)
tab_content = f.read()
f.close()

space_content = tab_content.replace("\t", " "*4)
print(space_content)

f = open(dst, 'w')
f.write(space_content)
f.close()

 

 

 

7. 다음 명령을 수행해 프로그램을 실행하자.

 

아래 명령을 실행하면 b.txt파일이 C:\ipmes1 디렉터리에 생성된다.

에디터로 b.txt를 열어 탭이 4개의 공백 문자로 변경되었는지 확인해 보자.

프로그램을 작성할 때 사용하는 에디터는 대부분 탭과 공백 문자를 다르게 표시하므로 눈으로 확인할 수 있다.

## 코드 명령 실행
C:\ipmes1> python tabto4.py a.txt b.txt
→ Life     is     too     short
→ you     need        python

 

 

하위 디렉터리 검색하기

 

특정 디렉터리부터 시작해서 그 하위의 모든 파일 중 파이썬 파일(*.py)만 출력해 주는 프로그램을 만들고 싶다.

필요한 기능은 파이썬 파일만 찾아서 출력하기, 입력받는 값은 검색을 시작할 디렉터리, 출력하는 값은 파이썬 파일명 이다.

 

1. 다음과 같이 sub_dir_search.py 파일을 작성해 보자.

search 함수를 만들고 시작 디렉터리를 입력받도록 코드를 작성했다.

sub_dir_search.py 파일은 ipmes1 파일에 저장하였다.

## sub_dir_search.py 파일 코드
def search(dirname):
    print(dirname)
    
search("C:/")

 

 

2. 이제 이 디렉터리에 있는 파일을 검색할 수 있도록 소스를 변경해 보자.

 

os.listdir를 사용하면 해당 디렉터리에 있는 파일의 리스트를 구할 수 있다.

여기에서 구하는 파일 리스트는 파일 이름만 포함되어 있으므로 경로를 포함한 파일 이름을 구하기 위해서는 입력으로 받은

dirname을 앞에 덧붙여 주어야 한다.

 

os 모듈에는 디렉터리와 파일 이름을 이어 주는 os.path.join 함수가 있으므로 이 함수를 사용하면 디렉터리를 포함한

전체 경로를 쉽게 구할 수 있다.

## sub_dir_search.py 코드 변경
import os

def search(dirname):
    filenames = os.listdir(dirname)
    for filename in filenames:
        full_filename = os.path.join(dirname, filename)
        print(full_filename)
    
search("C:/")

 

 

 

코드를 수행하면 C:/ 디렉터리에 있는 파일이 다음과 같이 출력될 것이다.

## sub_dir_search.py 실행 결과 값
→ C:/$Recycle.Bin
→ C:/$SysReset
→ C:/$Windows.~WS
→ C:/Config.Msi
→ C:/Documents and Settings
.
.
.

 

 

3. 이제 C:/ 디렉터리에 있는 파일들 중 확장자가 .py인 파일만을 출력하도록 코드를 변경해 보자.

 

파일 이름에서 확장자만 추출하기 위해 os 모듈의 os.path.splitext 함수를 사용하였다.

os.path.splitext는 파일 이름을 확장자를 기준으로 두 부분으로 나누어 준다.

 

따라서 os.path.splitext(full_filename)[-1]은 해당 파일의 확장자 이름이 된다.

앞의 코드는 확장자 이름이 .py인 경우만 출력하도록 작성했다.

C:/디렉터리에 파이썬 파일이 없어 아무것도 출력되지 않는다.

 

※ os.path.splitext(filename)은 파일 이름을 확장자와 나머지 부분으로 나누는 함수이다.

이 함수는 튜플(파일이름, 확장자)를 반환한다. 이때 [-1]은 리스트나 튜플에서 마지막 요소를 의미하는 인덱스이다.

따라서 마지막 요소인 확장자를 가져온다는 의미이다.

## C:/ 디렉터리에서 '.py'확장자 파일 찾기
import os

def search(dirname):
    filenames = os.listdir(dirname)
    for filename in filenames:
        full_filename = os.path.join(dirname, filename)
        ext = os.path.splitext(full_filename)[-1]
        if ext == '.py':
            print(full_filename)
            
search("C:/")

 

 

4. 이제는 C:/ 디렉터리 바로 밑에 있는 파일뿐만 아니라 그 하위 디렉터리를 포함한 모든 파이썬 파일을 검색해보자.

하위 디렉터리도 검색이 가능하도록 다음과 같이 코드를 변경해야 한다.

 

try-except 문으로 함수 전체를 감싼 이유는 os.listdir를 수행할 때 권한이 없는 디렉터리에 접근하더라도

프로그램이 오류로 종료되지 않고 그냥 수행되도록 하기 위해서이다.

full_filename이 디렉터리인지 파일인지 구별하기 위해 os.path.isdir 함수를 사용했고

디렉터리일 경우 해당 경로를 입력받아 다시 search 함수를 호출했다.

 

이렇게 해당 디렉터리의 파일이 디렉터리일 경우 다시 search 함수를 호출해 나가면(재귀 호출) 해당 디렉터리의 하위 파일을

다시 검색하기 시작하므로 결국 모든 파일을 검색할 수 있다.

 

여기서 재귀호출이란 자기 자신을 다시 호출하는 프로그래밍 기법이다.

이 코드에서 보면 search 함수에서 다시 자기 자신인 search 함수를 호출한다. 이것이 바로 재귀 호출이다.

## C:/ 하위 디렉터리에서 '.py'확장자 파일 찾기
import os

def search(dirname):
    try:
        filenames = os.listdir(dirname)
        for filename in filenames:
            full_filename = os.path.join(dirname, filename)
            if os.path.isdir(full_filename):
                search(full_filename)
            else:
                ext = os.path.splitext(full_filename)[-1]
                if ext == '.py':
                    print(full_filename)
    except PermissionError:
        pass
    
search("C:/")

 

 

위의 코드를 실행하면 다음과 같은 결괏값이 나온다.

결괏값이 너무 많아 생략하여 작성하였다.

## C:/ 하위 디렉터리에서 '.py'확장자 파일 결과
→ C:/$Recycle.Bin\S-1-5-21-3560836347-1011063389-2886450844-1001\$I0M8NS5.py
→ C:/$Recycle.Bin\S-1-5-21-3560836347-1011063389-2886450844-1001\$I1SK67G.py
→ C:/$Recycle.Bin\S-1-5-21-3560836347-1011063389-2886450844-1001\$I2AKICV.py
→ C:/$Recycle.Bin\S-1-5-21-3560836347-1011063389-2886450844-1001\$I3944GD.py
→ C:/$Recycle.Bin\S-1-5-21-3560836347-1011063389-2886450844-1001\$I3Q3KFT.py
.
.
.

 

 

○ 하위 디렉터리 검색을 쉽게 해주는 os.walk

 

os.walk를 사용하면 앞에서 작성한 코드를 더 간단하게 만들 수 있다.

os.walk는 시작 디렉터리부터 시작해 하위에 있는 모든 디렉터리를 차례대로 방문하는 함수이다.

디렉터리와 파일을 검색하는 일반적인 경우라면 os.walk를 사용하기를 추천한다.

 

이 결괏값 역시 너무 많아 생략하였다.

## 하위 디렉터리 검색을 쉽게 해주는 os.walk
import os

for (path, dir, files) in os.walk("c:/"):
    for filename in files:
        ext = os.path.splitext(filename)[-1]
        if ext == '.py':
            print(os.path.join(path, filename))
→ c:/Users\ipmes\anaconda3\Lib\site-packages\plotly\io/_sg_scraper.py
→ c:/Users\ipmes\anaconda3\Lib\site-packages\plotly\io/_templates.py
→ c:/Users\ipmes\anaconda3\Lib\site-packages\plotly\io/_utils.py
→ c:/Users\ipmes\anaconda3\Lib\site-packages\plotly\io/__init__.py

 

 

 

 

 

TEL (062-226-1777, 010-9891-7244), E-mail (ipmes@ipmes.co.kr)

임베디드 시스템 | PCB 설계 제작 | 펌웨어 개발 | 신호처리 | 응용프로그램