Integrated

Author: remram44

reprounzip-vis/reprounzip_vis/__init__.py

@@ -2,13 +2,16 @@
 import BaseHTTPServer
 import SocketServer
 import mimetypes
+import os
+import pkg_resources
+from rpaths import Path
 import shutil
+import tempfile
 import webbrowser
 
-import pkg_resources
-
+from reprounzip.common import RPZPack
 from reprounzip.unpackers.common import COMPAT_OK
-
+from reprounzip.unpackers.graph import generate
 
 __version__ = '0.1'
 
@@ -21,7 +24,9 @@ def do_GET(self):
         if self.path == '/provenance.json':
             self.send_response(200)
             self.send_header('Content-Type', 'application/json')
-            self.wfile.write(self.provenance_json)
+            self.end_headers()
+            with open(self.provenance_json, 'rb') as f:
+                shutil.copyfileobj(f, self.wfile)
         else:
             try:
                 f = pkg_resources.resource_stream('reprounzip_vis',
@@ -35,30 +40,41 @@ def do_GET(self):
                 else:
                     ctype = mimetypes.guess_type(self.path)[0]
                 self.send_header('Content-Type', ctype)
+                self.end_headers()
                 shutil.copyfileobj(f, self.wfile)
                 f.close()
 
 
 def show_vis(args):
     # Extract JSON from package
-    VisHTTPHandler.provenance_json = '{}'  # TODO
-
-    # Serve static files and JSON document to browser
-    port = 8003
-
-    httpd = SocketServer.TCPServer(('', port), VisHTTPHandler)
-    print("serving at port %d" % port)
-
-    # Open web browser
-    webbrowser.open('http://localhost:%d/index.html' % port)
-
-    # Serve until killed
+    fd, json_file = tempfile.mkstemp(prefix='reprounzip_vis_', suffix='.json')
     try:
-        httpd.serve_forever()
-    except KeyboardInterrupt:
-        pass
+        rpz_pack = RPZPack(args.pack)
+        with rpz_pack.with_config() as config:
+            with rpz_pack.with_trace() as trace:
+                generate(Path(json_file), config, trace, graph_format='json')
+        os.close(fd)
+
+        VisHTTPHandler.provenance_json = json_file
+
+        # Serve static files and JSON document to browser
+        port = 8002
+
+        httpd = SocketServer.TCPServer(('', port), VisHTTPHandler)
+        print("serving at port %d" % port)
+
+        # Open web browser
+        webbrowser.open('http://localhost:%d/index.html' % port)
+
+        # Serve until killed
+        try:
+            httpd.serve_forever()
+        except KeyboardInterrupt:
+            pass
+        finally:
+            httpd.server_close()
     finally:
-        httpd.server_close()
+        os.remove(json_file)
 
 
 def setup_vis(parser, **kwargs):