Merge pull request #13 from fedorov/truncate-time-fraction

Add truncate time option
pieper · Jun 28, 2019 · 615cede · 615cede
2 parents 8312a30 + 5466a1c
commit 615cede
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/dicomsort.py b/dicomsort.py
@@ -65,7 +65,9 @@ def __init__(self):
                 '-t': 'test',
                 '--test': 'test',
                 '-u': 'unsafe',
-                '--unsafe': 'unsafe'
+                '--unsafe': 'unsafe',
+                '-r': 'truncateTime',
+                '--truncateTime': 'truncateTime'
                 }
 
         self.defaultOptions = {
@@ -77,7 +79,8 @@ def __init__(self):
                 'keepGoing': False,
                 'verbose': False,
                 'test': False,
-                'unsafe': False
+                'unsafe': False,
+                'truncateTime': False
                 }
 
         self.requiredOptions = [ 'sourceDir', 'targetPattern', ]
@@ -128,6 +131,9 @@ def pathFromDatasetPattern(self,ds,safe=True):
                 value = ""
             if value == "":
                 value = "Unknown%s" % key
+            if self.options['truncateTime']:
+              if key.endswith("Time") and str(value)[str(value).find('.')+1:] == '000000':
+                value = str(value)[:str(value).find('.')]
             if safe:
               replacements[key] = self.safeFileName(str(value))
             else: