Mercurial > hg > orthanc
annotate Resources/Samples/Python/RestToolbox.py @ 3841:be7df7fe3d80
avoid one memcpy of the DICOM buffer on "POST /instances"
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Thu, 16 Apr 2020 16:58:37 +0200 |
parents | 94f4a18a79cc |
children |
rev | line source |
---|---|
747
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
1 # Orthanc - A Lightweight, RESTful DICOM Store |
1900 | 2 # Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
1288
6e7e5ed91c2d
upgrade to year 2015
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1184
diff
changeset
|
3 # Department, University Hospital of Liege, Belgium |
3640
94f4a18a79cc
upgrade to year 2020
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
3060
diff
changeset
|
4 # Copyright (C) 2017-2020 Osimis S.A., Belgium |
747
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
5 # |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
6 # This program is free software: you can redistribute it and/or |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
7 # modify it under the terms of the GNU General Public License as |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
8 # published by the Free Software Foundation, either version 3 of the |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
9 # License, or (at your option) any later version. |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
10 # |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
11 # This program is distributed in the hope that it will be useful, but |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
12 # WITHOUT ANY WARRANTY; without even the implied warranty of |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
14 # General Public License for more details. |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
15 # |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
16 # You should have received a copy of the GNU General Public License |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
17 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
18 |
44382c8bcd15
added explicit licensing terms for samples
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
424
diff
changeset
|
19 |
340 | 20 import httplib2 |
21 import json | |
1184 | 22 import sys |
23 | |
24 if (sys.version_info >= (3, 0)): | |
25 from urllib.parse import urlencode | |
26 else: | |
27 from urllib import urlencode | |
28 | |
340 | 29 |
346 | 30 _credentials = None |
31 | |
1184 | 32 |
33 def _DecodeJson(s): | |
34 try: | |
35 if (sys.version_info >= (3, 0)): | |
36 return json.loads(s.decode()) | |
37 else: | |
38 return json.loads(s) | |
39 except: | |
40 return s | |
41 | |
42 | |
346 | 43 def SetCredentials(username, password): |
44 global _credentials | |
45 _credentials = (username, password) | |
46 | |
47 def _SetupCredentials(h): | |
48 global _credentials | |
49 if _credentials != None: | |
50 h.add_credentials(_credentials[0], _credentials[1]) | |
51 | |
2795
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
52 def _ComputeGetUri(uri, data): |
340 | 53 d = '' |
54 if len(data.keys()) > 0: | |
55 d = '?' + urlencode(data) | |
56 | |
2795
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
57 return uri + d |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
58 |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
59 def DoGet(uri, data = {}, interpretAsJson = True): |
340 | 60 h = httplib2.Http() |
346 | 61 _SetupCredentials(h) |
2795
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
62 resp, content = h.request(_ComputeGetUri(uri, data), 'GET') |
340 | 63 if not (resp.status in [ 200 ]): |
64 raise Exception(resp.status) | |
424 | 65 elif not interpretAsJson: |
1184 | 66 return content.decode() |
340 | 67 else: |
1184 | 68 return _DecodeJson(content) |
340 | 69 |
70 | |
2795
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
71 def DoRawGet(uri, data = {}): |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
72 h = httplib2.Http() |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
73 _SetupCredentials(h) |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
74 resp, content = h.request(_ComputeGetUri(uri, data), 'GET') |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
75 if not (resp.status in [ 200 ]): |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
76 raise Exception(resp.status) |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
77 else: |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
78 return content |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
79 |
f2e49b953e86
RestToolbox.DoRawGet
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2447
diff
changeset
|
80 |
340 | 81 def _DoPutOrPost(uri, method, data, contentType): |
82 h = httplib2.Http() | |
346 | 83 _SetupCredentials(h) |
340 | 84 |
85 if isinstance(data, str): | |
86 body = data | |
87 if len(contentType) != 0: | |
88 headers = { 'content-type' : contentType } | |
354 | 89 else: |
90 headers = { 'content-type' : 'text/plain' } | |
340 | 91 else: |
92 body = json.dumps(data) | |
93 headers = { 'content-type' : 'application/json' } | |
94 | |
95 resp, content = h.request( | |
96 uri, method, | |
97 body = body, | |
98 headers = headers) | |
99 | |
100 if not (resp.status in [ 200, 302 ]): | |
101 raise Exception(resp.status) | |
102 else: | |
1184 | 103 return _DecodeJson(content) |
340 | 104 |
105 | |
106 def DoDelete(uri): | |
107 h = httplib2.Http() | |
346 | 108 _SetupCredentials(h) |
340 | 109 resp, content = h.request(uri, 'DELETE') |
346 | 110 |
340 | 111 if not (resp.status in [ 200 ]): |
112 raise Exception(resp.status) | |
113 else: | |
1184 | 114 return _DecodeJson(content) |
340 | 115 |
116 | |
117 def DoPut(uri, data = {}, contentType = ''): | |
118 return _DoPutOrPost(uri, 'PUT', data, contentType) | |
119 | |
120 | |
121 def DoPost(uri, data = {}, contentType = ''): | |
122 return _DoPutOrPost(uri, 'POST', data, contentType) |