-
Notifications
You must be signed in to change notification settings - Fork 0
/
database.py
291 lines (259 loc) · 9.76 KB
/
database.py
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
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
import sqlite3
class Database:
def __init__(self, db_file):
self.con = sqlite3.connect(db_file, check_same_thread=False)
self.cur = self.con.cursor()
def create_tables(self):
# causers = [(1, 'КО'), (2, 'Производство'), (3, 'Снабжение')]
with self.con:
self.cur.executescript('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
external_id INTEGER,
first_name TEXT,
last_name TEXT,
work_email TEXT);
CREATE TABLE IF NOT EXISTS trailers (
id INTEGER PRIMARY KEY,
designation TEXT UNIQUE);
CREATE TABLE IF NOT EXISTS causers (
id INTEGER PRIMARY KEY,
name TEXT);
CREATE TABLE IF NOT EXISTS troubles (
id INTEGER PRIMARY KEY,
date TEXT,
order_num TEXT,
problem TEXT,
document TEXT,
status INTEGER,
trailer_id INTEGER NOT NULL,
causer_id INTEGER,
user_id INTEGER NOT NULL,
FOREIGN KEY(trailer_id) REFERENCES trailers(id),
FOREIGN KEY(user_id) REFERENCES users(id),
FOREIGN KEY(causer_id) REFERENCES causers(id));
''')
# self.cur.executemany(
# 'INSERT INTO causers VALUES(?, ?);',
# (causers))
# Создание пользователя
def create_user(self, data):
with self.con:
self.cur.execute('''
INSERT INTO users(
external_id,
first_name,
last_name)
VALUES(?, ?, ?);''', (data))
# Поиск пользователя
def check_user(self, user_id):
with self.con:
result = self.cur.execute(
'SELECT * FROM users WHERE external_id = ?;',
(user_id,)).fetchall()
return bool(len(result))
# Поиск почты
def check_email(self, user_id):
with self.con:
result = self.cur.execute(
'SELECT work_email FROM users WHERE external_id = ?;',
(user_id,)).fetchall()[0]
return result[0]
# Создание почты
def create_email(self, email, user_id):
with self.con:
self.cur.execute(
'UPDATE users SET work_email=? WHERE external_id=?',
(email, user_id))
# Список прицепов
def read_trailers(self):
with self.con:
list_trailers = self.cur.execute(
'SELECT id, designation FROM trailers'
).fetchall()
return list_trailers
# Поиск прицепа по id
def search_trailer(self, id):
with self.con:
select_trailer = self.cur.execute(
'SELECT designation FROM trailers WHERE id = ?;',
(id,)).fetchall()
return select_trailer[0]
# Поиск прицепа по designation
def search_trailer_designation(self, designation):
with self.con:
select_trailer = self.cur.execute(
'SELECT id FROM trailers WHERE designation = ?;',
(designation,)).fetchall()
return select_trailer[0]
# Поиск виновника id
def search_causer_id(self, name):
with self.con:
causer_id = self.cur.execute(
'SELECT id FROM causers WHERE name = ?;',
(name,)).fetchall()
return causer_id[0]
# Поиск виновника name
def search_causer_name(self, id):
with self.con:
causer_name = self.cur.execute(
'SELECT name FROM causers WHERE id = ?;',
(id,)).fetchall()
return causer_name[0]
# Поиск пользователя
def search_user(self, external_id):
with self.con:
result = self.cur.execute(
'SELECT id FROM users WHERE external_id = ?;',
(external_id,)).fetchall()
return result[0]
# Создание записи в таблице проблем
def create_trouble(self, create_list):
with self.con:
self.cur.execute('''
INSERT INTO troubles(
date,
order_num,
problem,
document,
status,
trailer_id,
causer_id,
user_id)
VALUES(?, ?, ?, ?, ?, ?, ?, ?);''', (create_list))
# Создание записи в таблице прицепов
def create_trailer(self, designation):
with self.con:
self.cur.execute(
'INSERT INTO trailers(designation) VALUES(?)',
(designation,))
def search_designation_trailer(self, designation):
designation = ('%' + designation + '%')
with self.con:
result = self.cur.execute('''
SELECT
id,
designation
FROM trailers
WHERE designation LIKE ?;''', (designation,)).fetchall()
return result
# Функции для скрпита парсинга таблиц экселя
# ===================================================================================
def script_trailer_id(self, designation):
with self.con:
result = self.cur.execute(
'SELECT * FROM trailers WHERE designation = ?;',
(designation,)).fetchall()
return result[0]
def script_causer_id(self, name):
with self.con:
result = self.cur.execute(
'SELECT * FROM causers WHERE name = ?;',
(name,)).fetchall()
if result == []:
result = ' '
else:
result = result[0]
return result
def script_user_external_id(self, last_name):
with self.con:
result = self.cur.execute('''
SELECT
external_id,
last_name
FROM users
WHERE last_name = ?;''', (last_name,)).fetchall()
return result[0]
# ===================================================================================
def search_by_trailer_in_troubles(self, id):
with self.con:
result = self.cur.execute(
'SELECT * FROM troubles WHERE trailer_id = ? ORDER BY date;',
(id,)).fetchall()
return result
# Поиск пользователя, фамилия
def search_user_last_name(self, external_id):
with self.con:
result = self.cur.execute(
'SELECT last_name FROM users WHERE external_id = ?;',
(external_id,)).fetchall()
return result[0]
def search_user_first_name(self, external_id):
with self.con:
result = self.cur.execute(
'SELECT first_name FROM users WHERE external_id = ?;',
(external_id,)).fetchall()
return result[0]
# Поиск по дате
def search_by_date_in_troubles(self, date):
date = ('%' + date + '%')
with self.con:
result = self.cur.execute(
'SELECT * FROM troubles WHERE date LIKE ? ORDER BY date;',
(date,)).fetchall()
return result
# Поиск по периоду
def search_by_period_in_troubles(self, start_date, end_date):
# date = ('%' + date + '%')
with self.con:
result = self.cur.execute('''
SELECT
*
FROM troubles
WHERE date >= ? AND date <= ?
ORDER BY date;''', (start_date, end_date,)).fetchall()
return result
# Количество записей
def count_troubles(self):
with self.con:
result = self.cur.execute(
'SELECT COUNT(*) FROM troubles;').fetchall()
return result[0]
# Макс дата
def min_date(self):
with self.con:
result = self.cur.execute(
'SELECT MIN(date) FROM troubles;').fetchall()
return result[0]
# Мин дата
def max_date(self):
with self.con:
result = self.cur.execute(
'SELECT MAX(date) FROM troubles;').fetchall()
return result[0]
# Поиск по id записи
def search_by_id_in_troubles(self, id):
with self.con:
result = self.cur.execute(
'SELECT * FROM troubles WHERE id = ?;',
(id,)).fetchall()
return result[0]
# Поиск пользователя, фамилия
def search_user_emal(self, external_id):
with self.con:
result = self.cur.execute(
'SELECT work_email FROM users WHERE external_id = ?;',
(external_id,)).fetchall()
return result[0]
# Редактирование записи
def update_trouble(self, update_list, id):
with self.con:
self.cur.execute('''
UPDATE troubles
SET date = ?,
order_num = ?,
problem = ?,
document = ?,
status = ?,
trailer_id = ?,
causer_id = ?,
user_id = ?
WHERE id = ?;''', (*update_list, id))
return True
# Удаление записи
def delete_trouble(self, id):
with self.con:
self.cur.execute(
'DELETE FROM troubles WHERE id=?;',
(id,))
return True