LPTHW 笨方法学习python 16章

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

根据16章的内容作了一些扩展。

比如，判断文件如果存在，就在文件后追加，如不存在则创建。

同时借鉴了shell命令中类似 cat <<EOF　＞　test的方法，提示用户输入一个结尾符。

现在有一个小坑，怎么使用python去读取一个文件的行数，原来有os.system("wc -l filename")倒是可以，但是windows下如何操作呢？回头补填。

#!/usr/bin/env python

# -*- coding:utf-8 -*-

from sys import argv

from os import path

script, filename = argv

if path.exists(filename):

    print "We're going to erase %r." % filename

    t_module = "a"

else:

    print "We will create %r." % filename

    t_module = "w"

end_EOF = raw_input("please give me a EOF string")

with open(filename,t_module) as fo:

    line_count = 1

    while True:

        line_content = raw_input("please input %d line string:\n" % line_count)

        if line_content == end_EOF:

            break

        else:

            fo.write(line_content)

            fo.write("\n")