Source tree reorg, phase 1. Almost everything moved, no file contents changed.

svn path=/branches/tk685/; revision=5757

1 files changed, 72 insertions, 0 deletions

diff --git a/potpourri/csvgrep.py b/potpourri/csvgrep.py
new file mode 100644
index 00000000..68bdd259
--- /dev/null
+++ b/potpourri/csvgrep.py
@@ -0,0 +1,72 @@
+# $Id$
+# 
+# Copyright (C) 2010-2012  Internet Systems Consortium ("ISC")
+# 
+# Permission to use, copy, modify, and distribute this software for any
+# purpose with or without fee is hereby granted, provided that the above
+# copyright notice and this permission notice appear in all copies.
+# 
+# THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
+# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+# AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
+# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
+# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+# PERFORMANCE OF THIS SOFTWARE.
+
+"""
+Utility to simplify finding handles in one of the pseudo-RIR databases.
+
+Usage: python csvgrep.py datum [datum ...]
+
+where each datum is an ASN, IP address, or IP prefix.
+
+ASNs are recognized by being pure integers; IP addreses are recognized
+by having dots (IPv4) or colons (IPv6).
+
+After eating all of the command line arguments, we search asns.csv for
+any ASNs given, and prefixes.csv for any prefixes given.
+"""
+
+import sys
+from rpki.resource_set import resource_set_as, resource_set_ipv4, resource_set_ipv6
+from rpki.csv_utils import csv_reader
+
+asn  = resource_set_as()
+ipv4 = resource_set_ipv4()
+ipv6 = resource_set_ipv6()
+
+for datum in sys.argv[1:]:
+  if datum.replace("-", "").isdigit():
+    t = asn
+  else:
+    t = ipv6 if ":" in datum else ipv4
+    if "-" not in datum and "/" not in datum:
+      datum = datum + "-" + datum
+  try:
+    t.append(t.parse_str(datum))
+  except:
+    print "Error attempting to parse", datum
+    raise
+
+#print "Looking for: ASNs %s IPv4 %s IPv6 %s" % (asn, ipv4, ipv6)
+
+def matches(set1, datum):
+  set2 = set1.__class__(datum)
+  if set1.intersection(set2):
+    return set2
+  else:
+    return False
+
+if asn:
+  for h, a in csv_reader("asns.csv", columns = 2):
+    m = matches(asn, a)
+    if m:
+      print h, m
+
+if ipv4 or ipv6:
+  for h, a in csv_reader("prefixes.csv", columns = 2):
+    t = ipv6 if ":" in a else ipv4
+    m = t and matches(t, a)
+    if m:
+      print h, m