PyBounds/PyBounds.py at master · MAXZVER/PyBounds · 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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
#!/usr/bin/python3.5

__author__ = 'admin'

import getopt
import sys
import TaskExecutor
from shutil import rmtree
from tempfile import mkdtemp
from External.TaskParser import TaskParser
from Shared.Files import *


def usage():
    print()
    print("USAGE:")
    print()
    print("  -h | --help                          - prints this message")
    print()
    print(" [-v | -s] [-t secs] [-m kilobytes] task_file")
    print("                                       - run task from 'task_file'")
    print("                                          in normal, (v)erbose or (s)ilent mode")
    print("                                          with optional time and/or memory limit")
    print()
    print("  -r | --reg module cls1 [ cls2 [...]] - register argument classes 'cls1'")
    print("                                          'cls2', ... from module 'module'")
    print()
    print("  -u | --unreg cls1 [ cls2 [...] ]     - unregister argument classes 'cls1'")
    print("                                          'cls2', ... ")
    print()
    print("  -p | --print                         - show registered classes")
    print()
    print("  -c | --clear                         - clear registered classes")
    print()
pass

verbose = False
silent = False

def main():

    global verbose
    global silent
    smart = False
    arch = "x86"

    try:
        opts, args = getopt.getopt(sys.argv[1:], "vst:m:h:r:upc", ["help", "register=", "unregister", "print", "clear", "smart="])
    except getopt.GetoptError as err:
        print("    " + str(err))
        usage()
        sys.exit(2)
    pass

    # if not os.path.exists(os.path.expanduser("~/.PyBounds")):
    # 	os.mkdir(os.path.expanduser("~/.PyBounds"))

    dir_temp= mkdtemp()

    files = Files(dir_temp, '')
    if not os.path.exists(files.regclasses):
        try:
            open(files.regclasses, 'w').close()
        except:
            pass

    if len(sys.argv) == 1:
        usage()
        sys.exit()

    time = 1.5        # sec
    memory = 1000000  # Kb

    for o, a in opts:

        if o in ("-h", "--help"):
            usage()
            sys.exit()

        elif o == "--smart":
            arch = a
            smart = True

        elif o == "-v":
            if silent:
                print("    silent and verbose are exclusive options")
                usage()
                sys.exit(1)
            verbose = True

        elif o == "-s":
            if verbose:
                print("    silent and verbose are exclusive options")
                usage()
                sys.exit(1)
            silent = True

        elif o == "-t":
            try:
                time = float(a)
            except:
                print("    time must be a number")
                usage()
                exit(1)

            if time < 0:
                print("    time must be positive")
                usage()
                exit(1)

            if time < 1.5:
                print("    recommended time limit: > 1.5 seconds")

        elif o == "-m":
            try:
                memory = int(a)
            except:
                print("    memory must be a number")
                usage()
                exit(1)

            if memory < 0:
                print("    memory must be positive")
                usage()
                exit(1)

            if memory < 100000:
                print("    recommended memory limit: > 100000 Kb")

        elif o in ("-r", "--register"):
            module = a
            classes = args
            parser = TaskParser()
            parser.load_registered_classes(files.regclasses, silent=True)
            #print "registering classes " + " ".join(classes) + " in module " + module
            parser.register_by_query(module + " " + " ".join(classes), silent=False)
            parser.save_registered_classes(files.regclasses)
            sys.exit()

        elif o in ("-u", "--unregister"):
            classes = args
            parser = TaskParser()
            parser.load_registered_classes(files.regclasses, silent=True)
            #print "unregistering classes" + " ".join(classes) + " in module " + module
            parser.unregister_by_query(" ".join(classes), silent=False)
            parser.save_registered_classes(files.regclasses)
            sys.exit()

        elif o in ("-p", "--print"):
            os.system("cat " + files.regclasses + " 2>/dev/null")
            sys.exit()

        elif o in ("-c", "--clear"):
            os.system("rm " + files.regclasses + " 2>/dev/null")
            sys.exit()

        else:
            pass

    pass

    if len(args) > 0:
        TaskExecutor.main(dir_temp, args[0], time, memory, verbose, silent, smart, arch)
        rmtree(dir_temp)
    else:
        print("    required task_file")
        usage()
        exit(1)

pass

if __name__ == "__main__":
    main()