Mercurial > hg > orthanc
annotate Resources/Samples/ImportDicomFiles/ImportDicomFiles.py @ 370:22bb88181e06
credentials
author | jodogne |
---|---|
date | Tue, 12 Mar 2013 11:14:45 +0100 |
parents | 77e526e6fdf8 |
children | 7000726bade7 |
rev | line source |
---|---|
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
1 #!/usr/bin/python |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
2 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
3 import os |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
4 import sys |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
5 import os.path |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
6 import httplib2 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
7 |
370 | 8 if len(sys.argv) != 4 and len(sys.argv) != 6: |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
9 print(""" |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
10 Sample script to recursively import in Orthanc all the DICOM files |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
11 that are stored in some path. Please make sure that Orthanc is running |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
12 before starting this script. The files are uploaded through the REST |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
13 API. |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
14 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
15 Usage: %s [hostname] [HTTP port] [path] |
370 | 16 Usage: %s [hostname] [HTTP port] [path] [username] [password] |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
17 For instance: %s localhost 8042 . |
370 | 18 """ % (sys.argv[0], sys.argv[0], sys.argv[0])) |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
19 exit(-1) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
20 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
21 URL = 'http://%s:%d/instances' % (sys.argv[1], int(sys.argv[2])) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
22 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
23 success = 0 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
24 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
25 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
26 # This function will upload a single file to Orthanc through the REST API |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
27 def UploadFile(path): |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
28 global success |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
29 |
370 | 30 f = open(path, "rb") |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
31 content = f.read() |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
32 f.close() |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
33 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
34 try: |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
35 sys.stdout.write("Importing %s" % path) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
36 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
37 h = httplib2.Http() |
370 | 38 |
39 if len(sys.argv) == 6: | |
40 h.add_credentials(sys.argv[4], sys.argv[5]) | |
41 | |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
42 resp, content = h.request(URL, 'POST', |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
43 body = content, |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
44 headers = { 'content-type' : 'application/dicom' }) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
45 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
46 if resp.status == 200: |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
47 sys.stdout.write(" => success\n") |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
48 success += 1 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
49 else: |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
50 sys.stdout.write(" => failure (is it a DICOM file?)\n") |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
51 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
52 except: |
370 | 53 sys.stdout.write(" => unable to connect (is there a password?)\n") |
280
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
54 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
55 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
56 if os.path.isfile(sys.argv[3]): |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
57 # Upload a single file |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
58 UploadFile(sys.argv[3]) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
59 else: |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
60 # Recursively upload a directory |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
61 for root, dirs, files in os.walk(sys.argv[3]): |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
62 for f in files: |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
63 UploadFile(os.path.join(root, f)) |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
64 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
65 |
77e526e6fdf8
sample script to import DICOM files in Orthanc
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
66 print("\nSummary: %d DICOM file(s) have been imported" % success) |