|
0
|
1 import sys
|
|
|
2 import shutil
|
|
|
3 import os
|
|
|
4
|
|
|
5 def check_data_format(fn):
|
|
|
6 f = open(fn,'r')
|
|
|
7 l = f.readline()
|
|
|
8 if l[0:5] == '#mega':
|
|
|
9 return 'meg'
|
|
|
10 os.environ['MDTYPE'] = 1
|
|
|
11 return 'fasta'
|
|
|
12
|
|
|
13 data_name = str(sys.argv[1])
|
|
|
14 data_mod_name = 'mega_data.'+check_data_format(data_name)
|
|
|
15 os.symlink(data_name,data_mod_name)
|
|
|
16 if len(sys.argv) == 3:
|
|
|
17 tree_name = str(sys.argv[2])
|
|
|
18 tree_mod_name = 'mega_tree.nwk'
|
|
|
19 os.symlink(tree_name,tree_mod_name)
|
|
|
20 lf = open('mega_data.txt','w')
|
|
|
21 lf.write(data_mod_name+'\n')
|
|
|
22
|
