diff Plugins/DicomWeb/DicomWeb.py @ 33:eb6d219af210

test stow
author Sebastien Jodogne <s.jodogne@gmail.com>
date Sun, 02 Aug 2015 14:01:32 +0200
parents
children 97acfdf0dbce
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/Plugins/DicomWeb/DicomWeb.py	Sun Aug 02 14:01:32 2015 +0200
@@ -0,0 +1,77 @@
+#!/usr/bin/python
+
+# Orthanc - A Lightweight, RESTful DICOM Store
+# Copyright (C) 2012-2015 Sebastien Jodogne, Medical Physics
+# Department, University Hospital of Liege, Belgium
+#
+# This program is free software: you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+
+import os
+import sys
+import email
+import urllib2
+
+from email.mime.multipart import MIMEMultipart
+from email.mime.application import MIMEApplication
+
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', 'Tests'))
+from Toolbox import *
+
+
+def _AttachPart(related, path, contentType):
+    with open(path, 'rb') as f:
+        part = MIMEApplication(f.read(), contentType, email.encoders.encode_noop)
+        related.attach(part)
+
+
+def SendStow(orthanc, uri, dicom):
+    related = MIMEMultipart('related')
+    related.set_boundary('boundary_0123456789_boundary')
+
+    if isinstance(dicom, list):
+        for i in range(dicom):
+            _AttachPart(related, dicom[i], 'dicom')
+    else:
+        _AttachPart(related, dicom, 'dicom')
+
+    headers = dict(related.items())
+    body = related.as_string()
+
+    # Discard the header
+    body = body.split('\n\n', 1)[1]
+
+    headers['Content-Type'] = 'multipart/related; type=application/dicom; boundary=%s' % related.get_boundary()
+    headers['Accept'] = 'application/json'
+
+    return DoPost(orthanc, uri, body, headers = headers)
+
+
+def GetMultipart(uri, headers = {}):
+    tmp = urllib2.urlopen(uri)
+    info = str(tmp.info())
+    answer = tmp.read()
+
+    s = info + "\n" + answer
+
+    msg = email.message_from_string(s)
+
+    result = []
+
+    for i, part in enumerate(msg.walk(), 1):
+        payload = part.get_payload(decode = True)
+        if payload != None:
+            result.append(payload)
+
+    return result