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 EmailTab(tk.Frame): def __init__(self, master): self.emailHandler = EmailSheetHandler(FILENAME) tk.Frame.__init__(self, master) self.grid_columnconfigure(0, weight = 1) self.grid_rowconfigure(0, weight = 1) self.frame = ttk.Frame(self, height=450, width=400) 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 = 300, startup_select = (0,1,"rows"), data=self.populate_sheet(), total_columns = 3, #if you want to set empty sheet dimensions at startup height = 450, #height and width arguments are optional width = 600 #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", "arrowkeys", "row_height_resize", "double_click_row_resize", "right_click_popup_menu", "rc_select", "edit_cell")) self.sheet.extra_bindings("end_edit_cell", self.end_edit_cell) self.frame.grid(row = 0, column = 0, sticky = "nsw") self.sheet.grid(row = 0, column = 0, sticky = "nw") self.email_button_frame = ttk.Frame(self.frame, padding=(3,3,12,12), borderwidth=5, width=100, heigh=200) self.email_add_button = tk.Button(self.email_button_frame, text="Add email", command=self.add_email) self.email_merge_button = tk.Button(self.email_button_frame, text="Merge email (with file)", command=self.merge_mails) self.email_update_button = tk.Button(self.email_button_frame, text="Update file with table", command=self.email_file_update ) self.email_button_frame.grid(row=0, column=2, sticky="nswe") self.email_add_button.grid(row=1, column=1) self.email_merge_button.grid(row=2, column=1) self.email_update_button.grid(row=3, column=1) # print(self.sheet.get_sheet_data(get_index=0, get_header=0)) def end_edit_cell(self, event): print(event) print(self.sheet.get_cell_data(event[0], event[1 ])) def add_email(self): package = [] AddDialogue(package, "Add Email", "Add Email") print("Rsult: ", package) self.emailHandler.append_to_file(package) self.update_table() def email_file_update(self): self.emailHandler.update_email_file(self.sheet.get_sheet_data()) def update_table(self): data = self.populate_sheet() self.sheet.set_sheet_data(data) def merge_mails(self): file = filedialog.askopenfile().name print(file) # TODO: handle merge print(self.emailHandler.validate(file)) def populate_sheet(self): self.data = [] try: with open(FILENAME, 'r') as f: L = f.readlines() self.data = [i.strip("\n").split(",") for i in L] # for i in self.data: i.append("50") return self.data except FileNotFoundError: return self.data
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']))