utils.py 1.6 KB
Newer Older
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
1
2
3
4
5
6
import argparse
from datetime import date


def add_data_args(parser: argparse.ArgumentParser):
    parser.add_argument(
Gustaf Ahdritz's avatar
Fixes  
Gustaf Ahdritz committed
7
        '--uniref90_database_path', type=str, default=None,
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
8
9
    )
    parser.add_argument(
Gustaf Ahdritz's avatar
Fixes  
Gustaf Ahdritz committed
10
        '--mgnify_database_path', type=str, default=None,
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
11
12
    )
    parser.add_argument(
Gustaf Ahdritz's avatar
Fixes  
Gustaf Ahdritz committed
13
        '--pdb70_database_path', type=str, default=None,
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
14
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
15
16
17
    parser.add_argument(
        '--pdb_seqres_database_path', type=str, default=None,
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
18
    parser.add_argument(
Gustaf Ahdritz's avatar
Fixes  
Gustaf Ahdritz committed
19
        '--uniclust30_database_path', type=str, default=None,
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
20
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
21
22
23
    parser.add_argument(
        '--uniprot_database_path', type=str, default=None,
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
24
25
26
27
28
29
30
31
32
33
34
35
    parser.add_argument(
        '--bfd_database_path', type=str, default=None,
    )
    parser.add_argument(
        '--jackhmmer_binary_path', type=str, default='/usr/bin/jackhmmer'
    )
    parser.add_argument(
        '--hhblits_binary_path', type=str, default='/usr/bin/hhblits'
    )
    parser.add_argument(
        '--hhsearch_binary_path', type=str, default='/usr/bin/hhsearch'
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
36
37
38
39
40
41
    parser.add_argument(
        '--hmmsearch_binary_path', type=str, default='/usr/bin/hmmsearch'
    )
    parser.add_argument(
        '--hmmbuild_binary_path', type=str, default='/usr/bin/hmmbuild'
    )
Gustaf Ahdritz's avatar
Gustaf Ahdritz committed
42
43
44
45
46
47
48
49
50
51
    parser.add_argument(
        '--kalign_binary_path', type=str, default='/usr/bin/kalign'
    )
    parser.add_argument(
        '--max_template_date', type=str, 
        default=date.today().strftime("%Y-%m-%d"),
    )
    parser.add_argument(
        '--obsolete_pdbs_path', type=str, default=None
    )
52
53
54
    parser.add_argument(
        '--release_dates_path', type=str, default=None
    )