Add handle translation hack

svn path=/myrpki.rototill/arin-to-csv.py; revision=3129

1 files changed, 10 insertions, 4 deletions

diff --git a/myrpki.rototill/arin-to-csv.py b/myrpki.rototill/arin-to-csv.py
index fc98bb64..5b4caae1 100644
--- a/myrpki.rototill/arin-to-csv.py
+++ b/myrpki.rototill/arin-to-csv.py
@@ -53,7 +53,7 @@ class ASHandle(Handle):
 
   def finish(self, ctx):
     if self.check():
-      ctx.asns.writerow((self.OrgID, self.ASNumber))
+      ctx.asns.writerow((ctx.translations.get(self.OrgID, self.OrgID), self.ASNumber))
 
 class NetHandle(Handle):
 
@@ -63,7 +63,7 @@ class NetHandle(Handle):
 
   def finish(self, ctx):
     if self.NetType in ("allocation", "assignment") and self.check():
-      ctx.prefixes.writerow((self.OrgID, self.NetRange))
+      ctx.prefixes.writerow((ctx.translations.get(self.OrgID, self.OrgID), self.NetRange))
 
   def __repr__(self):
     return "<%s %s.%s %s %s>" % (self.__class__.__name__,
@@ -76,8 +76,8 @@ class V6NetHandle(NetHandle):
 
   def __repr__(self):
     return "<%s %s.%s %s %s>" % (self.__class__.__name__,
-                                 self.OrgID, self.V6NetHandle,
-                                 self.NetType, self.NetRange)
+                                 ctx.translations.get(self.OrgID, self.OrgID),
+                                 self.V6NetHandle, self.NetType, self.NetRange)
 
 class main(object):
 
@@ -86,6 +86,8 @@ class main(object):
     "NetHandle"   : NetHandle,
     "V6NetHandle" : V6NetHandle }
 
+  translations = {}
+
   @staticmethod
   def parseline(line):
     tag, sep, val = line.partition(":")
@@ -99,6 +101,10 @@ class main(object):
   def __init__(self):
     self.asns = self.csvout("asns.csv")
     self.prefixes = self.csvout("prefixes.csv")
+    try:
+      self.translations = dict((src, dst) for src, dst in myrpki.csv_open("translations.csv"))
+    except IOError:
+      pass
     f = gzip.open("arin_db.txt.gz")
     cur = None
     for line in f: