sobjects/parsers.py
changeset 6963 5774d4ba4306
parent 6960 822f2530570d
child 6970 a6ccbfbacf3d
--- a/sobjects/parsers.py	Wed Feb 09 18:06:27 2011 +0100
+++ b/sobjects/parsers.py	Wed Feb 09 18:06:29 2011 +0100
@@ -19,6 +19,7 @@
 
 import urllib2
 import StringIO
+import os.path as osp
 from cookielib import CookieJar
 from datetime import datetime, timedelta
 
@@ -193,6 +194,10 @@
         if not url.startswith('http'):
             stream = StringIO.StringIO(url)
         else:
+            for mappedurl in HOST_MAPPING:
+                if url.startswith(mappedurl):
+                    url = url.replace(mappedurl, HOST_MAPPING[mappedurl], 1)
+                    break
             self.source.info('GET %s', url)
             stream = _OPENER.open(url)
         return _entity_etree(etree.parse(stream).getroot())
@@ -334,3 +339,13 @@
         else:
             rql = 'SET %s, Y eid IN (%s), NOT X %s Y' % (rql, eidstr, rtype)
         self._cw.execute(rql, {'x': entity.eid})
+
+def registration_callback(vreg):
+    global HOST_MAPPING
+    vreg.register_all(globals().values(), __name__)
+    host_mapping_file = osp.join(vreg.config.apphome, 'hostmapping.py')
+    if osp.exists(host_mapping_file):
+        HOST_MAPPING = eval(file(host_mapping_file).read())
+        vreg.info('using host mapping %s from %s', HOST_MAPPING, host_mapping_file)
+    else:
+        HOST_MAPPING = {}