#! /usr/bin/env python
# -*- coding: utf-8 -*-
import sys
import argparse
import requests
def get_page(url):
try:
res = requests.get(url, timeout=2)
return res.status_code
except requests.exceptions.ReadTimeout:
return None
def main(argv=sys.argv[1:]):
parser = argparse.ArgumentParser()
parser.add_argument('--urlfile', type=argparse.FileType('rt'), default=sys.stdin)
parser.add_argument('urls', nargs='*')
args = parser.parse_args(argv)
for url in args.urls:
code = get_page(url)
print(code, url)
for line in args.urlfile:
url = line.strip()
code = get_page(url)
print(code, url)
if __name__ == '__main__':
main()
Created
October 21, 2016 14:24
-
-
Save TakesxiSximada/ab5f72f65907f4a431dd8e1594810e64 to your computer and use it in GitHub Desktop.
標準入力で入力されたURLをrequestsでGETする
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #! /usr/bin/env python | |
| # -*- coding: utf-8 -*- | |
| import sys | |
| import argparse | |
| import requests | |
| def get_page(url): | |
| try: | |
| res = requests.get(url, timeout=2) | |
| return res.status_code | |
| except requests.exceptions.ReadTimeout: | |
| return None | |
| def main(argv=sys.argv[1:]): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument('--urlfile', type=argparse.FileType('rt'), default=sys.stdin) | |
| parser.add_argument('urls', nargs='*') | |
| args = parser.parse_args(argv) | |
| for url in args.urls: | |
| code = get_page(url) | |
| print(code, url) | |
| for line in args.urlfile: | |
| url = line.strip() | |
| code = get_page(url) | |
| print(code, url) | |
| if __name__ == '__main__': | |
| main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment