Python/PatchCLIProcessor.py at master · danengh/Python · 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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
#!/usr/local/bin/python

# Python processor for running the patchlib (patchcli) created by
# Bryson Tyrell
# Copyright (C) 2019  Dan Engh

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import os
import shutil
import sys
from xml.etree import ElementTree
from xml.sax.saxutils import escape
from subprocess import call
from collections import OrderedDict

class patchUpdater():
    def __init__(self):
        self.application_path = None
        self.version = None
        self.os_requirements = None
        self.publisher = None
        self.extension_attribute = None
        self.file_output_location = None
        self.patch_profile = None

def patchcliProc():
    try:
        patchcli = os.path.exists("/usr/local/bin/patchcli")
        return patchcli
    except FileNotFoundError:
        print("patchlib has not been installed")

def userInput(arg):
    returnVar = input(str("Input value for {0}: ").format(arg))
    if returnVar == "":
        returnVar = None
    return returnVar

def xmlInput(xmlPath, xmlKey):
    if not os.path.exists(xmlPath):
        raise FileNotFoundError
        print(xmlPath, " does not exist")
    xmlElements = ElementTree.parse(xmlPath)
    return tree.get(xmlKey)
    pass

def patchcliOptions(key):
    if key == "file_output_location":
        return "-o"
    elif key == "publisher":
        return "-p"
    elif key == "extension_attribute":
        return "-e"
    elif key == "version":
        return "--app-version"
    elif key == "patch_profile":
        return "--profile"
    elif key == "os_requirements":
        return "--min-sys-version"
    else:
        pass

def runpatchcli(**kwargs):
    patchOptions = OrderedDict()
    for arg in kwargs:
        if kwargs[arg] != None:
            cliOption = patchcliOptions(arg)
            if arg == "application_path":
                pass
            else:
                patchOptions[cliOption] = kwargs[arg]
                # print(cliOption, patchOptions[cliOption])
        else:
            exit
    if len(patchOptions) != 0:
        if len(patchOptions) == 1:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], "--patch-only"])
        elif len(patchOptions) == 2:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], optionList[1], patchOptions[optionList[1]], "--patch-only"])
        elif len(patchOptions) == 3:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], optionList[1], patchOptions[optionList[1]], optionList[2], patchOptions[optionList[2]], "--patch-only"])
        elif len(patchOptions) == 4:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], optionList[1], patchOptions[optionList[1]], optionList[2], patchOptions[optionList[2]], optionList[3], patchOptions[optionList[3]], "--patch-only"])
        elif len(patchOptions) == 5:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], optionList[1], patchOptions[optionList[1]], optionList[2], patchOptions[optionList[2]], optionList[3], patchOptions[optionList[3]], optionList[4], patchOptions[optionList[4]],"--patch-only"])
        else:
            optionList=list(patchOptions.keys())
            call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], optionList[0], patchOptions[optionList[0]], optionList[1], patchOptions[optionList[1]], optionList[2], patchOptions[optionList[2]], optionList[3], patchOptions[optionList[3]], optionList[4], patchOptions[optionList[4]], optionList[5], patchOptions[optionList[5]],"--patch-only"])
    else:
        call(["/usr/local/bin/patchcli", "patch", kwargs["application_path"], "--patch-only"])


def main():
    keyInputs = {
        "application_path": patch1.application_path,
        "version": patch1.version,
        "os_requirements": patch1.os_requirements,
        "publisher": patch1.publisher,
        "extension_attribute": patch1.extension_attribute,
        "file_output_location": patch1.file_output_location,
        "patch_profile": patch1.patch_profile
    }
    for patchVar in keyInputs:
        value = userInput(patchVar)
        keyInputs[patchVar] = value
    runpatchcli(**keyInputs)

if __name__ == "__main__":
    patch1 = patchUpdater()
    main()