[02] 간단한 메모장

파일에 원하는 메모를 저장하고 수정및 확인을 할 수 있는 간단한 메모장을 만들어 보도록 하자.

# memo.py

import sys

import time

 

def usage():

    print """

Usage

=====

python %s -v : View memo

python %s -a : Add memo

""" % (sys.argv[0], sys.argv[0]) #sys.argv[0]는 프로그램명입니다, sys.argv[1:] 은 입력파라미터를 뜻합니다.

#sys.argv는 프로그램에 인수로 전달된 값들을 리스트형으로 반환해 준다.

if not sys.argv[1:] or sys.argv[1] not in ['-v', '-a']: #파라미터가 없거나 또는 첫번째 파라미터가 -v나 -a가 아닌경우에

    usage()

elif sys.argv[1] == '-v':

    try: print open("memo.txt").read()

    except IOError: print "memo does not exist!"

elif sys.argv[1] == '-a':

    word = raw_input("Enter memo: ")

    f = open("memo.txt", 'a')

    f.write(time.ctime() + ': ' + word+'\n')

    f.close()

    print "Added"

사용자가 명령행에서 ‘python memo.py -v’라고 입력하면 지금까지의 메모를 출력해주고 만약 하나도 추가된 것이 없을 때는 "memo does not exist!"라는 문장을 출력해 준다.

 

‘python memo.py -a’라고 입력하면 메모를 입력받아서 memo.txt라는 파일에 입력된 내용과 현재의 시간을 함께 파일에 적는다.

 

사용자가 명령행에서 ‘python memo.py -v’나 ‘python memo.py -a’라고 입력하지 않을 경우에는 usage() 함수를 호출한다. sys.argv를 어떻게 활용하는지 그리고 try.. except 구문을 어떻게 활용했는지를 주목해서 보도록 하자.

출처 : wikidocs 점프 투 파이썬