def Tabella(master):
if len(GLOBE.titolo) == 0:
return
righe = len(list(GLOBE.titolo.keys()))
colonne = len(list(GLOBE.titolo.get(list(GLOBE.titolo.keys())[0]).keys())) - 1
tabella = Sheet(master, data = [[f"Row {r}, Column {c}" for c in range(colonne)] for r in range(righe)], height = 400, width = 1550)
tabella.headers((f"Header{c}" for c in range(colonne)))
key_titolo = list(GLOBE.titolo.keys())
key_isin = GLOBE.titolo.get(key_titolo[0]).keys()
header = list(key_isin)
index = 0
for i in header: #mi crea gli header della tabella
tabella.headers(i.capitalize(), index)
# tabella.column_width(column = index, width = 150)
index = index + 1
if index == colonne:
break
for j in range(len(key_titolo)): #mi riempe le celle della tabella
for k in range(len(header) - 1):
tabella.set_cell_data(j, k, value = GLOBE.titolo.get(key_titolo[j]).get(header[k]))
tabella.set_all_cell_sizes_to_text()
tabella.grid(row = 0, column = 0, sticky = "nswe")
def auto_organize_dialog(self, moved_files): # noqa
dialog = Dialogs.create_dialog(
on_close_callback=self.on_auto_organize_dialog_close,
title='Auto Organize')
label = tk.Label(
dialog,
text='Following files were moved / renamed -',
)
label.grid(row=0, column=0, sticky='w', ipadx=10, ipady=10)
dialog.columnconfigure(0, weight=1)
# show a dialog with the output.
sheet = Sheet(
Dialogs.dialog,
header_font=(self.header_font, self.header_font_size, "bold"),
font=(self.content_font, self.content_font_size, "normal"),
align='w',
row_height="1", # str value for row height in number of lines.
row_index_align="w",
auto_resize_default_row_index=False,
row_index_width=40,
header_align='center',
empty_horizontal=0,
empty_vertical=0,
)
sheet.headers(['Source', '', 'Destination'])
target_parent = os.path.dirname(self.impartus.download_dir)
for row, (source, destination) in enumerate(moved_files.items()):
source = source[len(target_parent) + 1:]
destination = destination[len(target_parent) + 1:]
sheet.insert_row([source, Icons.MOVED_TO, destination])
sheet.set_all_column_widths()
sheet.grid(row=1, column=0, sticky='nsew')
ok_button = tk.Button(dialog,
text='OK',
command=self.on_auto_organize_dialog_close)
ok_button.grid(row=2, column=0, padx=10, pady=10)
class AnalyzeStockScreen(Frame):
def __init__(self, master=None, session=None):
super().__init__(master)
self.master = master
self.session = session
self.analyze_controller = FAnalyzeStockController(session)
self.pack(padx=20, pady=20)
self.create_widgets()
def create_widgets(self):
# Khởi tạo tab phân tích chứng khoán
## Phần nhập ngày bắt đầu
self.start_date_frame = Frame(self)
self.start_date_frame.pack(fill=X)
self.start_date_label = Label(self.start_date_frame,
text="Start date",
width=10)
self.start_date_label.pack(side=LEFT, padx=5, pady=5)
self.start_date_input = Entry(self.start_date_frame)
self.start_date_input.pack(fill=X, padx=5, expand=True)
## Phần nhập ngày kết thúc
self.end_date_frame = Frame(self)
self.end_date_frame.pack(fill=X)
self.end_date_label = Label(self.end_date_frame,
text="End date",
width=10)
self.end_date_label.pack(side=LEFT, padx=5, pady=5)
self.end_date_input = Entry(self.end_date_frame)
self.end_date_input.pack(fill=X, padx=5, expand=True)
## Phần nút bấm
self.stock_submit_frame = Frame(self)
self.stock_submit_frame.pack(fill=X, padx=10)
## Phần button phân tích
self.stock_submit = Button(self.stock_submit_frame)
self.stock_submit["text"] = "Tính"
self.stock_submit["command"] = lambda: self.start_calc()
self.stock_submit.pack(side=RIGHT)
# Phầns button export excel
self.export_excel_button = Button(self.stock_submit_frame,
state=DISABLED)
self.export_excel_button["text"] = "Export file"
self.export_excel_button["command"] = lambda: self.export_file_excel()
self.export_excel_button.pack(side=RIGHT, padx=20)
## Phần bản tính
self.tabel_frame = Frame(self)
self.tabel_frame.pack(fill=X, pady=10)
self.sheet = Sheet(self.tabel_frame)
self.sheet.pack(fill=X)
self.header = ("Mã", "Giá đóng nhỏ nhất", "Giá đóng lớn nhất")
self.sheet.headers(self.header)
### Cấu hình cho tkinter sheet
self.sheet.enable_bindings(
("single_select", "row_select", "column_width_resize", "arrowkeys",
"right_click_popup_menu", "rc_select", "rc_insert_row",
"rc_delete_row", "copy", "cut", "paste", "delete", "undo",
"edit_cell"))
def start_calc(self):
## Disable trang export file button
self.export_excel_button["state"] = DISABLED
datetime_format = "%d/%m/%Y"
startDate = datetime.strptime(self.start_date_input.get().strip(),
datetime_format)
endDate = datetime.strptime(self.end_date_input.get(), datetime_format)
self.analyzeStocks = self.analyze_controller.get_analyze_stocks_data(
startDate, endDate)
## Tạo data cho tkinter sheet
self.data_sheet = []
for anaStock in self.analyzeStocks:
self.data_sheet.append([
anaStock.code, anaStock.closedPriceMin, anaStock.closedPriceMax
])
## Thêm vào tkinter sheet
self.sheet.set_sheet_data(self.data_sheet)
self.export_excel_button["state"] = NORMAL
def export_file_excel(self):
## Tạo bảnh tính excel với tên file là file_excel_<Thời gian tính theo giây>
workbook = xlsxwriter.Workbook('file_excel_{}.xlsx'.format(
time.time()))
worksheet = workbook.add_worksheet()
## Tạo kiểu kiểu chữ
bold = workbook.add_format({'bold': True})
## Phần header cho trang tính excel
worksheet.write('A1', 'Mã', bold)
worksheet.write('B1', 'Giá đóng nhỏ nhất', bold)
worksheet.write('C1', 'Giá đóng lớn nhất', bold)
## Thêm dòng vào bảng tính
row = 1
for record in self.analyzeStocks:
worksheet.write(row, 0, record.code)
worksheet.write(row, 1, record.closedPriceMin)
worksheet.write(row, 2, record.closedPriceMax)
row += 1
## Lưu file và đóng bảng tính
workbook.close()
class Content:
def __init__(self, app: tk.Tk, login: LoginForm, toolbar: Toolbar,
menubar: Menubar, impartus: Impartus):
self.app = app
self.login = login
self.toolbar = toolbar
self.menubar = menubar
self.impartus = impartus
conf = Config.load(ConfigType.IMPARTUS)
self.content_font = conf.get('content_font').get(platform.system())
self.content_font_size = conf.get('content_font_size')
self.header_font = conf.get('header_font').get(platform.system())
self.header_font_size = conf.get('header_font_size')
self.frame_content = None
self.sheet = None
self.videos = None
self.video_slide_mapping = None
self.expected_real_paths_differ = False
self.all_captions_found = True
self.offline_video_ttid_mapping = None
# sort options
self.sort_by = None
self.sort_order = None
# threads for downloading videos / slides.
self.threads = None
self.logger = logging.getLogger(self.__class__.__name__)
def _init_content(self):
if self.frame_content:
self.frame_content.destroy()
self.frame_content = self.add_content_frame(self.app)
self.sheet = Sheet(
self.frame_content,
header_font=(self.header_font, self.header_font_size, "bold"),
font=(self.content_font, self.content_font_size, "normal"),
align='w',
row_height="1", # str value for row height in number of lines.
row_index_align="w",
auto_resize_default_row_index=False,
row_index_width=40,
header_align='center',
empty_horizontal=0,
empty_vertical=0,
)
self.sheet.enable_bindings((
"single_select",
"column_select",
"column_width_resize",
"double_click_column_resize",
"edit_cell",
"copy",
))
self.sheet.grid(row=0, column=0, sticky='nsew')
self.set_headers()
self.set_display_columns()
self.frame_content.columnconfigure(0, weight=1)
self.frame_content.rowconfigure(0, weight=1)
self.sheet.extra_bindings('column_select', self.sort_table)
self.sheet.extra_bindings('cell_select', self.on_click_button_handler)
# sort options
self.sort_by = None
self.sort_order = None
# threads for downloading videos / slides.
self.threads = dict()
def add_content_frame(self, anchor) -> tk.Frame: # noqa
frame_content = tk.Frame(anchor, padx=0, pady=0)
frame_content.grid(row=2, column=0, sticky='nsew')
return frame_content
def set_headers(self, sort_by=None, sort_order=None):
"""
Set the table headers.
"""
# set column title to reflect sort status
headers = list()
for name, value in Columns.display_columns.items():
if value.get('sortable'):
if name == sort_by:
sort_icon = Icons.SORT_DESC if sort_order == 'desc' else Icons.SORT_ASC
else:
sort_icon = Icons.UNSORTED
text = '{} {}'.format(value['display_name'], sort_icon)
else:
text = value['display_name']
if value.get('editable'):
text = '{} {}'.format(Icons.EDITABLE, text)
headers.append(text)
self.sheet.headers(headers)
def set_display_widgets(self):
"""
Create the table/sheet.
Fill in the data for table content, Set the buttons and their states.
"""
self.fetch_content()
self.fill_content()
def on_click_button_handler(self, args):
"""
On click handler for all the buttons, calls the corresponding function as defined by self.button_columns
"""
(event, row, col) = args
real_col = self.get_real_col(col)
self.sheet.highlight_rows(rows=[row], redraw=False)
self.sheet.highlight_columns(columns=[col], redraw=False)
# is subject field
col_name = Columns.column_names[real_col]
if Columns.all_columns[col_name].get('editable'):
old_value = self.sheet.get_cell_data(row, real_col)
self.sheet.create_text_editor(row=row,
column=real_col,
text=old_value,
set_data_ref_on_destroy=False,
binding=partial(
self.end_edit_cell, old_value))
# not a button.
if Columns.all_columns[col_name].get('type') != 'button':
self.sheet.refresh()
return
state_button_col_name, state_button_col_num = self.get_state_button(
col_name)
state = self.sheet.get_cell_data(row, state_button_col_num)
if state == 'False': # data read from sheet is all string.
self.sheet.refresh()
return
# disable the button if it is one of the Download buttons, to prevent a re-download.
if col_name == 'download_video':
self.sheet.set_cell_data(row,
real_col,
Icons.PAUSE_DOWNLOAD,
redraw=False)
elif col_name == 'download_slides':
cs = Config.load(
ConfigType.COLORSCHEMES)[Variables().colorscheme_var().get()]
self.disable_button(row, real_col, cs)
func_name = Columns.all_columns[col_name]['function']
getattr(self, func_name)(row, real_col)
def get_state_button(self, button_name): # noqa
if Columns.all_columns[button_name].get('state'):
state_col_name = Columns.all_columns[button_name].get('state')
state_col_number = Columns.column_names.index(state_col_name)
return state_col_name, state_col_number
else:
return None, None
def end_edit_cell(self, old_value, event=None):
row, col = (event[0], event[1])
new_value = self.sheet.get_text_editor_value(
event,
r=row,
c=col,
set_data_ref_on_destroy=True,
move_down=False,
redraw=True,
recreate=True)
# empty value or escape pressed.
if not new_value or new_value == '':
return
# no changes made.
if old_value == new_value:
return
col_name = Columns.column_names[self.get_real_col(col)]
columns_item = Columns.data_columns[col_name]
orig_values_col_name = columns_item.get('original_values_col')
original_value = self.sheet.get_cell_data(
row, Columns.column_names.index(orig_values_col_name))
for i, data in enumerate(
self.sheet.get_column_data(
Columns.column_names.index(orig_values_col_name))):
if data == original_value:
self.sheet.set_cell_data(i, col, new_value)
self.expected_real_paths_differ = True
Mappings.update_mappings(orig_values_col_name, original_value,
new_value)
self.reset_column_sizes()
self.sheet.deselect(row=row, column=col, redraw=False)
self.sheet.refresh()
if self.expected_real_paths_differ or not self.all_captions_found:
self.toolbar.auto_organize_button.config(state='normal')
self.menubar.actions_menu.entryconfig(Labels.AUTO_ORGANIZE,
state='normal')
def reset_column_sizes(self):
"""
Adjust column sizes after data has been filled.
"""
# resize cells
self.sheet.set_all_column_widths()
# reset column widths to fill the screen
pad = 5
column_widths = self.sheet.get_column_widths()
table_width = self.sheet.RI.current_width + sum(column_widths) + len(
column_widths) + pad
diff_width = self.frame_content.winfo_width() - table_width
# adjust extra width only to top N data columns
n = 3
column_states = [
v.get() for v in Variables().display_columns_vars().values()
]
count = 0
for k, v in enumerate(column_states):
if Columns.column_names[k] == 'downloaded':
break
count += v
# range(0..count) is all data columns.
data_col_widths = {k: v for k, v in enumerate(column_widths[:count])}
top_n_cols = sorted(data_col_widths,
key=data_col_widths.get,
reverse=True)[:n]
for i in top_n_cols:
self.sheet.column_width(i, column_widths[i] + diff_width // n)
def get_index(self, row):
"""
Find the values stored in the hidden column named 'Index', given a row record.
In case the row value has been updated due to sorting the table, Index field helps identify the new location
of the associated record.
"""
# find where is the Index column
index_col = Columns.column_names.index('index')
# original row value as per the index column
return int(self.sheet.get_cell_data(row, index_col))
def get_row_after_sort(self, index_value):
# find the new correct location of the row_index
col_index = Columns.column_names.index('index')
col_data = self.sheet.get_column_data(col_index)
return col_data.index(str(index_value))
def progress_bar_text(self, value, processed=False):
"""
return progress bar text, calls the unicode/ascii implementation.
"""
conf = Config.load(ConfigType.IMPARTUS)
if conf.get('progress_bar') == 'unicode':
text = self.progress_bar_text_unicode(value)
else:
text = self.progress_bar_text_ascii(value)
pad = ' ' * 2
if 0 < value < 100:
percent_text = '{:2d}%'.format(value)
status = percent_text
elif value == 0:
status = '{}{}{}'.format(pad, Icons.VIDEO_NOT_DOWNLOADED, pad)
else: # 100 %
if processed:
status = '{}{}{}'.format(pad, Icons.VIDEO_DOWNLOADED, pad)
else:
status = '{}{}{}'.format(pad, Icons.VIDEO_PROCESSING, pad)
return '{} {}{}'.format(text, status, pad)
def progress_bar_text_ascii(self, value): # noqa
"""
progress bar implementation with ascii characters.
"""
bars = 50
ascii_space = " "
if value > 0:
progress_text = '{}'.format('❘' * (value * bars // 100))
empty_text = '{}'.format(ascii_space * (bars - len(progress_text)))
full_text = '{}{} '.format(progress_text, empty_text)
else:
full_text = '{}'.format(ascii_space * bars)
return full_text
def progress_bar_text_unicode(self, value): # noqa
"""
progress bar implementation with unicode blocks.
"""
chars = ['▏', '▎', '▍', '▌', '▋', '▊', '▉', '█']
# 1 full unicode block = 8 percent values
# => 13 unicode blocks needed to represent counter 100.
unicode_space = ' '
if value > 0:
# progress_text: n characters, empty_text: 13-n characters
progress_text = '{}{}'.format(chars[-1] * (value // 8),
chars[value % 8])
empty_text = '{}'.format(unicode_space * (13 - len(progress_text)))
full_text = '{}{}'.format(progress_text, empty_text)
else:
# all 13 unicode whitespace.
full_text = '{} '.format(unicode_space * 13)
return full_text
def progress_bar_callback(self, count, row, col, processed=False):
"""
Callback function passed to the backend, where it computes the download progress.
Every time the function is called, it will update the progress bar value.
"""
updated_row = self.get_row_after_sort(row)
new_text = self.progress_bar_text(count, processed)
if new_text != self.sheet.get_cell_data(updated_row, col):
self.sheet.set_cell_data(updated_row, col, new_text, redraw=True)
def get_real_col(self, col): # noqa
"""
with configurable column list, the col number returned by tksheet may not be the same as
column no from self.all_columns/self.display_columns. Use self.display_column_vars to identify and return
the correct column.
"""
# find n-th visible column, where n=col
i = 0
for c, state in enumerate(Variables().display_columns_vars().values()):
if state.get() == 1:
if i == col:
return c
i += 1
def fetch_content(self):
self.videos = dict()
self.video_slide_mapping = dict()
self.expected_real_paths_differ = False
self.offline_video_ttid_mapping = None
root_url = self.login.url_box.get()
subject_dicts = self.impartus.get_subjects(root_url)
has_flipped_lectures = False
for subject_dict in subject_dicts:
videos_by_subject = self.impartus.get_lectures(
root_url, subject_dict)
flipped_videos_by_subject = self.impartus.get_flipped_lectures(
root_url, subject_dict)
if len(flipped_videos_by_subject):
has_flipped_lectures = True
all_videos_by_subject = [
*videos_by_subject, *flipped_videos_by_subject
]
slides = self.impartus.get_slides(root_url, subject_dict)
mapping_dict = self.impartus.map_slides_to_videos(
all_videos_by_subject, slides)
for key, val in mapping_dict.items():
self.video_slide_mapping[key] = val
self.videos[subject_dict.get('subjectId')] = {
x['ttid']: x
for x in all_videos_by_subject
}
state = 'normal' if has_flipped_lectures else 'disabled'
self.toolbar.flipped_video_quality_dropdown.configure(state=state)
self.menubar.main_menu.entryconfig(Labels.VIDEO, state=state)
def fill_content(self):
# A mapping dict containing previously downloaded, and possibly moved around / renamed videos.
# extract their ttid and map those to the correct records, to avoid forcing the user to re-download.
self.offline_video_ttid_mapping = self.impartus.get_mkv_ttid_map()
row = 0
sheet_rows = list()
for subject_id, videos in self.videos.items():
for ttid, video_metadata in videos.items():
video_metadata = Utils.add_new_fields(video_metadata,
self.video_slide_mapping)
video_path = self.impartus.get_mkv_path(video_metadata)
if not os.path.exists(video_path):
# or search from the downloaded videos, using video_ttid_map
video_path_moved = self.offline_video_ttid_mapping.get(
str(ttid))
if video_path_moved:
# For now, use the offline path if a video found. Also set the flag to enable move/rename button
video_path = video_path_moved
self.expected_real_paths_differ = True
captions_path = self.impartus.get_captions_path(video_metadata)
if not os.path.exists(captions_path):
self.all_captions_found = False
slides_path = self.impartus.get_slides_path(video_metadata)
video_exists_on_disk = video_path and os.path.exists(
video_path)
slides_exist_on_server = self.video_slide_mapping.get(ttid)
slides_exist_on_disk, slides_path = self.impartus.slides_exist_on_disk(
slides_path)
metadata = {
'video_metadata': video_metadata,
'video_path': video_path,
'video_exists_on_disk': video_exists_on_disk,
'slides_exist_on_server': slides_exist_on_server,
'slides_exist_on_disk': slides_exist_on_disk,
'slides_url': self.video_slide_mapping.get(ttid),
'slides_path': slides_path,
'captions_path': captions_path,
}
row_items = list()
button_states = list()
# data items
for col, (key, item) in enumerate(Columns.all_columns.items()):
text = ''
if item['type'] == 'data':
text = video_metadata[key]
# title case
if item.get('title_case'):
text = " ".join(text.splitlines()).strip().title()
elif item['type'] == 'auto':
text = row
elif item['type'] == 'progressbar':
if video_exists_on_disk:
text = self.progress_bar_text(100, processed=True)
else:
text = self.progress_bar_text(0)
elif item['type'] == 'button':
button_states.append(
self.get_button_state(key, video_exists_on_disk,
slides_exist_on_server,
slides_exist_on_disk))
text = item.get('text')
elif item['type'] == 'button_state':
text = button_states.pop(0)
elif item['type'] == 'metadata':
text = metadata
elif item['type'] == 'original_value':
text = video_metadata[key]
row_items.append(text)
row += 1
sheet_rows.append(row_items)
self._init_content()
self.sheet.insert_rows(sheet_rows, idx='end')
self.reset_column_sizes()
self.decorate()
# update button status
self.set_button_status(redraw=True)
self.sheet.grid(row=0, column=0, sticky='nsew')
def sort_table(self, args):
"""
Sorts the table content.
"""
col = args[1]
real_col = self.get_real_col(col)
self.sheet.deselect("all")
col_name = Columns.column_names[real_col]
if not Columns.all_columns[col_name].get('sortable'):
return
sort_by = col_name
if sort_by == self.sort_by:
sort_order = 'asc' if self.sort_order == 'desc' else 'desc'
else:
sort_order = 'desc'
self.sort_by = sort_by
self.sort_order = sort_order
reverse = True if sort_order == 'desc' else False
table_data = self.sheet.get_sheet_data()
table_data.sort(key=lambda x: x[real_col], reverse=reverse)
self.set_headers(sort_by, sort_order)
self.set_button_status()
self.sheet.refresh()
def save_captions_if_needed(self, video_metadata, root_url, captions_path):
chat_msgs = self.impartus.get_chats(video_metadata, root_url)
date_format = "%Y-%m-%d %H:%M:%S"
start_epoch = int(
datetime.strptime(video_metadata['startTime'],
date_format).timestamp())
try:
vtt_content = Captions.get_vtt(chat_msgs, start_epoch)
Captions.save_vtt(vtt_content, captions_path)
except CaptionsNotFound as ex:
self.logger.info(
"no lecture chat found for {}".format(captions_path))
return
return True
def _download_video(self, video_metadata, filepath, captions_path,
root_url, row, col, pause_ev, resume_ev): # noqa
"""
Download a video in a thread. Update the UI upon completion.
"""
# create a new Impartus session reusing existing token.
imp = Impartus(self.impartus.token)
pb_col = Columns.column_names.index('downloaded')
# # voodoo alert:
# It is possible for user to sort the table while download is in progress.
# In such a case, the row index supplied to the function call won't match the row index
# required to update the correct progressbar/open/play buttons, which now exists at a new
# location.
# The hidden column index keeps the initial row index, and remains unchanged.
# Use row_index to identify the new correct location of the progress bar.
row_index = self.get_index(row)
imp.process_video(video_metadata,
filepath,
root_url,
pause_ev,
resume_ev,
partial(self.progress_bar_callback,
row=row_index,
col=pb_col),
video_quality=Variables().lecture_quality_var())
# also download lecture chats, and create a webvtt subtitles file.
self.save_captions_if_needed(video_metadata, root_url, captions_path)
# download complete, enable open / play buttons
updated_row = self.get_row_after_sort(row_index)
# update progress bar status to complete.
self.progress_bar_callback(row=row_index,
col=pb_col,
count=100,
processed=True)
self.sheet.set_cell_data(updated_row,
Columns.column_names.index('download_video'),
Icons.DOWNLOAD_VIDEO)
self.disable_button(updated_row,
Columns.column_names.index('download_video'))
# enable buttons.
self.enable_button(updated_row,
Columns.column_names.index('open_folder'))
self.enable_button(updated_row,
Columns.column_names.index('play_video'))
def add_slides(self, row, col): # noqa
conf = Config.load(ConfigType.IMPARTUS)
file_types = [(str(ext).upper(), '*.{}'.format(ext))
for ext in conf.get('allowed_ext')]
filepaths = tkinter.filedialog.askopenfilenames(filetypes=file_types)
data = self.read_metadata(row)
slides_folder_path = os.path.dirname(data.get('video_path'))
for filepath in filepaths:
shutil.copy(filepath, slides_folder_path)
def pause_resume_button_click(self, row, col, pause_event, resume_event):
row_index = self.get_index(row)
updated_row = self.get_row_after_sort(row_index)
if pause_event.is_set():
self.sheet.set_cell_data(updated_row,
col,
Icons.PAUSE_DOWNLOAD,
redraw=True)
resume_event.set()
pause_event.clear()
else:
self.sheet.set_cell_data(updated_row,
col,
Icons.RESUME_DOWNLOAD,
redraw=True)
pause_event.set()
resume_event.clear()
def download_video(self, row, col):
"""
callback function for Download button.
Creates a thread to download the request video.
"""
data = self.read_metadata(row)
video_metadata = data.get('video_metadata')
filepath = data.get('video_path')
captions_path = data.get('captions_path')
root_url = self.login.url_box.get()
real_row = self.get_index(row)
if self.threads.get(real_row):
pause_ev = self.threads.get(real_row)['pause_event']
resume_ev = self.threads.get(real_row)['resume_event']
self.pause_resume_button_click(row, col, pause_ev, resume_ev)
return
from threading import Event
pause_event = Event()
resume_event = Event()
# note: args is a tuple.
thread = threading.Thread(target=self._download_video,
args=(
video_metadata,
filepath,
captions_path,
root_url,
row,
col,
pause_event,
resume_event,
))
self.threads[real_row] = {
'thread': thread,
'pause_event': pause_event,
'resume_event': resume_event,
}
thread.start()
def _download_slides(self, ttid, file_url, filepath, root_url, row):
"""
Download a slide doc in a thread. Update the UI upon completion.
"""
# create a new Impartus session reusing existing token.
imp = Impartus(self.impartus.token)
if imp.download_slides(ttid, file_url, filepath, root_url):
# download complete, enable show slides buttons
self.enable_button(row, Columns.column_names.index('show_slides'))
else:
tkinter.messagebox.showerror(
'Error',
'Error downloading slides, see console logs for details.')
self.enable_button(row,
Columns.column_names.index('download_slides'))
def download_slides(self, row, col): # noqa
"""
callback function for Download button.
Creates a thread to download the request video.
"""
data = self.read_metadata(row)
video_metadata = data.get('video_metadata')
ttid = video_metadata['ttid']
file_url = data.get('slides_url')
filepath = data.get('slides_path')
root_url = self.login.url_box.get()
# note: args is a tuple.
thread = threading.Thread(target=self._download_slides,
args=(
ttid,
file_url,
filepath,
root_url,
row,
))
# self.threads.append(thread)
thread.start()
def read_metadata(self, row):
"""
We saved a hidden column 'metadata' containing metadata for each record.
Extract it, and eval it as python dict.
"""
metadata_col = Columns.column_names.index('metadata')
data = self.sheet.get_cell_data(row, metadata_col)
return ast.literal_eval(data)
def open_folder(self, row, col): # noqa
"""
fetch video_path's folder from metadata column's cell and open system launcher with it.
"""
data = self.read_metadata(row)
video_folder_path = os.path.dirname(data.get('video_path'))
Utils.open_file(video_folder_path)
def play_video(self, row, col): # noqa
"""
fetch video_path from metadata column's cell and open system launcher with it.
"""
data = self.read_metadata(row)
Utils.open_file(data.get('video_path'))
def show_slides(self, row, col): # noqa
"""
fetch slides_path from metadata column's cell and open system launcher with it.
"""
data = self.read_metadata(row)
Utils.open_file(data.get('slides_path'))
def auto_organize(self):
self.toolbar.auto_organize_button.config(state='disabled')
self.menubar.actions_menu.entryconfig(Labels.AUTO_ORGANIZE,
state='disabled')
moved_files = dict()
conf = Config.load(ConfigType.IMPARTUS)
for subject_id, videos in self.videos.items():
for ttid, video_metadata in videos.items():
video_metadata = Utils.add_new_fields(video_metadata,
self.video_slide_mapping)
# for videos
expected_video_path = self.impartus.get_mkv_path(
video_metadata)
real_video_path = self.offline_video_ttid_mapping.get(
str(ttid))
if real_video_path and \
pathlib.PurePath(expected_video_path) != pathlib.PurePath(real_video_path) \
and os.path.exists(real_video_path):
Utils.move_and_rename_file(real_video_path,
expected_video_path)
self.logger.info('moved {} -> {}'.format(
real_video_path, expected_video_path))
moved_files[real_video_path] = expected_video_path
# also update the offline_video_ttid_mapping
self.offline_video_ttid_mapping[str(
ttid)] = expected_video_path
# also check any slides.
for ext in conf.get('allowed_ext'):
slides_path = '{}.{}'.format(
real_video_path[:-len(".mkv")], ext)
if os.path.exists(slides_path):
expected_slides_path = '{}.{}'.format(
expected_video_path[:-len(".mkv")], ext)
Utils.move_and_rename_file(slides_path,
expected_slides_path)
self.logger.info('moved {} -> {}'.format(
slides_path, expected_slides_path))
moved_files[slides_path] = expected_slides_path
# is the folder empty, remove it.? [also any empty parent folders]
old_video_dir = os.path.dirname(real_video_path)
sys_name = platform.system()
if conf.get('ignore_files').get(sys_name):
ignore_files = conf.get('ignore_files')[sys_name]
else:
ignore_files = []
while True:
dir_files = [
x for x in os.listdir(old_video_dir)
if x not in ignore_files
]
if len(dir_files) > 0:
break
for file in ignore_files:
filepath = os.path.join(old_video_dir, file)
if os.path.exists(filepath):
os.unlink(filepath)
os.rmdir(old_video_dir)
self.logger.info('removed empty directory: {}'.format(
old_video_dir))
# parent path.
old_video_dir = Path(old_video_dir).parent.absolute()
# captions
expected_captions_path = self.impartus.get_captions_path(
video_metadata)
if not os.path.exists(expected_captions_path):
if self.save_captions_if_needed(video_metadata,
self.login.url_box.get(),
expected_captions_path):
self.logger.info('downloaded captions: {}'.format(
expected_captions_path))
self.all_captions_found = True
if len(moved_files) > 0:
self.auto_organize_dialog(moved_files)
self.expected_real_paths_differ = False
def set_display_columns(self):
column_states = [
i
for i, v in enumerate(Variables().display_columns_vars().values())
if v.get() == 1
]
self.sheet.display_columns(indexes=column_states,
enable=True,
redraw=False)
self.reset_column_sizes()
self.sheet.refresh()
def odd_even_color(self, cs: Dict, redraw=False):
"""
Apply odd/even colors for table for better looking UI.
"""
num_rows = self.sheet.total_rows()
self.sheet.highlight_rows(list(range(0, num_rows, 2)),
bg=cs['even_row']['bg'],
fg=cs['even_row']['fg'],
redraw=redraw)
self.sheet.highlight_rows(list(range(1, num_rows, 2)),
bg=cs['odd_row']['bg'],
fg=cs['odd_row']['fg'],
redraw=redraw)
def progress_bar_color(self, cs: Dict, redraw=True):
"""
Set progress bar color.
"""
col = Columns.column_names.index('downloaded')
num_rows = self.sheet.total_rows()
for row in range(num_rows):
odd_even_bg = cs['odd_row']['bg'] if row % 2 else cs['even_row'][
'bg']
self.sheet.highlight_cells(row,
col,
fg=cs['progressbar']['fg'],
bg=odd_even_bg,
redraw=redraw)
def set_button_status(self, redraw=False):
"""
reads the states of the buttons from the hidden state columns, and sets the button states appropriately.
"""
col_indexes = [
x for x, v in enumerate(Columns.all_columns.values())
if v['type'] == 'button_state'
]
num_buttons = len(col_indexes)
for row, row_item in enumerate(self.sheet.get_sheet_data()):
for col in col_indexes:
# data set via sheet.insert_row retains tuple/list's element data type,
# data set via sheet.set_cell_data makes everything a string.
# Consider everything coming out of a sheet as string to avoid any issues.
state = str(row_item[col])
if state == 'True':
self.enable_button(row, col - num_buttons, redraw=redraw)
elif state == 'False':
self.disable_button(row, col - num_buttons, redraw=redraw)
return
def get_button_state(self, key, video_exists_on_disk,
slides_exist_on_server, slides_exist_on_disk): # noqa
"""
Checks to identify when certain buttons should be enabled/disabled.
"""
state = True
if key == 'download_video' and video_exists_on_disk:
state = False
elif key == 'open_folder' and not video_exists_on_disk:
state = False
elif key == 'play_video' and not video_exists_on_disk:
state = False
elif key == 'download_slides' and (slides_exist_on_disk
or not slides_exist_on_server):
state = False
elif key == 'show_slides' and not slides_exist_on_disk:
state = False
return state
def disable_button(self, row, col, redraw=False):
"""
Disable a button given it's row/col position.
"""
cs = Config.load(
ConfigType.COLORSCHEMES)[Variables().colorscheme_var().get()]
self.sheet.highlight_cells(row,
col,
bg=cs['disabled']['bg'],
fg=cs['disabled']['fg'],
redraw=redraw)
# update state field.
state_button_col_name, state_button_col_num = self.get_state_button(
Columns.column_names[col])
self.sheet.set_cell_data(row,
state_button_col_num,
False,
redraw=redraw)
def enable_button(self, row, col, redraw=False):
"""
Enable a button given it's row/col position.
"""
cs = Config.load(
ConfigType.COLORSCHEMES)[Variables().colorscheme_var().get()]
odd_even_bg = cs['odd_row']['bg'] if row % 2 else cs['even_row']['bg']
odd_even_fg = cs['odd_row']['fg'] if row % 2 else cs['even_row']['fg']
self.sheet.highlight_cells(row,
col,
bg=odd_even_bg,
fg=odd_even_fg,
redraw=redraw)
# update state field.
state_button_col_name, state_button_col_num = self.get_state_button(
Columns.column_names[col])
self.sheet.set_cell_data(row,
state_button_col_num,
True,
redraw=redraw)
def set_readonly_columns(self, redraw=False):
readonly_cols = [
i for i, (k, v) in enumerate(Columns.all_columns.items())
if not v.get('editable')
]
self.sheet.readonly_columns(columns=readonly_cols,
readonly=True,
redraw=redraw)
def set_colorscheme(self, cs):
if self.frame_content:
self.frame_content.configure(bg=cs['root']['bg'])
if self.sheet:
self.sheet.set_options(frame_bg=cs['table']['bg'],
table_bg=cs['table']['bg'],
table_fg=cs['table']['fg'],
header_bg=cs['header']['bg'],
header_fg=cs['header']['fg'],
header_grid_fg=cs['table']['grid'],
index_grid_fg=cs['table']['grid'],
header_border_fg=cs['table']['grid'],
index_border_fg=cs['table']['grid'],
table_grid_fg=cs['table']['grid'],
top_left_bg=cs['header']['bg'],
top_left_fg=cs['header']['bg'])
self.odd_even_color(cs, redraw=False)
self.progress_bar_color(cs, redraw=False)
self.set_button_status(redraw=False)
self.set_readonly_columns(redraw=False)
self.sheet.refresh()
def decorate(self):
"""
calls multiple ui related tweaks.
"""
self.align_columns()
cs = Config.load(
ConfigType.COLORSCHEMES)[Variables().colorscheme_var().get()]
self.set_colorscheme(cs)
self.odd_even_color(cs)
self.progress_bar_color(cs)
def align_columns(self):
# data and progressbar west/left aligned, button center aligned.
self.sheet.align_columns([
Columns.column_names.index(k) for k in Columns.data_columns.keys()
],
align='w')
self.sheet.align_columns([
Columns.column_names.index(k)
for k in Columns.progressbar_column.keys()
],
align='w')
self.sheet.align_columns([
Columns.column_names.index(k)
for k in Columns.button_columns.keys()
],
align='center')
def show_video_callback(self, impartus: Impartus, event=None): # noqa
if threading.activeCount(
) > 1: # 1. main thread, 2,3... download threads.
response = tk.messagebox.askquestion(
'Download(s) in progress!',
"Reloading the content will lose the downloads in progress.\n"
+ "Do you want to continue?",
icon='warning')
if response != 'yes':
return
self.toolbar.reload_button.config(state='disabled')
self.menubar.actions_menu.entryconfig(Labels.RELOAD, state='disabled')
self.toolbar.auto_organize_button.config(state='disabled')
self.menubar.actions_menu.entryconfig(Labels.AUTO_ORGANIZE,
state='disabled')
self.toolbar.frame_toolbar.grid(row=1, column=0, sticky='ew')
self.login.authenticate(impartus)
self.set_display_widgets()
self.toolbar.reload_button.config(state='normal')
self.menubar.actions_menu.entryconfig(Labels.RELOAD, state='normal')
if self.expected_real_paths_differ or not self.all_captions_found:
auto_organize_button_state = 'normal'
else:
auto_organize_button_state = 'disabled'
self.toolbar.auto_organize_button.config(
state=auto_organize_button_state)
self.menubar.actions_menu.entryconfig(Labels.AUTO_ORGANIZE,
state=auto_organize_button_state)
def on_auto_organize_dialog_close(self):
self.login.authenticate(self.impartus)
Dialogs.on_dialog_close()
self.show_video_callback(self.impartus)
def auto_organize_dialog(self, moved_files): # noqa
dialog = Dialogs.create_dialog(
on_close_callback=self.on_auto_organize_dialog_close,
title='Auto Organize')
label = tk.Label(
dialog,
text='Following files were moved / renamed -',
)
label.grid(row=0, column=0, sticky='w', ipadx=10, ipady=10)
dialog.columnconfigure(0, weight=1)
# show a dialog with the output.
sheet = Sheet(
Dialogs.dialog,
header_font=(self.header_font, self.header_font_size, "bold"),
font=(self.content_font, self.content_font_size, "normal"),
align='w',
row_height="1", # str value for row height in number of lines.
row_index_align="w",
auto_resize_default_row_index=False,
row_index_width=40,
header_align='center',
empty_horizontal=0,
empty_vertical=0,
)
sheet.headers(['Source', '', 'Destination'])
target_parent = os.path.dirname(self.impartus.download_dir)
for row, (source, destination) in enumerate(moved_files.items()):
source = source[len(target_parent) + 1:]
destination = destination[len(target_parent) + 1:]
sheet.insert_row([source, Icons.MOVED_TO, destination])
sheet.set_all_column_widths()
sheet.grid(row=1, column=0, sticky='nsew')
ok_button = tk.Button(dialog,
text='OK',
command=self.on_auto_organize_dialog_close)
ok_button.grid(row=2, column=0, padx=10, pady=10)
class PageScreener(tk.Frame):
def __init__(self, parent, controller):
tk.Frame.__init__(self, parent)
self.controller = controller
self.flag = 0
self.FLAG_THRD_SCREEN_DATA = 1
# 处理数据队列
self.queData = Queue()
# 数据缓存，用于重复使用，
self.listData = []
topFrame = tk.Frame(self)
self.btnMainPage = tk.Button(topFrame,
text="主页面",
command=self.on_btn_main_page).pack(
side=tk.LEFT, padx=4)
# 日期选择
# Calendar((x, y), 'ur').selection() 获取日期，x,y为点坐标
date_start_gain = lambda: [
self.date_start.set(date)
for date in [CalendarCustom(None, 'ul').selection()] if date
]
self.btnStart = tk.Button(topFrame,
text='选股日期:',
command=date_start_gain)
self.btnStart.pack(side=tk.LEFT, padx=4)
# ret = self.btnStart.winfo_geometry()
self.date_start = tk.StringVar()
ttk.Entry(topFrame, textvariable=self.date_start).pack(side=tk.LEFT)
today = get_last_valid_trade_date()
dt = datetime.datetime.strptime(today, '%Y%m%d')
today = dt.strftime('%Y-%m-%d')
self.date_start.set(today)
tk.Label(topFrame, text="选股类型：").pack(side=tk.LEFT)
self.screenTypeStr = tk.StringVar()
cmbScreenType = ttk.Combobox(topFrame,
width=15,
textvariable=self.screenTypeStr,
state='readonly')
# Adding combobox drop down list
cmbScreenType['values'] = ('平台突破', '均线附近')
cmbScreenType.current(1)
cmbScreenType.pack(side=tk.LEFT, padx=4)
# cmbScreenType.bind("<<ComboboxSelected>>", self.on_cmb_screen_select)
# 选股周期数
self.screenIntervalCountStr = tk.StringVar(value='25')
# state=tk.DISABLED 默认禁止输入
self.screenIntervalCount = ttk.Entry(
topFrame, width=10, textvariable=self.screenIntervalCountStr)
self.screenIntervalCount.pack(side=tk.LEFT, padx=4)
# 选股周期类型
self.screenIntervalTypeStr = tk.StringVar()
cmbScreenInterval = ttk.Combobox(
topFrame,
width=8,
textvariable=self.screenIntervalTypeStr,
state='readonly')
# Adding combobox drop down list
cmbScreenInterval['values'] = ('日', '周', '月')
cmbScreenInterval.current(0)
cmbScreenInterval.pack(side=tk.LEFT, padx=4)
# cmbScreenInterval.bind("<<ComboboxSelected>>", self.on_cmb_screen_interval_select)
self.chkST = tk.IntVar()
tk.Checkbutton(topFrame, text="包括ST",
variable=self.chkST).pack(side=tk.LEFT, padx=4)
self.chkTech = tk.IntVar()
tk.Checkbutton(topFrame, text="包括科创板",
variable=self.chkTech).pack(side=tk.LEFT, padx=4)
self.btnStart = tk.Button(topFrame,
text="选股",
command=self.on_btn_start)
self.btnStart.pack(side=tk.LEFT, padx=4)
topFrame.pack(side=tk.TOP, fill=tk.BOTH)
self.tipsStr = tk.StringVar()
tk.Label(topFrame, textvariable=self.tipsStr,
font=("simsun", 12)).pack(side=tk.LEFT)
self.tipsStr.set('状态：准备...')
# Progress bar widget
self.progress = Progressbar(topFrame,
orient=tk.HORIZONTAL,
length=100,
mode='determinate')
# self.progress.pack(side=tk.LEFT, padx=4)
# self.progress.pack_forget()
# 列表框
self.frameReport = tk.Frame(self)
self.sheet = Sheet(self.frameReport)
self.sheet.enable_bindings((
# "single_select", # "single_select" or "toggle_select"
# "drag_select", # enables shift click selection as well
# "column_drag_and_drop",
# "row_drag_and_drop",
# "column_select",
"row_select",
"column_width_resize",
"double_click_column_resize",
# "row_width_resize",
# "column_height_resize",
# "arrowkeys",
# "row_height_resize",
# "double_click_row_resize",
# "right_click_popup_menu",
# "rc_select",
# "rc_insert_column",
# "rc_delete_column",
# "rc_insert_row",
# "rc_delete_row",
# "hide_columns",
# "copy",
# "cut",
# "paste",
# "delete",
# "undo",
# "edit_cell"
))
self.sheet.pack(side=tk.TOP, fill=tk.BOTH, expand=1)
self.frameReport.pack(side=tk.TOP, fill=tk.BOTH, expand=1, pady=4)
self.sheet.headers(['编码', '名称', '板块', '收盘价', '涨幅'])
self.sheet.refresh()
def on_btn_main_page(self):
self.controller.show_frame('PageMain')
pass
def on_btn_start(self):
if self.FLAG_THRD_SCREEN_DATA == self.flag & self.FLAG_THRD_SCREEN_DATA:
return
self.flag |= self.FLAG_THRD_SCREEN_DATA
self.progress.pack(side=tk.LEFT, padx=4)
# 获取数据线程，取一个放入队列，由另一个线程处理
# 周期参数
iInterval = int(self.screenIntervalCountStr.get())
# TODO 测试代码，
iInterval = 43
type = 'd'
if self.screenIntervalTypeStr.get() == '周':
type = 'w'
elif self.screenIntervalTypeStr.get() == '月':
type = 'm'
start_date = self.date_start.get().replace('-', '')
# 多获取 10 组数据
thrd = threading.Thread(target=dbService.get_data_thread,
args=(
iInterval + 30,
type,
self.chkST.get(),
self.chkTech.get(),
self.queData,
start_date,
))
thrd.setDaemon(True) # 守护线程
thrd.start()
cnt = self.sheet.get_total_rows()
for i in range(cnt):
self.sheet.delete_row(i)
self.sheet.refresh()
if self.screenTypeStr.get() == '平台突破':
thrd = threading.Thread(target=self.screen_platform,
args=(self.queData, ))
thrd.setDaemon(True) # 守护线程
thrd.start()
elif self.screenTypeStr.get() == '均线附近':
thrd = threading.Thread(target=self.screen_ma_around,
args=(self.queData, ))
thrd.setDaemon(True) # 守护线程
thrd.start()
pass
def screen_platform(self, in_q):
"""
平台突破
找出N天内创新高的股票，
:return:
"""
self.btnStart['state'] = 'disabled'
self.tipsStr.set('状态：正在读取数据，请耐心等待...')
# 准备数据
iInterval = int(self.screenIntervalCountStr.get())
type = 'd'
if self.screenIntervalTypeStr.get() == '周':
type = 'w'
elif self.screenIntervalTypeStr.get() == '月':
type = 'm'
self.progress['value'] = 5
# root.update_idletasks()
# 避免中间缺数据 * 2
count = iInterval + 5
datas = dbService.get_data(count, type, self.chkST.get(),
self.chkTech.get())
screenCount = 0
self.tipsStr.set('状态：正在选股，请耐心等待...')
per_interval = len(datas) / 95
step_count = 0
progress_step = 5
pickup_list = []
for it in datas:
step_count += 1
if step_count >= per_interval:
step_count = 0
progress_step += 1
self.progress['value'] = progress_step
# root.update_idletasks()
result = platform_break_through(it, iInterval)
if result:
screenCount += 1
print(result)
pickup_list.append(result)
pickup_sorted = sorted(pickup_list, key=itemgetter(3), reverse=True)
for it in pickup_sorted:
self.sheet.insert_row(values=it)
self.sheet.refresh()
self.progress['value'] = 100
# root.update_idletasks()
self.tipsStr.set('状态：共选出 {:d} 只股票'.format(screenCount))
self.flag &= (~self.FLAG_THRD_SCREEN_DATA)
self.btnStart['state'] = 'normal'
self.progress.pack_forget()
print("screen_platform exit ==========")
pass
def screen_ma_around(self, in_q):
"""
参数 120,2 表示 120 日均线，2%附近 ma120*0.98 < 收盘价 < ma120*1.2
:return:
"""
self.btnStart['state'] = 'disabled'
screenCount = 0
self.tipsStr.set('状态：正在处理，请耐心等待...')
# 准备数据
param = [(self.screenIntervalCountStr.get())]
per_interval = get_code_count() / 95
step_count = 0
progress_step = 5
pickup_list = []
while True:
try:
it = in_q.get_nowait()
except queue.Empty as e1:
continue
if it is None:
break
step_count += 1
if step_count >= per_interval:
step_count = 0
progress_step += 1
self.progress['value'] = progress_step
result = ma25_around(it, param)
if result:
screenCount += 1
print(result)
self.sheet.insert_row(values=result)
# self.sheet.refresh()
# pickup_list.append(result)
# pickup_sorted = sorted(pickup_list, key=itemgetter(3), reverse=True)
#
# for it in pickup_sorted:
# self.sheet.insert_row(values=it)
self.progress['value'] = 100
# root.update_idletasks()
self.tipsStr.set('状态：共选出 {:d} 只股票'.format(screenCount))
self.flag &= (~self.FLAG_THRD_SCREEN_DATA)
self.btnStart['state'] = 'normal'
self.progress.pack_forget()
self.progress['value'] = 0
print("screen_ma_around exit ==========")
pass
def tree_solution_selected(self, selected):
print('tree_solution_selected items:', selected)
pass
class PageDiary(tk.Frame):
def __init__(self, parent, controller):
tk.Frame.__init__(self, parent)
self.controller = controller
topFrame = tk.Frame(self)
self.btnMainPage = tk.Button(topFrame,
text="主页面",
command=self.on_btn_main_page).pack(
side=tk.LEFT, padx=4)
# 日期选择
# Calendar((x, y), 'ur').selection() 获取日期，x,y为点坐标
date_start_gain = lambda: [
self.str_date.set(date)
for date in [CalendarCustom(None, 'ur').selection()] if date
]
tk.Button(topFrame, text='日期:',
command=date_start_gain).pack(side=tk.LEFT, padx=4)
self.str_date = tk.StringVar()
ttk.Entry(topFrame, textvariable=self.str_date,
width=12).pack(side=tk.LEFT)
today = datetime.date.today()
self.str_date.set(today)
tk.Label(topFrame, text="代码：").pack(side=tk.LEFT)
self.strCode = tk.StringVar(value='')
ttk.Entry(topFrame, width=10,
textvariable=self.strCode).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame, text="名称：").pack(side=tk.LEFT)
self.strName = tk.StringVar(value='')
ttk.Entry(topFrame, width=10,
textvariable=self.strName).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame, text="操作时间：").pack(side=tk.LEFT)
self.strTime = tk.StringVar(value='')
ttk.Entry(topFrame, width=8,
textvariable=self.strTime).pack(side=tk.LEFT, padx=4)
# 操作类型
self.operateTypeStr = tk.StringVar()
cmbOperateType = ttk.Combobox(topFrame,
width=8,
textvariable=self.operateTypeStr,
state='readonly')
# Adding combobox drop down list
cmbOperateType['values'] = ('买入', '卖出')
cmbOperateType.current(0)
cmbOperateType.pack(side=tk.LEFT, padx=4)
cmbOperateType.bind("<<ComboboxSelected>>",
self.on_cmb_operate_type_select)
topFrame2 = tk.Frame(self)
tk.Label(topFrame2, text="价格：").pack(side=tk.LEFT, padx=4, pady=2)
self.strPrice = tk.StringVar(value='')
ttk.Entry(topFrame2, width=8,
textvariable=self.strPrice).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame2, text="数量：").pack(side=tk.LEFT)
self.strAmount = tk.StringVar(value='')
ttk.Entry(topFrame2, width=8,
textvariable=self.strAmount).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame2, text="目标价位：").pack(side=tk.LEFT)
self.strTargetPrice = tk.StringVar(value='')
ttk.Entry(topFrame2, width=8,
textvariable=self.strTargetPrice).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame2, text="止损价位：").pack(side=tk.LEFT)
self.strStopPrice = tk.StringVar(value='')
ttk.Entry(topFrame2, width=8,
textvariable=self.strStopPrice).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame2, text="收益比：").pack(side=tk.LEFT)
self.strProfitRate = tk.StringVar(value='')
ttk.Entry(topFrame2, width=8,
textvariable=self.strProfitRate).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame, text="盈利：").pack(side=tk.LEFT)
self.strProfit = tk.StringVar(value='')
ttk.Entry(topFrame, width=8,
textvariable=self.strProfit).pack(side=tk.LEFT, padx=4)
tk.Label(topFrame, text="备注：").pack(side=tk.LEFT)
self.text = tk.Text(topFrame, width=50, height=4)
self.text.pack(side=tk.LEFT, padx=4)
tk.Button(topFrame2, text="添加", width=10,
command=self.on_btn_add).pack(side=tk.LEFT, padx=4)
topFrame.pack(side=tk.TOP, fill=tk.BOTH)
topFrame2.pack(side=tk.TOP, fill=tk.BOTH)
# 列表框
self.frameReport = tk.Frame(self)
self.sheet = Sheet(self.frameReport)
self.sheet.enable_bindings((
"single_select", # "single_select" or "toggle_select"
"drag_select", # enables shift click selection as well
"column_drag_and_drop",
"row_drag_and_drop",
"column_select",
"row_select",
"column_width_resize",
"double_click_column_resize",
"row_width_resize",
"column_height_resize",
"arrowkeys",
"row_height_resize",
"double_click_row_resize",
"right_click_popup_menu",
"rc_select",
"rc_insert_column",
"rc_delete_column",
"rc_insert_row",
"rc_delete_row",
"hide_columns",
"copy",
"cut",
"paste",
"delete",
"undo",
"edit_cell"))
self.sheet.pack(side=tk.TOP, fill=tk.BOTH, expand=1)
self.frameReport.pack(side=tk.TOP, fill=tk.BOTH, expand=1, pady=4)
self.sheet.headers([
'日期', '编码', '名称', '操作时间', '操作方向', '价格', '数量', '目标价位', '止损', '收益比',
'盈利', '备注'
])
self.sheet.refresh()
def on_btn_main_page(self):
self.controller.show_frame('PageMain')
pass
def on_btn_add(self):
self.sheet.insert_row(values=(self.str_date.get(), self.strCode.get(),
self.strName.get(), self.strTime.get(),
self.operateTypeStr.get(),
self.strPrice.get(),
self.strAmount.get(),
self.strTargetPrice.get(),
self.strStopPrice.get(),
self.strProfitRate.get(),
self.strProfit.get(),
self.text.get('0.0', tk.END)))
self.sheet.refresh()
pass
def on_cmb_operate_type_select(self, event):
if not self.operateTypeStr.get():
pass
# self.nameEntered.delete(0, tk.END)
# self.nameEntered.insert(0, self.measureTypeStr.get())
pass
def tree_solution_selected(self, selected):
print('tree_solution_selected items:', selected)
pass
class demo(tk.Tk):
def __init__(self):
tk.Tk.__init__(self)
self.data = [["Average Motor Stator Current", 0],
["Average Motor Phase Voltage", 0], ["Target Torque", 0],
["Motor Actual Torque ", 0], ["Steering Angle", 0],
["Proportional Speed Limit ", 0],
["Motor RPM ", 0],
["Calculated Battery Current ", 0],
["Controller Capacitor Voltage", 0],
["Throttle input ", 0], ["Motor Temp ", 0],
["Controller Temperature ", 0],
["Distance travelled ", 0],
["Forward switch ", 0],
["Reverse Switch ", 0], ["Seat Switch ", 0]]
self.grid_columnconfigure(0, weight=1)
self.grid_rowconfigure(0, weight=1)
self.frame = tk.Frame(self)
self.frame.grid_columnconfigure(0, weight=1)
self.frame.grid_rowconfigure(0, weight=1)
self.sheet = Sheet(
self.frame,
page_up_down_select_row=True,
#empty_vertical = 0,
column_width=150,
startup_select=(0, 1, "rows"),
data=self.data,
theme="black",
height=400, #height and width arguments are optional
width=360, #For full startup arguments see DOCUMENTATION.md
)
self.sheet.enable_bindings((
"single_select", #"single_select" or "toggle_select"
"drag_select", #enables shift click selection as well
"column_drag_and_drop",
"row_drag_and_drop",
"column_select",
"row_select",
"column_width_resize",
"double_click_column_resize",
# "row_width_resize",
# "column_height_resize",
"arrowkeys",
"row_height_resize",
"double_click_row_resize",
"right_click_popup_menu",
"rc_select",
"rc_insert_column",
"rc_delete_column",
"rc_insert_row",
"rc_delete_row",
"hide_columns",
"copy",
"cut",
"paste",
"delete",
"undo",
"edit_cell"))
self.frame.grid(row=0, column=0, sticky="nswe")
self.sheet.grid(row=0, column=0, sticky="nswe")
self.sheet.headers(
(f"Header {c}" for c in range(2))) #any iterable works
self.sheet.headers("CAN Data", 0)
self.sheet.headers("CAN Values", 1)
def all_extra_bindings(self, event):
print(event)
def begin_edit_cell(self, event):
print(event) # event[2] is keystroke
return event[
2] # return value is the text to be put into cell edit window
def end_edit_cell(self, event):
print(event)
def window_resized(self, event):
pass
#print (event)
def mouse_motion(self, event):
region = self.sheet.identify_region(event)
row = self.sheet.identify_row(event, allow_end=False)
column = self.sheet.identify_column(event, allow_end=False)
print(region, row, column)
def deselect(self, event):
print(event, self.sheet.get_selected_cells())
def rc(self, event):
print(event)
def cell_select(self, response):
#print (response)
pass
def shift_select_cells(self, response):
print(response)
def drag_select_cells(self, response):
pass
#print (response)
def ctrl_a(self, response):
print(response)
def row_select(self, response):
print(response)
def shift_select_rows(self, response):
print(response)
def drag_select_rows(self, response):
pass
#print (response)
def column_select(self, response):
print(response)
#for i in range(50):
# self.sheet.create_dropdown(i, response[1], values=[f"{i}" for i in range(200)], set_value="100",
# destroy_on_select = False, destroy_on_leave = False, see = False)
#print (self.sheet.get_cell_data(0, 0))
#self.sheet.refresh()
def shift_select_columns(self, response):
print(response)
def drag_select_columns(self, response):
pass
# print (response)
def null_print(self):
print("NULL Printing\n")
class columnselection(tk.Frame):
def __init__(self,parent,C):
tk.Frame.__init__(self,parent)
self.C = C
self.parent_cols = []
self.rowlen = 0
self.grid_rowconfigure(0,weight=1)
self.grid_rowconfigure(1,weight=1)
self.grid_columnconfigure(1,weight=1)
self.flattened_choices = flattened_base_ids_choices(self,command=self.flattened_mode_toggle)
self.flattened_choices.grid(row=1,column=0,pady=(0,5),sticky="wnse")
self.flattened_selector = flattened_column_selector(self)
self.selector = id_and_parent_column_selector(self)
self.selector.grid(row=1,column=0,sticky="wnse")
self.sheetdisplay = Sheet(self,
theme = self.C.theme,
header_font = ("Calibri", 13, "normal"),
outline_thickness=1)
self.sheetdisplay.enable_bindings("enable_all")
self.sheetdisplay.extra_bindings([("row_index_drag_drop", self.drag_row),
("column_header_drag_drop", self.drag_col),
("ctrl_x", self.reset_selectors),
("delete_key", self.reset_selectors),
("rc_delete_column", self.reset_selectors),
("rc_delete_row", self.reset_selectors),
("rc_insert_column", self.reset_selectors),
("rc_insert_row", self.reset_selectors),
("ctrl_v", self.reset_selectors),
("ctrl_z", self.reset_selectors),
("edit_cell", self.edit_cell_in_sheet)
])
self.sheetdisplay.grid(row=0,column=1,rowspan=3,sticky="nswe")
self.cont_ = button(self,
text="Build tree with selections ",
style="TF.Std.TButton",command=self.try_to_build_tree)
self.cont_.grid(row=2,column=0,sticky="wns",padx=10,pady=(10, 50))
self.cont_.config(width=40)
self.flattened_selector.grid(row=0,column=0,pady=(0,9),sticky="nswe")
self.selector.grid_forget()
self.selector.grid(row=0,column=0,sticky="nswe")
self.flattened_selector.grid_forget()
def flattened_mode_toggle(self):
x = self.flattened_choices.get_choices()[0]
if x:
self.flattened_selector.grid(row=0,column=0,pady=(0,9),sticky="nswe")
self.selector.grid_forget()
else:
self.selector.grid(row=0,column=0,sticky="nswe")
self.flattened_selector.grid_forget()
def drag_col(self, event):
colsiter = sorted(event[1])
c = event[3]
stins = colsiter[0]
endins = colsiter[-1] + 1
self.selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]])
self.flattened_selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]])
self.selector.detect_id_col()
self.selector.detect_par_cols()
def drag_row(self, event):
rowsiter = sorted(event[1])
r = event[3]
stins = rowsiter[0]
endins = rowsiter[-1] + 1
self.selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]])
self.flattened_selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]])
self.selector.detect_id_col()
self.selector.detect_par_cols()
def reset_selectors(self, event = None):
idcol = self.selector.get_id_col()
parcols = self.selector.get_par_cols()
ancparcols = self.flattened_selector.get_par_cols()
self.selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]] if self.sheetdisplay.get_sheet_data() else [])
self.flattened_selector.set_columns([h for h in self.sheetdisplay.get_sheet_data()[0]] if self.sheetdisplay.get_sheet_data() else [])
try:
if idcol is not None and self.sheetdisplay.get_sheet_data():
self.selector.set_id_col(idcol)
except:
pass
try:
if parcols and self.sheetdisplay.get_sheet_data():
self.selector.set_par_cols(parcols)
except:
pass
try:
if ancparcols and self.sheetdisplay.get_sheet_data():
self.flattened_selector.set_par_cols(ancparcols)
except:
pass
def edit_cell_in_sheet(self, event = None):
idcol = self.selector.get_id_col()
parcols = self.selector.get_par_cols()
ancparcols = self.flattened_selector.get_par_cols()
if event[1] == idcol or event[1] in parcols or event[1] in ancparcols or event[0] == 0:
self.reset_selectors()
def enable_widgets(self):
self.selector.enable_me()
self.flattened_selector.enable_me()
self.flattened_choices.enable_me()
self.cont_.config(state="normal")
self.sheetdisplay.basic_bindings(True)
self.sheetdisplay.enable_bindings("enable_all")
self.sheetdisplay.extra_bindings([("row_index_drag_drop", self.drag_row),
("column_header_drag_drop", self.drag_col),
("ctrl_x", self.ctrl_x_in_sheet),
("delete_key", self.del_in_sheet),
("rc_delete_column", self.del_in_sheet),
("rc_delete_row", self.del_in_sheet),
("rc_insert_column", self.reset_selectors),
("rc_insert_row", self.reset_selectors),
("ctrl_v", self.ctrl_v_in_sheet),
("ctrl_z", self.ctrl_z_in_sheet),
("edit_cell", self.edit_cell_in_sheet)
])
def disable_widgets(self):
self.selector.disable_me()
self.flattened_selector.disable_me()
self.flattened_choices.disable_me()
self.cont_.config(state="disabled")
self.sheetdisplay.basic_bindings(False)
self.sheetdisplay.disable_bindings("disable_all")
self.sheetdisplay.extra_bindings("unbind_all")
self.sheetdisplay.unbind("<Delete>")
def populate(self, columns, non_tsrgn_xl_file = False):
self.sheetdisplay.deselect("all")
self.non_tsrgn_xl_file = non_tsrgn_xl_file
self.rowlen = len(columns)
self.selector.set_columns([h for h in self.C.treeframe.sheet[0]])
self.flattened_selector.set_columns([h for h in self.C.treeframe.sheet[0]])
self.C.treeframe.sheet = self.sheetdisplay.set_sheet_data(data = self.C.treeframe.sheet,
redraw=True)
self.sheetdisplay.headers(newheaders=0)
if len(self.C.treeframe.sheet) < 3000:
self.sheetdisplay.set_all_cell_sizes_to_text()
self.selector.detect_id_col()
self.selector.detect_par_cols()
self.flattened_selector.detect_par_cols()
self.C.show_frame("columnselection")
def try_to_build_tree(self):
baseids, order, delcols = self.flattened_choices.get_choices()
if baseids:
hiers = list(self.flattened_selector.get_par_cols())
if len(hiers) < 2:
return
else:
hiers = list(self.selector.get_par_cols())
if not hiers:
return
idcol = self.selector.get_id_col()
if idcol in hiers or idcol is None:
return
self.C.status_bar.change_text("Loading... ")
self.C.disable_at_start()
self.C.treeframe.sheet = self.sheetdisplay.get_sheet_data()
if baseids:
if order == "Order: Base → Top":
idcol = hiers.pop(0)
elif order == "Order: Top → Base":
idcol = hiers.pop(len(hiers) - 1)
self.C.treeframe.sheet[:] = [row + list(repeat("",self.rowlen - len(row))) if len(row) < self.rowlen else row for row in self.C.treeframe.sheet]
self.C.treeframe.sheet, self.rowlen, idcol, newpc = self.C.treeframe.treebuilder.convert_flattened_to_normal(data = self.C.treeframe.sheet,
idcol = idcol,
parcols = hiers,
rowlen = self.rowlen,
order = order,
delcols = delcols,
warnings = self.C.treeframe.warnings)
hiers = [newpc]
self.C.treeframe.headers = [Header(name) for name in self.C.treeframe.fix_heads(self.C.treeframe.sheet.pop(0),self.rowlen)]
self.C.treeframe.ic = idcol
self.C.treeframe.hiers = hiers
self.C.treeframe.pc = hiers[0]
self.C.treeframe.row_len = int(self.rowlen)
self.C.treeframe.set_metadata(headers = True)
self.C.treeframe.sheet, self.C.treeframe.nodes, self.C.treeframe.warnings = self.C.treeframe.treebuilder.build(self.C.treeframe.sheet,
self.C.treeframe.new_sheet,
self.C.treeframe.row_len,
self.C.treeframe.ic,
self.C.treeframe.hiers,
self.C.treeframe.nodes,
self.C.treeframe.warnings)
self.C.treeframe.populate(non_tsrgn_xl_file = self.non_tsrgn_xl_file)
self.C.treeframe.show_warnings(str(self.C.open_dict['filepath']),str(self.C.open_dict['sheet']))
