Mercurial > hg > orthanc-tests
annotate Plugins/DicomWeb/DicomWeb.py @ 511:933fe1bbce4f
upgrade to year 2023
| author | Sebastien Jodogne <s.jodogne@gmail.com> |
|---|---|
| date | Sat, 25 Mar 2023 12:30:14 +0100 |
| parents | 06eea14c5906 |
| children | 9f8276ac1cdd |
| rev | line source |
|---|---|
| 33 | 1 #!/usr/bin/python |
| 2 | |
| 3 # Orthanc - A Lightweight, RESTful DICOM Store | |
| 73 | 4 # Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
| 33 | 5 # Department, University Hospital of Liege, Belgium |
|
511
933fe1bbce4f
upgrade to year 2023
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
449
diff
changeset
|
6 # Copyright (C) 2017-2023 Osimis S.A., Belgium |
|
933fe1bbce4f
upgrade to year 2023
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
449
diff
changeset
|
7 # Copyright (C) 2021-2023 Sebastien Jodogne, ICTEAM UCLouvain, Belgium |
| 33 | 8 # |
| 9 # This program is free software: you can redistribute it and/or | |
| 10 # modify it under the terms of the GNU General Public License as | |
| 11 # published by the Free Software Foundation, either version 3 of the | |
| 12 # License, or (at your option) any later version. | |
| 13 # | |
| 14 # This program is distributed in the hope that it will be useful, but | |
| 15 # WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 17 # General Public License for more details. | |
| 18 # | |
| 19 # You should have received a copy of the GNU General Public License | |
| 20 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 21 | |
| 22 | |
| 23 import os | |
| 24 import sys | |
| 25 import email | |
| 83 | 26 import uuid |
| 33 | 27 |
| 28 from email.mime.multipart import MIMEMultipart | |
| 29 from email.mime.application import MIMEApplication | |
| 30 | |
| 31 sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', 'Tests')) | |
| 32 from Toolbox import * | |
| 33 | |
| 34 | |
| 83 | 35 def _AttachPart(body, path, contentType, boundary): |
| 33 | 36 with open(path, 'rb') as f: |
| 83 | 37 body += bytearray('--%s\r\n' % boundary, 'ascii') |
| 38 body += bytearray('Content-Type: %s\r\n\r\n' % contentType, 'ascii') | |
| 39 body += f.read() | |
| 40 body += bytearray('\r\n', 'ascii') | |
| 33 | 41 |
| 42 | |
|
398
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
43 def SendStowRaw(orthanc, uri, dicom): |
| 83 | 44 # We do not use Python's "email" package, as it uses LF (\n) for line |
| 45 # endings instead of CRLF (\r\n) for binary messages, as required by | |
| 46 # RFC 1341 | |
| 47 # http://stackoverflow.com/questions/3086860/how-do-i-generate-a-multipart-mime-message-with-correct-crlf-in-python | |
| 48 # https://www.w3.org/Protocols/rfc1341/7_2_Multipart.html | |
| 49 | |
| 50 # Create a multipart message whose body contains all the input DICOM files | |
| 51 boundary = str(uuid.uuid4()) # The boundary is a random UUID | |
| 52 body = bytearray() | |
| 33 | 53 |
| 54 if isinstance(dicom, list): | |
| 55 for i in range(dicom): | |
| 83 | 56 _AttachPart(body, dicom[i], 'application/dicom', boundary) |
| 33 | 57 else: |
| 83 | 58 _AttachPart(body, dicom, 'application/dicom', boundary) |
| 33 | 59 |
| 83 | 60 # Closing boundary |
| 61 body += bytearray('--%s--' % boundary, 'ascii') | |
| 33 | 62 |
| 83 | 63 # Do the HTTP POST request to the STOW-RS server |
| 64 headers = { | |
| 65 'Content-Type' : 'multipart/related; type=application/dicom; boundary=%s' % boundary, | |
| 66 'Accept' : 'application/json', | |
| 67 } | |
| 33 | 68 |
|
398
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
69 (response, content) = DoPostRaw(orthanc, uri, body, headers = headers) |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
70 |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
71 return (response.status, DecodeJson(content)) |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
72 |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
73 |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
74 def SendStow(orthanc, uri, dicom): |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
75 (status, content) = SendStowRaw(orthanc, uri, dicom) |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
76 if not (status in [ 200 ]): |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
77 raise Exception('Bad status: %d' % status) |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
78 else: |
|
9528e2a03d3c
adapt DICOMweb tests following fix of issue 196 (STOW-RS: Should return 200 only when successfully stored all instances)
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
363
diff
changeset
|
79 return content |
|
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
80 |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
81 |
|
297
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
82 def DoGetMultipart(orthanc, uri, headers = {}, returnHeaders = False): |
|
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
83 answer = DoGetRaw(orthanc, uri, headers = headers) |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
84 |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
85 header = '' |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
86 for i in answer[0]: |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
87 header += '%s: %s\r\n' % (i, answer[0][i]) |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
88 |
|
226
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
89 b = bytearray() |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
90 b.extend(header.encode('ascii')) |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
91 b.extend(b'\r\n') |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
92 b.extend(answer[1]) |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
93 |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
94 if (sys.version_info >= (3, 0)): |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
95 msg = email.message_from_bytes(b) |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
96 else: |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
97 msg = email.message_from_string(b) |
|
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
98 |
|
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
99 if not msg.is_multipart(): |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
100 raise Exception('Not a multipart message') |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
101 |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
102 result = [] |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
103 |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
104 for part in msg.walk(): |
|
226
230aede7f8d5
test_bitbucket_issue_96
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
222
diff
changeset
|
105 payload = part.get_payload(decode = True) |
|
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
106 if payload != None: |
|
297
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
107 if returnHeaders: |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
108 h = {} |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
109 for (key, value) in part.items(): |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
110 h[key] = value |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
111 result.append((payload, h)) |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
112 else: |
|
f95cd3af1c7a
DICOMweb: test_transcoding
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
260
diff
changeset
|
113 result.append(payload) |
|
222
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
114 |
|
0f03ee6ffa80
DICOMweb: test_wado_hierarchy, test_wado_bulk, test_bitbucket_issue_112
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
195
diff
changeset
|
115 return result |