-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
231 lines (197 loc) · 12 KB
/
main.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
import logging
import tkinter
import customtkinter
import twain
from io import BytesIO
from PIL import Image, ImageTk
customtkinter.set_appearance_mode("System") # Modes: "System" (standard), "Dark", "Light"
customtkinter.set_default_color_theme("dark-blue") # Themes: "blue" (standard), "green", "dark-blue"
class App(customtkinter.CTk):
def __init__(self):
super().__init__()
self.sm = twain.SourceManager(self)
self.default_res = "600 dpi"
self.scanner_model = tkinter.StringVar(value="None")
self.src = None
self.scanned_image = ImageTk
self.acquire_requested = False
# Set font
customtkinter.FontManager.load_font("Noto_Sans.ttf")
# configure window
self.title(" CTkScannerUI - Scan utility")
self.geometry(f"{360}x{250}")
self.resizable(False, True)
# TODO can't find how to change the background to a darker one
self.configure(highlightbackground="#122F7B")
self.iconbitmap('icons/scanner.ico')
# self.overrideredirect(True)
# TODO probably useless, I should check if I can remove
# configure grid layout (1x1)
# a non-zero weight causes a row or column to grow if there's extra space
self.grid_columnconfigure(0, weight=0)
self.grid_rowconfigure(0, weight=0)
# create tabview
self.tabview = customtkinter.CTkTabview(self, width=300, fg_color="white")
self.tabview.grid(row=0, column=0, sticky="nsew")
self.tabview.pack(fill="both", expand=True, padx=5, pady=5)
self.tabview.add("Actions")
self.tabview.add("Settings")
self.tabview.add("Install")
# you can configure grid of individual tabs
self.tabview.tab("Actions").grid_columnconfigure(0, weight=0)
self.tabview.tab("Settings").grid_columnconfigure(0, weight=0)
# Main tab
self.scanner_model_label = customtkinter.CTkLabel(self.tabview.tab("Actions"),
text="Scanner:")
self.scanner_model_label.grid(row=0, column=0, padx=20, pady=(20, 10))
self.scanner_model_saved_label = customtkinter.CTkLabel(self.tabview.tab("Actions"),
justify="left",
textvariable=self.scanner_model,
)
self.scanner_model_saved_label.grid(row=0, column=1,
columnspan=2,
padx=20,
pady=(20, 10),
sticky="w") # to justify left
# Scan
self.scan_button_image = customtkinter.CTkImage(light_image=Image.open("icons/scanner.64x64.png"),
dark_image=Image.open("icons/scanner.64x64.png"),
size=(60, 60))
self.string_input_button = customtkinter.CTkButton(self.tabview.tab("Actions"),
text="",
command=self.scan, # self.open_input_dialog_event,
border_color="#FFBF63",
border_width=3,
fg_color="white",
text_color="black",
image=self.scan_button_image,
width=80, height=80)
self.string_input_button.grid(row=2, column=0, padx=15, pady=(10, 10))
self.scanner_button_label = customtkinter.CTkLabel(self.tabview.tab("Actions"),
font=("Nota Sans", 14),
text="Scan"
)
self.scanner_button_label.grid(row=3, column=0, padx=15) # , pady=(10, 10))
# Scan to PDF
self.scan_pdf_button_image = customtkinter.CTkImage(light_image=Image.open("icons/pdf.64x64.png"),
dark_image=Image.open("icons/scanner.64x64.png"),
size=(60, 60))
self.scan_pdf_button = customtkinter.CTkButton(self.tabview.tab("Actions"),
image=self.scan_pdf_button_image,
text="",
font=("Nota Sans", 20),
border_color="#FFBF63",
border_width=3,
fg_color="white",
text_color="black",
command=print("Hello"),
width=80,
height=80)
self.scan_pdf_button.grid(row=2, column=1, padx=15, pady=(10, 10))
self.scan_pdf_button_label = customtkinter.CTkLabel(self.tabview.tab("Actions"),
font=("Nota Sans", 14),
text="Scan to PDF"
)
self.scan_pdf_button_label.grid(row=3, column=1, padx=15) # , pady=(10, 10))
# Scan B&W
self.scan_bw_button_image = customtkinter.CTkImage(light_image=Image.open("icons/black-and-white.64x64.png"),
dark_image=Image.open("icons/black-and-white.64x64.png"),
size=(60, 60))
self.scan_button_bw = customtkinter.CTkButton(self.tabview.tab("Actions"),
text="",
image=self.scan_bw_button_image,
font=("Nota Sans", 20),
border_color="#FFBF63",
border_width=3,
fg_color="white",
text_color="black",
command=print("Hello B&W"),
width=80,
height=80)
self.scan_button_bw.grid(row=2, column=2, padx=15, pady=(10, 10))
self.scan_bw_button_label = customtkinter.CTkLabel(self.tabview.tab("Actions"),
font=("Nota Sans", 14),
text="Scan B&W"
)
self.scan_bw_button_label.grid(row=3, column=2, padx=15) # , pady=(10, 10))
# Tab Settings
self.label_tab_2 = customtkinter.CTkButton(self.tabview.tab("Settings"), text="Source...",
command=self.select_src)
self.label_tab_2.grid(row=0, column=0, padx=20, pady=20)
self.combobox_1 = customtkinter.CTkComboBox(self.tabview.tab("Settings"),
values=["My Scanner", "Value 2"])
self.combobox_1.grid(row=0, column=1, padx=20, pady=(10, 10))
self.res_label = customtkinter.CTkLabel(self.tabview.tab("Settings"),
text="Resolution:")
self.res_label.grid(row=1, column=0, padx=20, pady=(20, 10))
# TODO rename this properly
self.optionmenu_1 = customtkinter.CTkOptionMenu(self.tabview.tab("Settings"),
dynamic_resizing=True,
values=["600 dpi", "300 dpi", "Value Long Long Long"])
self.optionmenu_1.grid(row=1, column=1, padx=20, pady=(20, 10))
self.check1_label = customtkinter.CTkButton(self.tabview.tab("Install"),
text="Check 32 bit",
command="")
self.check1_label.grid(row=0, column=0, padx=20, pady=20)
self.check2_label = customtkinter.CTkButton(self.tabview.tab("Install"),
text="Check DLLs",
command="")
self.check2_label.grid(row=1, column=0, padx=20, pady=20)
# set default values
self.optionmenu_1.set(self.default_res)
self.combobox_1.set("My Super long Scanner")
def select_src(self):
# this will show UI to allow user to select source
self.src = self.sm.open_source()
if not self.src:
logging.error("No source selected")
else:
self.scanner_model.set(self.src.name)
logging.info(f"src = {self.src.name}")
logging.info(f"scanner_model = {self.scanner_model}")
def open_input_dialog_event(self):
dialog = customtkinter.CTkInputDialog(text="Type in a number:", title="CTkInputDialog")
print("CTkInputDialog:", dialog.get_input())
def change_appearance_mode_event(self, new_appearance_mode: str):
customtkinter.set_appearance_mode(new_appearance_mode)
def scan(self):
logging.info("Scanning...")
# this will show UI to allow user to select source
if self.src:
logging.info("Test Acquire request param")
if not self.acquire_requested:
logging.info("Acquire request = False, requesting...")
# don't touch the visibility params, UI freezes and scanner becomes weird
self.src.request_acquire(show_ui=False, modal_ui=False)
self.acquire_requested = True
logging.info("Acquire request is now True, ready to go!")
logging.info("Acquire request = True")
(handle, remaining_count) = self.src.xfer_image_natively()
logging.info(f"Remaining count = {remaining_count}")
logging.info("Acquire done")
bmp_bytes = twain.dib_to_bm_file(handle)
img = Image.open(BytesIO(bmp_bytes), formats=["bmp"])
width, height = img.size
factor = 600.0 / width
img.save("test.bmp")
logging.info("Image saved to disk")
# self.src.close()
# TODO extract class
secondary_window = (customtkinter.CTkToplevel(master=self))
secondary_window.title("Scan utility - scanned result")
# see https://github.com/TomSchimansky/CustomTkinter/issues/1486
secondary_window.attributes("-topmost", True)
# see https://github.com/TomSchimansky/CustomTkinter/issues/2302#issuecomment-1991632556
self.after(1000, lambda: secondary_window.iconbitmap("icons/scanner.ico"))
screen_width = self.winfo_screenwidth()
screen_height = self.winfo_screenheight()
secondary_window.config(width=screen_width / 2, height=screen_height / 2)
self.scanned_image = customtkinter.CTkImage(img, size=(int(width * factor), int(height * factor)))
customtkinter.CTkLabel(secondary_window, text="", image=self.scanned_image).pack(side="left", fill="both", expand=1)
else:
print("User clicked cancel")
if __name__ == "__main__":
logging.basicConfig(level=logging.DEBUG)
app = App()
scanner_list = twain.SourceManager(app)
app.mainloop()