NCTPtools/cli.py at main · alejopages/NCTPtools · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import argparse
import pathlib
from http_client import *
from upload import all, sites, varieties, results
from payloads import all_payloads
from os.path import join, exists
from os import remove
from datetime import datetime as dt


def config(key):
    with open('config.json', 'r') as config_file:
        return json.load(config_file).get(key, None)


class UploadAction(argparse.Action):

    def __init__(self, option_strings, dest, **kwargs):
        self.domain_mappings = config('domain_mappings')
        super().__init__(option_strings, dest, 1, None, kwargs.get('default', 'local'), None, self.domain_mappings.keys(), True, "Prepare for or upload to the specified environment")

    def __call__(self, parser, namespace, values, option_string=None):
        setattr(namespace, 'domain', self.domain_mappings.get(values[0]))


class Cli:

    def __init__(self):
        self.__parser = self.__define_parser()
        self.args = self.__parser.parse_args()
        self.__load_config()
        if hasattr(self.args, 'upload'):
            self.client = VarietyTestingHttpClient(self.args.domain,self.config,self.args.loud)
            self.__fetch_year()
            self.__open_and_set_manifest()

    def run(self):
        self.args.op(self)

    def write_to_manifest(self, why, path, *args):
        args_str = ',' + ','.join(map(lambda arg: str(arg).replace(",", "\\,"), args))
        self.manifest.write(f"[{why}],{path}{args_str}\n")

    def __load_config(self):
        with open('config.json', 'r') as config_file:
            self.config = json.load(config_file)

    def __fetch_year(self):
        year_id = self.client.get_year_id(self.args.year)
        if year_id is None:
            print(f"\n> Have you created the harvest year publication?")
            exit("\nfailed\n")
        setattr(self.args, 'year_id', year_id)

    def __open_and_set_manifest(self):
        if exists(path:=join(self.args.inpath, f'manifest_{dt.today().isoformat()}.csv')):
            remove(path)
        self.manifest = open(path, 'w')

    def __del__(self):
        if hasattr(self, 'manifest'):
            self.manifest.close()

    @staticmethod
    def __define_parser() -> argparse.ArgumentParser:
        parser = argparse.ArgumentParser()

        parser.add_argument("-e", "--env", action=UploadAction)
        parser.add_argument("-l", "--loud", help="Log everything about outgoing requests", action='store_true')
        command_subparsers = parser.add_subparsers(title = 'actions', help="Actions")

        payloads_parser = command_subparsers.add_parser('payloads', help='Generate request payloads')
        payloads_parser.add_argument("-i", "--inpath", help="Path to trial csv data provided by PREC", type=pathlib.Path, required=True)
        payloads_parser.add_argument("-o", "--outpath", default="./json", help="Location to dump the payloads", type=pathlib.Path, required=True)
        payloads_parser.set_defaults(op=all_payloads)

        upload_parser = command_subparsers.add_parser('upload', help='Upload payloads to a variety testing environment')
        upload_parser.set_defaults(upload=True)
        upload_parser.add_argument("-i", "--inpath", help="Path to json payloads generated by the payloads command", type=pathlib.Path, required=True)
        upload_parser.add_argument('-y', '--year', help="The trial harvest year of this data", type=int)
        upload_parser.add_argument('-r', '--rewrite', help="Rewrite i.e. delete existing and write again", action='store_true', default=False)

        upload_types = upload_parser.add_subparsers()

        all_upload_parser = upload_types.add_parser('all')
        all_upload_parser.set_defaults(op=all)

        site_upload_parser = upload_types.add_parser('sites')
        site_upload_parser.set_defaults(op=sites)

        results_parser = upload_types.add_parser('results')
        results_parser.set_defaults(op=results)

        varieties_parser = upload_types.add_parser('varieties')
        varieties_parser.set_defaults(op=varieties)

        return parser


cli = Cli()
cli.run()