[tor-commits] [metrics-tasks/master] Add --download option (#6329).

[karsten at torproject.org]

commit 00693abce05d177a388aa7adef6a2d371579e008
Author: Karsten Loesing <karsten.loesing at gmx.net>
Date:   Fri Jul 20 15:45:38 2012 +0200

    Add --download option (#6329).
---
 task-6329/tor-relays-stats.py |   16 +++++++++++++++-
 1 files changed, 15 insertions(+), 1 deletions(-)

diff --git a/task-6329/tor-relays-stats.py b/task-6329/tor-relays-stats.py
index 4fedf92..fc9b14c 100755
--- a/task-6329/tor-relays-stats.py
+++ b/task-6329/tor-relays-stats.py
@@ -11,6 +11,7 @@ import operator
 import sys
 import os.path
 from optparse import OptionParser, OptionGroup
+import urllib
 
 class RelayStats(object):
     def __init__(self):
@@ -129,8 +130,17 @@ class RelayStats(object):
                   selection_weights[2] * 100.0, selection_weights[3] * 100.0,
                   selection_weights[4] * 100.0)
 
+def download_details_file():
+    url = urllib.urlopen('https://onionoo.torproject.org/details?type=relay&running=true')
+    details_file = open("details.json", 'w')
+    details_file.write(url.read())
+    url.close()
+    details_file.close()
+
 if '__main__' == __name__:
     parser = OptionParser()
+    parser.add_option("-d", "--download", action="store_true",
+                      help="download details.json from Onionoo service")
     group = OptionGroup(parser, "Filtering options")
     group.add_option("-a", "--as", dest="ases", action="append",
                      help="select only relays from autonomous system number AS",
@@ -155,8 +165,12 @@ if '__main__' == __name__:
     (options, args) = parser.parse_args()
     if len(args) > 0:
         parser.error("Did not understand positional argument(s), use options instead.")
+
+    if options.download:
+        download_details_file()
+
     if not os.path.exists('details.json'):
-        parser.error("Did not find details.json.  Please download this file using the following command:\ncurl -o details.json 'https://onionoo.torproject.org/details?type=relay&running=true'")
+        parser.error("Did not find details.json.  Re-run with --download.")
 
     stats = RelayStats()
     relays = stats.get_relays(countries=options.country,