SaveWizard/util.py at dev · JDM170/SaveWizard · 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
#!/usr/bin/python3
# -*- coding: utf-8 -*-

from os.path import isfile
from re import search, match
# from re import search, match, sub
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import (
    QMessageBox, QProgressDialog, QApplication
)
from hashlib import md5
from statics import hash_chunk_size


class CustomFuncs:
    # Custom functions
    def __init__(self):
        self.lines = []

    def set_lines(self, new_lines):
        self.lines = new_lines

    def get_lines(self, index=None):
        if index is not None:
            return self.lines[index]
        return self.lines

    @staticmethod
    def generate_md5(fn):
        if not isfile(fn):
            return False
        hash_md5 = md5()
        with open(fn, "rb") as f:
            for chunk in iter(lambda: f.read(hash_chunk_size), b""):
                hash_md5.update(chunk)
        return hash_md5.hexdigest()

    @staticmethod
    def show_progress_bar(title, text, length):
        if (not title) or (not text) or (not length) or (length <= 0):
            return
        progress_bar = QProgressDialog(text + "...", None, 0, length, flags=Qt.Window | Qt.WindowTitleHint)
        progress_bar.setWindowTitle(title)
        progress_bar.setWindowModality(Qt.WindowModal)
        progress_bar.show()
        return progress_bar

    @staticmethod
    def update_progress_bar(progress_bar, value=1):
        if not progress_bar:
            return
        progress_bar.setValue(progress_bar.value() + value)
        QApplication.processEvents()

    # Stock functions
    def search_line(self, term, start=0, cancel=r"this_string_must_not_exist"):
        if search(term, self.lines[start]):
            return start
        start += 1
        while start <= len(self.lines) - 1:
            if search(term, self.lines[start]):
                return start
            if search(cancel, self.lines[start]):
                return None
            start += 1
        return None

    def search_line_in_unit(self, term, unit):
        line = self.search_line(" : " + unit + " {")
        return self.search_line(term, start=line, cancel="}")

    def search_all_lines(self, term):
        matches = []
        line = 0
        while self.search_line(term, start=line + 1):
            line = self.search_line(term, start=line + 1)
            matches.append(line)
        if matches is None:
            return None
        return matches

    def get_value(self, line):
        return search(r": (.+)$", self.lines[line]).group(1)

    def set_value(self, line, value):
        name = match(r"(.+):", self.lines[line]).group(1)
        self.lines[line] = name + ": " + value

    # def get_unit_name(self, line):
    #     return search(r" : (.+) {$", self.lines[line]).group(1)

    def get_array_length(self, line):
        return int(search(r": ([0-9]+)$", self.lines[line]).group(1))

    # def get_array_value_by_index(self, line, index):
    #     return search(r": (.+)$", self.lines[line + index + 1]).group(1)

    # def get_array_index_by_value(self, line, value):
    #     count = 0
    #     for i in range(self.get_array_length(line)):
    #         if self.get_value(line + count + 1) == value:
    #             return count
    #         count += 1
    #     return None

    def get_array_items(self, line):
        items = []
        for i in range(self.get_array_length(line)):
            items.append(search(r": (.+)$", self.lines[line + i + 1]).group(1))
        if items is None:
            return None
        return items

    def add_array_value(self, line, value):
        name = match(r"(.+):", self.lines[line]).group(1)
        count = self.get_array_length(line)
        self.lines[line] = name + ": " + str(count + 1)
        self.lines.insert(line + count + 1, name + "[" + str(count) + "]: " + value)

    # def remove_array_value(self, line, value):
    #     name = match(r"(.+):", self.lines[line]).group(1)
    #     del self.lines[line + 1 + self.get_array_index_by_value(line, value)]
    #     count = self.get_array_length(line)
    #     self.lines[line] = name + ": " + str(count - 1)
    #     for i in range(count):
    #         self.lines[line + i + 1] = sub(r"\[[0-9]+]", "[" + str(i) + "]", lines[line + i + 1])

    # def change_array_value(self, line, index, value):
    #     line += index + 1
    #     self.set_value(line, value)


util = CustomFuncs()