microCT_auto_processing/listdir_ctan.py at main · Tiankuo528/microCT_auto_processing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#! /home/spl/ml/sitk/bin/python

# -*- coding: utf-8 -*-

import os
from pathlib import Path
import sys
import logging

if len(sys.argv) == 1:
    Masterdir = Path.cwd()
else:
    Masterdir = Path(sys.argv[1])

ctan_list =os.path.join(Masterdir,os.path.basename(Masterdir)+'_CTANlist.ctl')
info ='Info=0000000001000000FFFF0000000000000000000000000000000000000000000000000000000000000000000000000000E60E993B64EED03F00000000000000000000000000000000000000000000000004000000000000002C'
i=0

if os.path.exists(ctan_list):
    os.remove(ctan_list)

format = "%(asctime)s: %(message)s"
logging.basicConfig(format=format, level=logging.INFO, datefmt="%H:%M:%S")

#method 1
with open(ctan_list,'a') as thefile:
    thefile.write('[Dataset list]\n')
    for folder in sorted(os.listdir(Masterdir)):
        if os.path.isdir(os.path.join(Masterdir,folder)) :#and "composite" in folder:
            file =os.path.join(Masterdir,folder,os.listdir(os.path.join(Masterdir,folder))[2])
            thefile.write('Next=@{:d}\n'.format(i))
            thefile.write('[@{:d}]\n'.format(i))
            thefile.write('File={:s}\n'.format(file))
            thefile.write('{:s}\n'.format(info))
            i=i+1

logging.info("File saved as {}".format(ctan_list))
'''
#method 2
w = open(ctan_list,'a')
w.write('[Dataset list]\n')
for folder in sorted(os.listdir(Masterdir)):
        if os.path.isdir(os.path.join(Masterdir,folder)):
            file =os.path.join(Masterdir,folder,os.listdir(os.path.join(Masterdir,folder))[5])
            w.write('Next=@{:d}\n'.format(i))
            w.write('[@{:d}]\n'.format(i))
            w.write('File={:s}\n'.format(file))
            w.write('{:s}\n'.format(info))
            i=i+1
w.close()
'''

'''
#method 3
ctan = ['[Dataset list]\n']
for folder in sorted(os.listdir(Masterdir)):
        if os.path.isdir(os.path.join(Masterdir,folder)):
            file =os.path.join(Masterdir,folder,os.listdir(os.path.join(Masterdir,folder))[10])
            ctan.append('Next=@{:d}\n'.format(i))
            ctan.append('[@{:d}]\n'.format(i))
            ctan.append('File={:s}\n'.format(file))
            ctan.append('{:s}\n'.format(info))
            i=i+1
w = open(ctan_list,'w')
w.writelines(ctan)
w.close()
'''