OOHelper/utils.py at master · Eadral/OOHelper · 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
import os
import random
import shutil
import sys
import re
import threading

import sqlite3

conn = sqlite3.connect("test_data/datacheck.db3", check_same_thread=False)
cur = conn.cursor()


def get_paths_recursively(folder):
    paths = []
    files = os.listdir(folder)
    for file in files:
        path = os.path.join(folder, file)
        if os.path.isdir(path):
            paths += get_paths_recursively(path)
        else:
            paths.append(path)
    return paths


def datacheck(test_data_in):
    # print(test_data_in)
    time = get_time(test_data_in)
    return time, 200
    # print("\033[1;33mDatacheck unavailable now.\033[0m")
    # return 65535, 65535
    # result = os.popen(r"{} -i {}".format(os.path.join("test_data", "datacheck.exe"), test_data_in))
    # output = result.read()
    # basetime = re.search(r'base time is (\d+)', output, re.M)
    # maxtime = re.search(r'max time is (\d+)', output, re.M)
    # if basetime is None:
    #     print("\033[1;31mIllegal input: {}\033[0m".format(test_data_in))
    #     return 0, 0
    # return int(basetime.group(1)), int(maxtime.group(1))


def precompile(project_dir, main, main_path, jar, temp):
    if os.path.exists(temp):
        shutil.rmtree(temp)
        os.mkdir(temp)
    class_path = os.path.join(temp, "{}.class".format(main))
    if os.path.exists(class_path):
        os.remove(class_path)
    output_path = os.path.join(temp, "output.txt")
    if os.path.exists(output_path):
        os.remove(output_path)
    src_path = os.path.join(project_dir, "src")
    compile_source(src_path, main_path, jar)


def compile_source(src, main_path=".", jar=".;"):
    # src_path = os.path.join(project_dir, "src", "*.java")
    here = os.getcwd()
    os.chdir(src)
    cmd = 'javac -encoding UTF-8 {} -d {} -cp {}'.format(os.path.join(main_path, "*.java"), os.path.join(here, "temp"),
                                                         jar)
    print(cmd)
    os.system(cmd)
    os.chdir(here)


db_mutex = threading.Lock()


def insert(name, time):
    db_mutex.acquire()
    cur.execute(r"insert into datacheck (name, time) values ('{}', {})".format(name, time))
    db_mutex.release()


def select(name):
    db_mutex.acquire()
    cur.execute("select * from datacheck where name = '{}'".format(name))
    r = cur.fetchall()
    db_mutex.release()
    return r


def delete(name):
    db_mutex.acquire()
    cur.execute("delete from datacheck where name = '{}'".format(name))
    db_mutex.release()


def get_time(filename):
    r = select(filename)
    if len(r) == 0:
        return 65535
    else:
        return r[0][1]


def update_time(filename, time):
    t = get_time(filename)
    if time < t:
        delete(filename)
        insert(filename, time)


if __name__ == "__main__":
    print(update_time("qqq", 2))
    print(get_time("qqq"))