diff --git a/Util/MachineProductCfg.py b/Util/MachineProductCfg.py new file mode 100644 index 0000000..c0ad8f9 --- /dev/null +++ b/Util/MachineProductCfg.py @@ -0,0 +1,39 @@ +import json +import xmltodict + + +# Open the MachineProductCfg.xml file in the root directory +with open("MachineProductCfg.xml") as MPCxml: + MPCdict = xmltodict.parse(MPCxml.read()) + MPCdump = json.dumps(MPCdict) + data = json.loads(MPCdump) + + +def getPrimaryLocations(): + """ Returns all of the primary locations in the MachineProductCfg """ + locationIds = [] + # iterate on the json data and grab anything that has PrimaryLocation. + # Also needs to return anything in the Regional area. + for i in data['Config']['ConfigDef']['ConfigItems']['ConfigItem']: + if "PrimaryLocation" in i['@key'] and i['@value'] != "": + # Split the string up + locationIds.append(i['@value'].split("_")[2]) + + if "NearbyLocation" in i['@key'] and i['@value'] != "": + locationIds.append(i['@value'].split("_")[2]) + + return locationIds + + +def getAirportCodes(): + """ Returns all of the airport identifiers present in the MachineProductCfg """ + airports = [] + for i in data['Config']['ConfigDef']['ConfigItems']['ConfigItem']: + if "Airport" in i['@key'] and i['@value'] != "": + # Split the string up + airports.append(i['@value'].split("_")[2]) + + return airports + + +print(getPrimaryLocations()) diff --git a/py2Lib/MachineProductCfg.py b/py2Lib/MachineProductCfg.py deleted file mode 100644 index 8b13789..0000000 --- a/py2Lib/MachineProductCfg.py +++ /dev/null @@ -1 +0,0 @@ -