-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.pyw
800 lines (626 loc) · 30.8 KB
/
app.pyw
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
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
# ======================================================================================================================
import pytesseract
import os
import logging
import pyperclip
import datetime
import requests
import urllib.request
import subprocess
import markdown
import tkinter as tk
import ttkbootstrap as ttk
from tkinter import messagebox, filedialog
from PIL import ImageGrab, ImageTk, Image, ImageDraw
from PIL.PngImagePlugin import PngImageFile
from PIL.Image import Resampling
from time import time_ns
from pyzbar.pyzbar import decode
from dotenv import load_dotenv, set_key
from packaging import version
from tkhtmlview import HTMLScrolledText
# ---------------------------------------------
import clipbarcode.database as database
from clipbarcode.datetime_tools import timens_to_datetime
from clipbarcode.boleto import new_boleto, BoletoInvalidoException
from clipbarcode.version import __version__
# ======================================================================================================================
HISTORY_PATH = "./history"
CUR_VERSION = version.parse(__version__)
LABEL_FONTNAME = "Arial"
# ======================================================================================================================
class NoImageException(Exception): pass
class LeituraFalhaException(Exception):
def __init__(self, message):
self.message = message
# ======================================================================================================================
class MainWindow:
def __init__(self, themename="darkly", *args, **kwargs):
super().__init__(*args, **kwargs)
self.root = ttk.Window(alpha=0.99, iconphoto=None)
self.root.title(f"Clip Barcode {CUR_VERSION}")
self.root.geometry("1280x720")
self.root.iconbitmap("icon.ico")
self.root.place_window_center()
self.root.position_center()
self.last_width = 0
self.last_height = 0
self._change_theme(themename)
# -------------------------------------
menubar = ttk.Menu()
self.root.config(menu=menubar)
# Themes:
theme_menu = ttk.Menu()
themes = ["darkly", "yeti", "superhero", "vapor", "cosmo", "journal", "sandstone"]
avaiable_themes = [t for t in self.root.style.theme_names() if t in themes]
for theme_name in sorted(avaiable_themes):
theme_menu.add_command(
label = theme_name,
command = lambda t=theme_name: self._change_theme(t),
)
menubar.add_cascade (
label = "Temas",
menu = theme_menu,
)
# Ajuda:
help_menu = ttk.Menu()
help_menu.add_command(
label = "Ajuda",
command = self._on_ajuda_click,
)
help_menu.add_separator()
help_menu.add_command(
label = "Sobre",
command = self._on_sobre_click,
)
menubar.add_cascade (
label = "Ajuda",
menu = help_menu,
)
# -------------------------------------
# Frames:
self.btn_ler_print = ttk.Button(self.root, text="Ler Print", command=self._on_ler_print_click)
self.btn_ler_print.grid(row=0, column=0, padx=(10, 5), pady=(20,10), sticky="nsew")
self.f1 = ttk.Frame(self.root, )
self.f1.grid(row=1, column=0, padx=(10, 5), pady=(10, 10), sticky="nswe")
self.root.columnconfigure(index=0, weight=0, minsize=410)
self.f2 = ttk.Frame(self.root, )
self.f2.grid(row=1, column=1, padx=(5, 10), pady=(0, 10), sticky="nswe")
self.root.columnconfigure(index=1, weight=1)
self.root.rowconfigure(index=1, weight=1)
# -------------------------------------
# List Frame:
self.listbox = tk.Listbox(self.f1, font=(LABEL_FONTNAME, 16), selectmode="SINGLE", activestyle=tk.NONE)
self.listbox.bind('<<ListboxSelect>>', self._on_item_selected)
self.listbox.bind("<Down>", self._on_arrow_down_click)
self.listbox.bind("<Up>", self._on_arrow_up_click)
self.listbox.bind("<Delete>", self._on_del_click)
self.listbox.pack(fill="both", expand=True)
self.leituras = None
self.cur_index = None
self._fill_list()
# -------------------------------------
# Detail Frame:
lf_canvas = ttk.Labelframe(self.f2, text="Imagem", bootstyle='primary')
lf_canvas.grid(row=0, column=0, sticky="nsew", pady=(0, 5))
self.canvas = ttk.Label(lf_canvas)
self.canvas.pack(fill='both', expand='true', padx=10, pady=10)
# Data:
self.lbl_date = tk.StringVar()
tk.Label(self.f2, textvariable=self.lbl_date, font=(LABEL_FONTNAME, 16),).grid(row=1, column=0, sticky="nswe", pady=(5, 5))
# Tipo:
self.var_tipo = tk.StringVar()
tk.Label(self.f2, textvariable=self.var_tipo, font=(LABEL_FONTNAME, 16),).grid(row=2, column=0, sticky="nsew", pady=(5, 5))
# Leitura:
lf_leitura = ttk.Labelframe(self.f2, text="Leitura", bootstyle='primary')
lf_leitura.grid(row=3, column=0, sticky="nsew", pady=(5, 5))
self.var_leitura = tk.StringVar()
self.entry_leitura = ttk.Entry(lf_leitura, font=(LABEL_FONTNAME, 16), state='readonly', textvariable=self.var_leitura)
self.entry_leitura.grid(row=0, column=0, sticky="we", pady=10, padx=(10, 5))
self.btn_copiar_leitura = ttk.Button(lf_leitura, text="Copiar", command=self.on_copiar_leitura_click, width=7)
self.btn_copiar_leitura.grid(row=0, column=1, sticky="ew", pady=10, padx=(5,10))
lf_leitura.columnconfigure(0, weight=1)
# Descrição:
lf_descricao = ttk.Labelframe(self.f2, text="Descrição", bootstyle='primary')
lf_descricao.grid(row=4, column=0, sticky="nsew", pady=(5,0))
self.var_descricao = tk.StringVar()
self.entry_descricao = ttk.Entry(lf_descricao, font=(LABEL_FONTNAME, 16), state='readonly', textvariable=self.var_descricao)
self.entry_descricao.grid(row=0, column=0, sticky="we", pady=10, padx=(10, 5))
self.entry_descricao.bind("<Return>", self._salvar_descricao)
self.btn_descricao = ttk.Button(lf_descricao, text="Editar", command=self._on_btn_descricao_click, width=7)
self.btn_descricao.grid(row=0, column=1, sticky="ew", pady=10, padx=(5,10))
lf_descricao.columnconfigure(0, weight=1)
# Frame:
self.f2.rowconfigure(index=0, weight=1)
self.f2.columnconfigure(index=0, weight=1)
# -------------------------------------
self.root.bind('<Configure>', self._on_configure_callback)
# -------------------------------------
self.cur_img = None
self.cur_img_resized = None
self.photoimage = None
self._hot_read()
# -------------------------------------
def _change_theme(self, themename, *args, **kwargs):
self.root.style.theme_use(themename)
update_users_preferences(themename)
self.root.style.configure (
"TButton",
font = (LABEL_FONTNAME, 12),
)
self.root.style.configure (
'TLabelframe.Label',
font = (LABEL_FONTNAME, 10),
)
def _on_sobre_click(self, *args, **kwargs):
SobreToplevel(title="Sobre", resizable=(False, False), alpha=0.99, topmost=True)
def _on_ajuda_click(self, *args, **kwargs):
AjudaToplevel(title="Ajuda", minsize=(600,600), alpha=0.99, topmost=True)
def _fill_list(self, *args, **kwargs):
"""Método responsável por atualiza a lista com as leituras e atualiza o listbox.
"""
self.leituras = database.get_leituras()
self.listbox.delete(0, tk.END)
for leitura in self.leituras:
self.listbox.insert(tk.END, str(leitura))
def _ler_print(self):
"""Método responsáel por realizar a leitura do print, verificar se é duplicado, e etc.
"""
nova_leitura, img = realizar_leitura()
duplicado, leitura = verifica_se_duplicado(nova_leitura)
index = 0
if duplicado:
salvar = messagebox.askyesno(title="Duplicado", message=f"Código de barras já lido.\nDeseja salvar mesmo assim ?")
if salvar:
salvar_leitura(nova_leitura, img)
else:
index = self.leituras.index(leitura)
else:
salvar_leitura(nova_leitura, img)
self._fill_list()
self.listbox.selection_clear(0, tk.END)
self.listbox.selection_set(index)
self.listbox.event_generate("<<ListboxSelect>>") # Simulando um "item selecionado"
def _hot_read(self):
"""Método responsável por tentar realizar uma leitura logo na inicialização da aplicação.
"""
try:
self._ler_print()
except Exception:
pass
self.listbox.focus()
def _on_ler_print_click(self, *args, **kwargs):
"""Método responsável por lidar com o evento do Botão "Ler Print" ao ser pressionado.
"""
try:
self._ler_print()
except NoImageException:
messagebox.showwarning("Sem Imagem", "Tire um print antes")
except LeituraFalhaException as e:
messagebox.showwarning("Ops", e.message)
self.listbox.focus()
def _on_item_selected(self, *args, **kwargs):
"""Método responsável por lidar com os eventos de um item selecionado na Listbox.
Esse evento também é chamado caso o TkInter entenda que nenhum item foi selecionado.
"""
cur_selection = self.listbox.curselection()
if cur_selection:
selected_index = cur_selection[0]
self.cur_index = selected_index
leitura = self.leituras[self.cur_index]
self.update_frame_detail(leitura)
self.btn_descricao.configure(text="Editar")
self.entry_descricao.config(state="readonly")
def _on_arrow_up_click(self, *args, **kwargs):
"""Método responsável por lidar com o evento do botão "Setinha Para Cima" pressionado no Lisbox
"""
cur_selection = self.listbox.curselection()
if cur_selection:
self.selected_index = cur_selection[0]
if self.selected_index > 0:
self.listbox.select_clear(self.selected_index)
self.selected_index -= 1
self.listbox.select_set(self.selected_index)
self.listbox.event_generate("<<ListboxSelect>>") # Simulando um "item selecionado"
def _on_arrow_down_click(self, *args, **kwargs):
"""Método responsável por lidar com o evento do botão "Setinha Para Baixo" pressionado no Lisbox
"""
cur_selection = self.listbox.curselection()
if cur_selection:
self.selected_index = cur_selection[0]
if self.selected_index < self.listbox.size() - 1:
self.listbox.select_clear(self.selected_index)
self.selected_index += 1
self.listbox.select_set(self.selected_index)
self.listbox.event_generate("<<ListboxSelect>>") # Simulando um "item selecionado"
def _on_del_click(self, *args, **kwargs):
"""Método responsável por lidar com o evento do botão "Delete" pressionado no Lisbox
"""
ans = messagebox.askyesno(title="Excluir", message="Excluir registro ?")
if ans:
cur_selection = self.listbox.curselection()
if cur_selection:
selected_index = cur_selection[0]
leitura = self.leituras[selected_index]
delete_leitura(leitura)
self._fill_list()
self.clear()
else:
messagebox.showerror("Erro", "Não foi possível excluir o registro")
def on_copiar_leitura_click(self, *args, **kwargs):
"""Método responsável por lidar com o evento do botão "Copiar" pressionado.
No caso, envia para a Área de Transferência o que tiver no widget Leitura.
"""
pyperclip.copy(self.var_leitura.get())
self.btn_copiar_leitura.config(text="Copiado", bootstyle="success")
self.root.after(1000, lambda : self.btn_copiar_leitura.config(text="Copiar", bootstyle="default"))
def _on_btn_descricao_click(self, *args, **kwargs):
"""
Método responsável por lidar com o evento do botão "Editar/Salvar" pressionado.
O mesmo botão tem 2 funções:
1) Quando está com "Editar" e o usuário clicou, iremos habilitar os campos para que o usuário possa editar o EditText
2) Quando está com "Salvar", iremos salvar as informações e desabilitar o EditText.
"""
match (self.btn_descricao.cget("text")):
case "Editar":
self.btn_descricao.configure(text="Salvar", bootstyle="success")
self.entry_descricao.config(state="normal")
case "Salvar":
self._salvar_descricao()
def _on_configure_callback(self, event, *args, **kwargs):
"""Callback para qualquer evento de configuração da Janela.
Utilizaremos para saber se usuário alterou o tamanho da janela e assim recalcularmos o tamanho da imagem dentro do canvas.
Args:
event ():
"""
if event.widget == self.canvas: # Houve uma alteração no Canvas:
if abs(event.width - self.last_width) > 50 or abs(event.height - self.last_height) > 50: # Foi uma alteração de tamanho (usuário aumentou/diminui a janela ou simplesmente foi a inicialiação do GUI)
if self.last_width != 0 and self.last_height != 0: # Ignorando a inicialização do GUI
try:
res_img = self.resize_image_to_canvas(self.cur_img)
self.update_canvas(img_resized=res_img)
except NoImageException:
pass
self.last_width = event.width
self.last_height = event.height
def _salvar_descricao(self, *args, **kwargs):
self.btn_descricao.configure(text="Editar", bootstyle="default")
self.entry_descricao.config(state="readonly")
cur_selection = self.listbox.curselection()
if cur_selection:
self.cur_index = cur_selection[0]
leitura = self.leituras[self.cur_index]
update_value(leitura, descricao=self.var_descricao.get())
self._fill_list()
# TODO: Selecionar algum item da listbox (o anterior, o próximo, o primeiro...)
def mainloop(self, *args, **kwargs):
"""Mainloop do TkInter
"""
self.root.mainloop()
def update_frame_detail(self, leitura: database.Leitura, *args, **kwargs):
"""Atualiza todos os widgets presentes no frame "Detail"
Args:
leitura(database.Leitura): Instância da class Leitura que contém todos os dados necessários para atualizar os widgets.
"""
self.update_canvas(filename=os.path.join(HISTORY_PATH, f"{leitura.mili}.png"))
self.update_widget_data(leitura.data)
self.update_tipo(leitura.get_type_display())
self.update_widget_leitura(leitura.cod_conv)
self.update_widget_descricao(leitura.descricao)
def update_tipo(self, tipo:str, *args, **kwargs):
"""Insere um valor no widget "Tipo".
Args:
tipo (str): Valor a ser inserido no widget
"""
self.var_tipo.set(tipo)
def update_widget_data(self, value:datetime.datetime=None, *args, **kwargs):
"""Insere um valor no widget "Date".
Args:
new_text (str): Valor a ser inserido no widget
"""
if value:
self.lbl_date.set(value.strftime("%d/%m/%Y %H:%M:%S"))
else:
self.lbl_date.set('')
def update_widget_leitura(self, new_text:str, *args, **kwargs):
"""Insere um valor no widget "Leitura".
Args:
new_text (str): Valor a ser inserido no widget
"""
self.var_leitura.set(new_text)
def update_widget_descricao(self, new_text:str, *args, **kwargs):
"""Insere um valor no widget "Descrição".
Args:
new_text (str): Valor a ser inserido no widget
"""
self.var_descricao.set(new_text if new_text else "")
def resize_image_to_canvas(self, img:PngImageFile, *args, **kwargs) -> Image.Image:
"""Realiza o redimensionamento de uma imagem, mantendo as suas proporções, conforme o tamanho do Canvas.
Args:
img (PngImageFile): Imagem a ser redimensionada
Raises:
NoImageException: Caso não haja uma imagem
Returns:
Image.Image: Imagem redimensionada.
"""
if img:
cur_width, cur_height = img.size
ratio = min(self.canvas.winfo_width() / cur_width, self.canvas.winfo_height() / cur_height)
new_width = int(cur_width * ratio)
new_height = int(cur_height * ratio)
return img.resize((new_width, new_height), Resampling.LANCZOS)
else:
raise NoImageException
def update_canvas(self, filename:str=None, img_resized:Image.Image=None, *args, **kwargs):
"""Insere uma nova imagem ao canvas:
Ou através do nome do arquivo da imagem
Ou através de um objeto Image.
Args:
filename (str, optional): Nome do arquivo de imagem da leitura salva em history/. Defaults to None.
img_resized (Image.Image, optional): Imagem já redimensionada. Defaults to None.
"""
self.canvas.update()
if filename:
try:
self.cur_img = Image.open(filename)
self.cur_img_resized = self.resize_image_to_canvas(self.cur_img) # Vamos redimensionar a imagem e deixar a função calcular automaticamente o tamanho
self.photoimage = ImageTk.PhotoImage(self.cur_img_resized)
self.canvas["image"] = self.photoimage
except (FileNotFoundError, FileExistsError, NoImageException):
logging.error("Imagem não encontrada")
messagebox.showerror("Imagem", "Imagem não encontrada")
except ValueError:
pass
elif img_resized:
self.cur_img_resized = img_resized
self.photoimage = ImageTk.PhotoImage(self.cur_img_resized)
self.canvas["image"] = self.photoimage
else:
self.canvas["image"] = ""
def clear(self, *args, **kwargs):
"""Limpa todos os widgets.
"""
self.listbox.selection_clear(0, tk.END)
self.update_canvas()
self.update_widget_data()
self.update_tipo('')
self.update_widget_leitura('')
class BaseToplevel(ttk.Toplevel):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.iconbitmap("icon.ico")
self.grab_set()
self.position_center()
self.place_window_center()
class SobreToplevel(BaseToplevel):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
f = tk.Frame(self,)
f.pack(expand=True, fill='both', pady=10, padx=10)
tk.Label(f, text="ClipBarcode", font=(LABEL_FONTNAME, 16)) .pack(expand=False, fill="both",)
tk.Label(f, text=f"Versão {CUR_VERSION}") .pack(expand=False, fill="both",)
tk.Label(f, text="Vinícius Costa") .pack(expand=False, fill="both",)
tk.Label(f, text="https://github.com/viniciusccosta/ClipBarcode").pack(expand=False, fill="both",)
class AjudaToplevel(BaseToplevel):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
html_content = self._get_html_content()
self.html_label = HTMLScrolledText(self)
self.html_label.pack(expand=True, fill='both')
self.html_label.set_html(html_content)
def _get_html_content(self):
with open('README.md', 'r', encoding='utf8') as file:
markdown_content = file.read()
return markdown.markdown(markdown_content)
# ======================================================================================================================
def initial_config(*args, **kwargs):
"""Realiza as configurações inicias da aplicação.
- Logging
- SQLite
- Diretório de Imagens
- Tesseract
- Variáveis de Ambiente
"""
# ------------------------------------------
load_dotenv()
# ------------------------------------------
# Logging:
logging.basicConfig(
stream = open(f'.log', 'a', encoding='utf-8') ,
level = logging.INFO, datefmt='%Y-%m-%d %H:%M:%S' ,
format = '%(asctime)s %(levelname)-8s %(message)s' ,
)
# ------------------------------------------
# Banco de Dados:
json_path = os.path.join(HISTORY_PATH, "results.json")
database.create_tables()
if os.path.exists(json_path):
try:
database.from_json_to_sqlite(json_path)
os.remove(json_path) # Excluindo json após transferir as leituras para DB
except Exception as e:
print(e)
logging.error(e)
# ------------------------------------------
# Diretório de Imagens:
if not os.path.exists(HISTORY_PATH):
os.mkdir(HISTORY_PATH)
logging.info("Pasta HISTORY criada com sucesso")
# ------------------------------------------
# Tesseract:
while True:
try:
pytesseract.pytesseract.tesseract_cmd = os.environ.get('TESSERACT_CMD', r'C:/Program Files/Tesseract-OCR/tesseract.exe')
pytesseract.get_languages() # Apenas para testar se o Tesseract está no PATH
break
except pytesseract.pytesseract.TesseractNotFoundError:
pass
logging.error("Tesseract não encontrado")
messagebox.showerror("Erro", "Tesseract não encontrado!")
tesseract_path = filedialog.askopenfilename(title="Onde está tesseract.exe ?")
logging.info(f"Usuário informou |{tesseract_path}| como path para o Tesseract")
if tesseract_path:
set_key('.env', 'TESSERACT_CMD', tesseract_path)
logging.info("Path do Tesseract salvo com sucesso")
else:
logging.error("Encerrando programa pois o usuário não informou um path!")
exit(1)
def verificar_versao(*args, **kwargs):
try:
response = requests.get("https://api.github.com/repos/viniciusccosta/clipbarcode/releases", timeout=(1.0, 2.0))
match response.status_code:
case 200:
json_data = response.json()
if len(json_data) > 0:
release = json_data[0]
latest_version = version.parse(release["tag_name"])
assets = release["assets"]
logging.info(f"Versão mais recente: {latest_version}")
if latest_version > CUR_VERSION and len(assets) > 0:
logging.warning("Versão instalada não é a mais recente")
atualizar = messagebox.askyesno("Atualização", "Uma nova versão está disponível, deseja baixá-la?")
if atualizar:
logging.info("Usuário decidiu instalar a nova versão")
for asset in assets:
url = asset['browser_download_url']
filename = asset['name']
if url.endswith(".exe"):
urllib.request.urlretrieve(url, filename)
subprocess.run([filename])
messagebox.showinfo("Fim da Atualização", "Abra o programa novamente!")
exit(0)
case _:
pass
except Exception:
pass
def salvar_leitura(leitura: database.Leitura, img:PngImageFile, *args, **kwargs):
"""Salva a leitura no banco de dados juntamente com o print.
Args:
result (database.Leitura): Instância da classe Leitura com os dados da leitura
img (PngImageFile): Print em si
"""
# Salvando a imagem:
img.save(os.path.join(HISTORY_PATH, f"{leitura.mili}.png"))
logging.info(f"{leitura.mili}.png salvo com sucesso na pasta History")
# Incluíndo a leitura no arquivo de resultados:
database.create_leitura(leitura)
def update_value(leitura: database.Leitura, *args, **kwargs):
"""Faz uma requisição ao banco de dados para atualizar determinada leitura.
Args:
leitura (database.Leitura):
**kwargs (dict): Dicionário com os campos e valores a serem atualizados
"""
database.update_leitura(leitura.id, **kwargs)
def delete_leitura(leitura: database.Leitura, *args, **kwargs):
"""Faz a exclusão do print do diretório de imagens e faz uma requisição ao banco de dados para a exclusão da leitura.
Args:
leitura (database.Leitura):
"""
os.remove(os.path.join(HISTORY_PATH, f"{leitura.mili}.png"))
database.delete_leitura(leitura)
def realizar_leitura(*args, **kwargs):
"""Faz a leitura da imagem que estiver na Área de Transferência.
- Procura inicialmente por um Códigos de Barra
- Caso não identifique nenhum código de barras, realiza OCR
Raises:
NoImageException: Caso não seja encontrado nenhuma imagem na Área de Transferência
LeituraFalhaException: Caso qualquer problema ocorra ao realizar a leitura
"""
# -----------------------------------------------------------
timens = time_ns()
agora = timens_to_datetime(timens)
# -----------------------------------------------------------
img = ImageGrab.grabclipboard()
# -----------------------------------------------------------
# Código de Barras:
try:
results = decode(img)
except (TypeError, Exception):
raise NoImageException
if len(results) >= 1:
logging.info("Código de barras encontrado")
if len(results) > 1:
texto = "O seu print só deve conter apenas 1 código de barras"
logging.error(texto)
raise LeituraFalhaException(texto)
d = results[0]
text = d.data.decode("utf-8")
match(d.type):
case "I25": # Boletos de Cobraça e Arrecadação
logging.debug("Código de barrras do tipo I25 (boletos de cobrança e arrecadação)")
try:
boleto = new_boleto(cod_barras=text)
cod_conv = boleto.linha_digitavel
m_type = 1
except BoletoInvalidoException:
texto = f"Boleto Inválido: |{text}|"
logging.error(texto)
raise LeituraFalhaException(texto)
case "CODE128": # Código de Nota Fiscal
logging.debug("Código de barras do tipo CODE128 (notas fiscais)")
cod_conv = text
m_type = 2
case "QRCODE": # QRCode
logging.debug("Código de barras do tipo QRCODE")
cod_conv = text
m_type = 3
case _: # Nenhum dos tipos anteriores
texto = f"O código de barras do tipo {d.type} não é suportado"
logging.warning(texto)
raise LeituraFalhaException(texto)
x, y, wi, h = d.rect.left, d.rect.top, d.rect.width, d.rect.height
imgdraw = ImageDraw.Draw(img)
imgdraw.rectangle(xy=(x, y, x+wi, y+h), outline="#FF0000", width=2,)
logging.debug(f"Imagem encontrada em ({x},{y}) -> ({x+wi},{y+h})")
leitura = database.Leitura (
mili = f"{timens}",
data = agora,
type = m_type,
cod_lido = text,
cod_conv = cod_conv,
)
return(leitura, img)
# -----------------------------------------------------------
# OCR:
else:
logging.info("Nenhum código de barras encontrado, programa tentará fazer OCR.")
try:
text = pytesseract.image_to_string(img, lang="por",).strip("\n") # TODO: Tesseract está tendo dificuldades em ler números com mais de dois 0 seguidos
except TypeError:
raise NoImageException
if len(text) <= 0:
texto = "OCR não encontrou nada"
logging.warning(texto)
raise LeituraFalhaException(texto)
logging.debug(f"OCR realizado com sucesso |{text}|")
boleto = new_boleto(linha_digitavel=text)
leitura = database.Leitura(
mili = f"{timens}",
data = agora,
type = 0,
cod_lido = text,
cod_conv = boleto.linha_digitavel if boleto else text
)
return(leitura, img)
def verifica_se_duplicado(leitura: database.Leitura, *args, **kwargs):
"""Verifica se a linha digitável já foi inserida no banco de dados"""
leitura = database.get_leitura_por_cod_lido(leitura.cod_lido)
if leitura:
return True, leitura
return False, None
def get_users_preferences(*args, **kwargs):
"""Obter o nome da temática escolhida pelo usuário."""
preferencia = database.get_preferencia()
return preferencia
def update_users_preferences(themename, *args, **kwargs):
database.update_preferencia(themename=themename)
# ======================================================================================================================
if __name__ == '__main__':
# -------------------------------------
initial_config()
# -------------------------------------
verificar_versao()
# -------------------------------------
preferences = get_users_preferences()
w = MainWindow(themename=preferences.themename)
w.mainloop()