ManurePyrolysisIAM/read_GCAM_DB.py at master · PEESEgroup/ManurePyrolysisIAM · 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
import xml.etree.cElementTree as ET
from xml.dom import minidom
import subprocess
import constants as c
import os


def build_batch_query(version, RCP, fname):
    """
    Warning - requires hard-coded filepaths, which can be changed in the constants.py file
    Builds a batch query file with all the necessary things.
    :param fname: filename of the xml file
    :param version: the model version name
    :param RCP: the RCP pathway used for the model version
    :return: directory path and file name of output file to be created if not already exist
    """
    root = ET.Element("ModelInterfaceBatch")
    dbviewer = ET.SubElement(root, "class", name="ModelInterface.ModelGUI2.DbViewer")
    command = ET.SubElement(dbviewer, "command", name="XMLDB Batch File")

    ET.SubElement(command, "scenario", name="GCAM_SSP1")
    ET.SubElement(command, "scenario", name="GCAM_SSP2")
    ET.SubElement(command, "scenario", name="GCAM_SSP3")
    ET.SubElement(command, "scenario", name="GCAM_SSP4")
    ET.SubElement(command, "scenario", name="GCAM_SSP5")

    ET.SubElement(command, "queryFile").text = "xml/query_list.xml"

    outdir = "data/gcam_out/" + str(version) + "/" + str(RCP)
    outfile = outdir + "/ref.csv"

    ET.SubElement(command, "outFile").text = outfile

    ET.SubElement(command, "xmldbLocation").text = c.GCAMConstants.XML_DB_loc + str(version) + "_" + str(RCP)

    xmlstr = minidom.parseString(ET.tostring(root)).toprettyxml(indent="   ")
    with open(fname, "w") as f:
        f.write(xmlstr)

    return outdir, outfile


def main():
    """
    control program for querying GCAM databases
    """
    for i in range(len(c.GCAMConstants.version)):
        # build a batch file for every database
        filename = "xml/xmldb_batch_" + str(c.GCAMConstants.version[i][0]) + "_" + str(
            c.GCAMConstants.version[i][1]) + ".xml"
        print(filename)
        out_dir, out_file = build_batch_query(c.GCAMConstants.version[i][0], c.GCAMConstants.version[i][1], filename)

        # create output directory
        os.makedirs(out_dir, exist_ok=True)
        with open(out_file, "w") as f:
            f.write("foo")

        # edit the batch file to include the right xml query file
        # read the file into a list of lines
        bat_file = "xml/launch_model_interface.bat"
        lines = open(bat_file, 'r').readlines()

        # now edit the last line of the list of lines
        new_last_line = ("java ModelInterface.InterfaceMain -b " + filename)
        lines[-1] = new_last_line

        print(lines[-1])

        # now write the modified list back out to the file
        open(bat_file, 'w').writelines(lines)

        # execute the batch query on the command line
        subprocess.call([r'xml\launch_model_interface.bat'])


if __name__ == '__main__':
    main()