PythonRQR/file_manager.py at main · s-razoes/PythonRQR · 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
import os
#project libraries
import config
import humanize

files_path = os.path.join(os.getcwd(), config.FILES_PATH)

def id_created(idi):
    path = os.path.join(files_path, idi + config.LOG_PREFIX)
    return os.path.exists(path)

def id_type(idi):
    path = path_file(idi)
    if os.path.exists(path) is True:
        return config.CONTENT_TYPE.FILE
    file_path = path_type(idi)
    if os.path.exists(file_path) is True:
        f = open(file_path, "r", encoding='utf8')
        ch = f.read()[0]
        type = config.CONTENT_TYPE(int(ch))
        return type

    return config.CONTENT_TYPE.NONE

def is_expired(idi):
    file_path = path_type(idi)
    if os.path.exists(file_path) is True:
        f = open(file_path, "r", encoding='utf8')
        ch = f.read()[1:]
        #TODO
        '''
        created_time = os.path.getctime(file_path)
        if ch == "1h":
            os.path.getctime
        if ch == "10min":

        if ch == '1hit':
        if ch == '10hits':
        '''
    return None

def delete_id(idi):
    file_path = path_type(idi)
    if os.path.exists(file_path) is True:
        file_path = path_type(idi)
        f = open(file_path, "w", encoding='utf8')
        type = str(config.CONTENT_TYPE.DELETED.value)
        f.write(type)
        f.close()

def setup_id(idi, key, data):
    file_path = os.path.join(files_path, idi + config.SET_PREFIX)
    f = open(file_path, "a", encoding='utf8')
    f.write(data)
    f.close()
    file_path = os.path.join(files_path, idi + config.KEY_PREFIX)
    f = open(file_path, "w", encoding='utf8')
    f.write(key)
    f.close()

def check_key(idi, key):
    if is_id_setup(idi):
        path = os.path.join(files_path, idi + config.KEY_PREFIX)
        if os.path.exists(path):
            f = open(path, "r")#, encoding='utf8')
            strline = f.read().strip()
            f.close()
            lg = f"{key} - {strline}"
            if strline == key.strip():
                return True
    return False

def is_id_setup(idi):
    path = os.path.join(files_path, idi + config.SET_PREFIX)
    return os.path.exists(path)

def path_value(idi):
    return os.path.join(files_path, idi + config.VAL_PREFIX)

def path_type(idi):
    return os.path.join(files_path, idi + config.TYPE_PREFIX)

def path_file(idi):
    return os.path.join(files_path, idi + config.FILE_PREFIX)

def write_data(idi, data, type):
    file_path = path_value(idi)
    f = open(file_path, "a", encoding='utf8')
    f.write(data)
    f.close()
    if type is not config.CONTENT_TYPE.FILE:
        file_path = path_type(idi)
        f = open(file_path, "w", encoding='utf8')
        f.write(type)
        f.close()

def write_file_data(idi, file):
    file_path = path_file(idi)
    file.save(file_path)
    file.close()

def load_file_data(idi):
    file_path = path_file(idi)
    f = open(file_path, "rb")
    return f.read()

def get_file_size(idi):
    file_path = path_file(idi)
    size = os.path.getsize(file_path)
    return humanize.naturalsize(size)

def load_data(idi):
    file_path = path_value(idi)
    f = open(file_path, "r")
    return f.read()

#log to file
def log_data(request, idi, data):
    log_path = os.path.join(files_path, idi + config.LOG_PREFIX)
    f = open(log_path, "a", encoding='utf8')
    f.write(data)
    f.close()