python - 读取后从txt文件中删除行

Question

我正在尝试创建一个脚本，该脚本向 txt 文件中的随机网址发出请求

import urllib2

with open('urls.txt') as urls:
    for url in urls:
        try:
            r = urllib2.urlopen(url)
        except urllib2.URLError as e:
            r = e
        if r.code in (200, 401):
            print '[{}]: '.format(url), "Up!"
        elif r.code == 404:
            print '[{}]: '.format(url), "Not Found!" 

但我希望当某些 url 出现 404 not found 时从文件中删除。每个 url 都是每行，所以基本上是删除每个 404 未找到的 url。怎么办？!

Answer 1

您可以写入第二个文件:

import urllib2

with open('urls.txt', 'r') as urls, open('urls2.txt', 'w') as urls2:
    for url in urls:
        try:
            r = urllib2.urlopen(url)
        except urllib2.URLError as e:
            r = e

        if r.code in (200, 401):
            print '[{}]: '.format(url), "Up!"
            urls2.write(url + '\n')
        elif r.code == 404:
            print '[{}]: '.format(url), "Not Found!"