2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Updated build_outer_panes() *BROKEN*

Browse source 
* Script broken until update_progress() is fixed
This commit is contained in:
2Shirt 2018-08-01 23:56:25 -07:00
parent c568668fd0
commit d474c8b5d4
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 17 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
.bin/Scripts/functions/ddrescue.py
View file

@ -246,11 +246,14 @@ class ImageObj(BaseObj):
class RecoveryState(): class RecoveryState():
"""Object to track BlockPair objects and overall state.""" """Object to track BlockPair objects and overall state."""
def __init__(self, mode): def __init__(self, mode, source, dest):
self.mode = mode.lower()
self.source_name = source.name
self.dest_name = dest.name
self.block_pairs = [] self.block_pairs = []
self.current_pass = 0 self.current_pass = 0
self.finished = False self.finished = False
self.mode = mode.lower() self.progress_out = '{}/progress.out'.format(global_vars['LogDir'])
self.rescued = 0 self.rescued = 0
self.resumed = False self.resumed = False
self.started = False self.started = False
@ -338,7 +341,7 @@ class RecoveryState():
# Functions # Functions
def build_outer_panes(source, dest): def build_outer_panes(state):
"""Build top and side panes.""" """Build top and side panes."""
clear_screen() clear_screen()
result = run_program(['tput', 'cols']) result = run_program(['tput', 'cols'])
@ -346,12 +349,12 @@ def build_outer_panes(source, dest):
(int(result.stdout.decode().strip()) - SIDE_PANE_WIDTH) / 2) - 2 (int(result.stdout.decode().strip()) - SIDE_PANE_WIDTH) / 2) - 2
# Top panes # Top panes
source_str = source.name source_str = state.source_name
if len(source_str) > width: if len(source_str) > width:
source_str = '{}...'.format(source_str[:width-3]) source_str = '{}...'.format(source_str[:width-3])
dest_str = dest.name dest_str = state.dest_name
if len(dest_str) > width: if len(dest_str) > width:
if dest.is_dev(): if state.mode == 'clone':
dest_str = '{}...'.format(dest_str[:width-3]) dest_str = '{}...'.format(dest_str[:width-3])
else: else:
dest_str = '...{}'.format(dest_str[-width+3:]) dest_str = '...{}'.format(dest_str[-width+3:])
@ -375,12 +378,11 @@ def build_outer_panes(source, dest):
**COLORS)) **COLORS))
# Side pane # Side pane
# TODO FIX IT update_progress(state)
#update_progress(source) tmux_splitw(
#tmux_splitw( '-dhl', str(SIDE_PANE_WIDTH),
# '-dhl', '{}'.format(SIDE_PANE_WIDTH), 'watch', '--color', '--no-title', '--interval', '1',
# 'watch', '--color', '--no-title', '--interval', '1', 'cat', state.progress_out)
# 'cat', source['Progress Out'])
def create_path_obj(path): def create_path_obj(path):
@ -591,12 +593,11 @@ def menu_ddrescue(source_path, dest_path, run_mode):
dest.self_check() dest.self_check()
# Build BlockPairs # Build BlockPairs
state = RecoveryState(run_mode) state = RecoveryState(run_mode, source, dest)
if run_mode == 'clone': if run_mode == 'clone':
state.add_block_pair(source, dest) state.add_block_pair(source, dest)
else: else:
source_parts = select_parts(source) for part in select_parts(source):
for part in source_parts:
state.add_block_pair(part, dest) state.add_block_pair(part, dest)
# Update state # Update state
@ -617,7 +618,7 @@ def menu_ddrescue(source_path, dest_path, run_mode):
raise GenericAbort() raise GenericAbort()
# Main menu # Main menu
build_outer_panes(source, dest) build_outer_panes(state)
# TODO Fix # TODO Fix
#menu_main(source, dest) #menu_main(source, dest)
pause('Fake Main Menu... ') pause('Fake Main Menu... ')