Mercurial > hg > orthanc

comparison Resources/Samples/Python/AutoClassify.py @ 1184:4e9d517503ae

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
port to Python3
author Sebastien Jodogne <s.jodogne@gmail.com>
date Fri, 10 Oct 2014 09:13:48 +0200
parents 6ef2c81581cd
children 6e7e5ed91c2d
comparison
equal deleted inserted replaced
1183:6ef2c81581cd 1184:4e9d517503ae
21 21
22 import argparse 22 import argparse
23 import time 23 import time
24 import os 24 import os
25 import os.path 25 import os.path
26 import sys
26 import RestToolbox 27 import RestToolbox
27 28
28 parser = argparse.ArgumentParser( 29 parser = argparse.ArgumentParser(
29 description = 'Automated classification of DICOM files from Orthanc.', 30 description = 'Automated classification of DICOM files from Orthanc.',
30 formatter_class = argparse.ArgumentDefaultsHelpFormatter) 31 formatter_class = argparse.ArgumentDefaultsHelpFormatter)
42 help = 'Remove DICOM files from Orthanc once classified (disabled by default)') 43 help = 'Remove DICOM files from Orthanc once classified (disabled by default)')
43 parser.set_defaults(remove = False) 44 parser.set_defaults(remove = False)
44 45
45 46
46 def FixPath(p): 47 def FixPath(p):
47 return p.encode('ascii', 'ignore').strip() 48 return p.encode('ascii', 'ignore').strip().decode()
48 49
49 def GetTag(resource, tag): 50 def GetTag(resource, tag):
50 if ('MainDicomTags' in resource and 51 if ('MainDicomTags' in resource and
51 tag in resource['MainDicomTags']): 52 tag in resource['MainDicomTags']):
52 return resource['MainDicomTags'][tag] 53 return resource['MainDicomTags'][tag]
68 GetTag(series, 'SeriesDescription')) 69 GetTag(series, 'SeriesDescription'))
69 d = '%s.dcm' % GetTag(instance, 'SOPInstanceUID') 70 d = '%s.dcm' % GetTag(instance, 'SOPInstanceUID')
70 71
71 p = os.path.join(args.target, FixPath(a), FixPath(b), FixPath(c)) 72 p = os.path.join(args.target, FixPath(a), FixPath(b), FixPath(c))
72 f = os.path.join(p, FixPath(d)) 73 f = os.path.join(p, FixPath(d))
73 74
74 # Copy the DICOM file to the target path 75 # Copy the DICOM file to the target path
75 print('Writing new DICOM file: %s' % f) 76 print('Writing new DICOM file: %s' % f)
76 77
77 try: 78 try:
78 os.makedirs(p) 79 os.makedirs(p)