Initial table mode added. Probably bugs.
This commit is contained in:
Storm Dragon
@ -16,12 +16,350 @@ class TableManager:
|
||||
self.noOfHeadLineColumns = 0
|
||||
self.headColumnSep = ""
|
||||
self.rowColumnSep = ""
|
||||
self.headerRow = []
|
||||
self.tableMode = False
|
||||
self.currentColumn = 0
|
||||
self.currentRow = 0
|
||||
|
||||
def initialize(self, environment):
|
||||
self.env = environment
|
||||
|
||||
def shutdown(self):
|
||||
pass
|
||||
|
||||
def is_table_mode(self):
|
||||
return self.tableMode
|
||||
|
||||
def set_table_mode(self, active):
|
||||
self.tableMode = active
|
||||
if not active:
|
||||
self.clear_header_row()
|
||||
else:
|
||||
# Initialize current position when entering table mode
|
||||
self.sync_table_position_from_cursor()
|
||||
|
||||
def set_header_row_from_cursor(self):
|
||||
"""Set header row from current cursor position"""
|
||||
if not self.env["runtime"]["CursorManager"].is_review_mode():
|
||||
return False
|
||||
|
||||
cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
|
||||
if not cursor_pos:
|
||||
return False
|
||||
|
||||
# Get the line text at cursor position
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
|
||||
if not line_text:
|
||||
return False
|
||||
|
||||
# Parse the line into columns
|
||||
self.headerRow = self.parse_line_into_columns(line_text)
|
||||
return len(self.headerRow) > 0
|
||||
|
||||
def clear_header_row(self):
|
||||
"""Clear the stored header row"""
|
||||
self.headerRow = []
|
||||
|
||||
def has_header_row(self):
|
||||
"""Check if header row is set"""
|
||||
return len(self.headerRow) > 0
|
||||
|
||||
def get_column_header(self, column_index):
|
||||
"""Get header for a specific column (0-based)"""
|
||||
if column_index < 0:
|
||||
return f"Column {column_index + 1}"
|
||||
|
||||
if len(self.headerRow) == 0:
|
||||
return f"Column {column_index + 1}"
|
||||
|
||||
if column_index >= len(self.headerRow):
|
||||
return f"Column {column_index + 1}"
|
||||
|
||||
header = self.headerRow[column_index].strip()
|
||||
if not header: # If header is empty
|
||||
return f"Column {column_index + 1}"
|
||||
|
||||
return header
|
||||
|
||||
def get_column_count(self):
|
||||
"""Get number of columns based on header row"""
|
||||
return len(self.headerRow)
|
||||
|
||||
def parse_line_into_columns(self, line_text):
|
||||
"""Parse a line into columns using various separators"""
|
||||
if not line_text:
|
||||
return []
|
||||
|
||||
# Try different separators in order of preference
|
||||
separators = [',', '|', ';', '\t']
|
||||
|
||||
for sep in separators:
|
||||
if sep in line_text:
|
||||
columns = line_text.split(sep)
|
||||
if len(columns) > 1:
|
||||
return columns
|
||||
|
||||
# If no clear separator, try to detect aligned columns
|
||||
return self.detect_aligned_columns(line_text)
|
||||
|
||||
def detect_aligned_columns(self, line_text):
|
||||
"""Detect columns in space-aligned text"""
|
||||
if not line_text:
|
||||
return []
|
||||
|
||||
# Split on multiple spaces (2 or more)
|
||||
parts = re.split(r' +', line_text)
|
||||
if len(parts) > 1:
|
||||
return parts
|
||||
|
||||
# Fallback: treat as single column
|
||||
return [line_text]
|
||||
|
||||
def get_cell_content(self, line_text, column_index):
|
||||
"""Get content of a specific cell in a line"""
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if column_index < 0 or column_index >= len(columns):
|
||||
return ""
|
||||
return columns[column_index].strip()
|
||||
|
||||
def get_table_cell_info(self, cursor_pos):
|
||||
"""Get table cell information for speech output"""
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
|
||||
if not line_text:
|
||||
return None
|
||||
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns:
|
||||
return None
|
||||
|
||||
# Find which column we're in based on cursor x position
|
||||
column_index = self.get_column_index_from_x_position(line_text, cursor_pos["x"])
|
||||
|
||||
if column_index < 0 or column_index >= len(columns):
|
||||
return None
|
||||
|
||||
# Use header if available, otherwise generic column name
|
||||
if self.has_header_row():
|
||||
column_header = self.get_column_header(column_index)
|
||||
else:
|
||||
column_header = f"Column {column_index + 1}"
|
||||
|
||||
return {
|
||||
'column_index': column_index,
|
||||
'column_header': column_header,
|
||||
'cell_content': columns[column_index].strip(),
|
||||
'total_columns': len(columns),
|
||||
'row_number': cursor_pos["y"] + 1
|
||||
}
|
||||
|
||||
def get_column_index_from_x_position(self, line_text, x_pos):
|
||||
"""Determine which column an x position falls into"""
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns:
|
||||
return 0
|
||||
|
||||
# Handle CSV/delimited text
|
||||
if ',' in line_text or '|' in line_text or ';' in line_text or '\t' in line_text:
|
||||
current_pos = 0
|
||||
for i, column in enumerate(columns):
|
||||
column_end = current_pos + len(column)
|
||||
if current_pos <= x_pos <= column_end:
|
||||
return i
|
||||
current_pos = column_end + 1 # +1 for separator
|
||||
else:
|
||||
# Handle space-aligned text - find the column by position ranges
|
||||
current_pos = 0
|
||||
for i, column in enumerate(columns):
|
||||
# Find where this column starts in the original text
|
||||
if i == 0:
|
||||
column_start = 0
|
||||
else:
|
||||
# Find the column by searching from current position
|
||||
column_start = line_text.find(column.strip(), current_pos)
|
||||
if column_start == -1:
|
||||
column_start = current_pos
|
||||
|
||||
column_end = column_start + len(column.strip())
|
||||
if column_start <= x_pos <= column_end:
|
||||
return i
|
||||
current_pos = column_end
|
||||
|
||||
# If past the end, return last column
|
||||
return len(columns) - 1
|
||||
|
||||
def sync_table_position_from_cursor(self):
|
||||
"""Sync internal table position from current cursor position"""
|
||||
if not self.env["runtime"]["CursorManager"].is_review_mode():
|
||||
return
|
||||
|
||||
cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
|
||||
if cursor_pos:
|
||||
old_column = self.currentColumn
|
||||
self.currentRow = cursor_pos["y"]
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
|
||||
if line_text:
|
||||
self.currentColumn = self.get_column_index_from_x_position(line_text, cursor_pos["x"])
|
||||
else:
|
||||
self.currentColumn = 0
|
||||
|
||||
self.env["runtime"]["DebugManager"].write_debug_out(
|
||||
f"TableManager sync: old_column={old_column}, new_column={self.currentColumn}, cursor_x={cursor_pos['x']}, cursor_y={cursor_pos['y']}",
|
||||
debug.DebugLevel.INFO
|
||||
)
|
||||
|
||||
def move_to_next_column(self):
|
||||
"""Move to next column in table"""
|
||||
self.env["runtime"]["DebugManager"].write_debug_out(
|
||||
f"TableManager move_to_next_column: currentColumn={self.currentColumn}",
|
||||
debug.DebugLevel.INFO
|
||||
)
|
||||
|
||||
if not self.env["runtime"]["CursorManager"].is_review_mode():
|
||||
return None
|
||||
|
||||
cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
|
||||
if not cursor_pos:
|
||||
return None
|
||||
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
|
||||
if not line_text:
|
||||
return None
|
||||
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns:
|
||||
return None
|
||||
|
||||
self.env["runtime"]["DebugManager"].write_debug_out(
|
||||
f"TableManager: line_text='{line_text}', columns={columns}, currentColumn={self.currentColumn}",
|
||||
debug.DebugLevel.INFO
|
||||
)
|
||||
|
||||
# Don't sync from cursor position - maintain our own tracking
|
||||
# Check if we're already at the last column
|
||||
if self.currentColumn >= len(columns) - 1:
|
||||
# At end of line - return special indicator but keep position
|
||||
return {"at_end": True, "current_info": self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)}
|
||||
|
||||
# Move to next column
|
||||
self.currentColumn += 1
|
||||
|
||||
self.env["runtime"]["DebugManager"].write_debug_out(
|
||||
f"TableManager: moved to column {self.currentColumn}",
|
||||
debug.DebugLevel.INFO
|
||||
)
|
||||
|
||||
# Return info for the new column without moving cursor position
|
||||
return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
|
||||
|
||||
def move_to_prev_column(self):
|
||||
"""Move to previous column in table"""
|
||||
if not self.env["runtime"]["CursorManager"].is_review_mode():
|
||||
return None
|
||||
|
||||
cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
|
||||
if not cursor_pos:
|
||||
return None
|
||||
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(cursor_pos["y"])
|
||||
if not line_text:
|
||||
return None
|
||||
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns:
|
||||
return None
|
||||
|
||||
# Don't sync from cursor position - maintain our own tracking
|
||||
# Check if we're already at the first column
|
||||
if self.currentColumn <= 0:
|
||||
# At beginning of line - return special indicator but keep position
|
||||
return {"at_start": True, "current_info": self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)}
|
||||
|
||||
# Move to previous column
|
||||
self.currentColumn -= 1
|
||||
|
||||
# Return info for the new column without moving cursor position
|
||||
return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
|
||||
|
||||
def get_current_table_cell_info(self):
|
||||
"""Get current table cell info using internal position tracking"""
|
||||
if not self.env["runtime"]["CursorManager"].is_review_mode():
|
||||
return None
|
||||
|
||||
cursor_pos = self.env["runtime"]["CursorManager"].get_review_or_text_cursor()
|
||||
if not cursor_pos:
|
||||
return None
|
||||
|
||||
return self.get_table_cell_info_by_indices(cursor_pos["y"], self.currentColumn)
|
||||
|
||||
def get_table_cell_info_by_indices(self, row, column_index):
|
||||
"""Get table cell info for specific row and column indices"""
|
||||
line_text = self.env["runtime"]["ScreenManager"].get_line_text(row)
|
||||
if not line_text:
|
||||
return None
|
||||
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns or column_index < 0 or column_index >= len(columns):
|
||||
return None
|
||||
|
||||
# Always get column header (fallback to generic if needed)
|
||||
column_header = self.get_column_header(column_index)
|
||||
|
||||
cell_content = columns[column_index].strip()
|
||||
if not cell_content: # Handle empty cells
|
||||
cell_content = "blank"
|
||||
|
||||
return {
|
||||
'column_index': column_index,
|
||||
'column_header': column_header,
|
||||
'cell_content': cell_content,
|
||||
'total_columns': len(columns),
|
||||
'row_number': row + 1
|
||||
}
|
||||
|
||||
def get_column_start_position(self, line_text, column_index):
|
||||
"""Get the starting x position of a specific column"""
|
||||
columns = self.parse_line_into_columns(line_text)
|
||||
if not columns or column_index < 0 or column_index >= len(columns):
|
||||
return 0
|
||||
|
||||
# For CSV/delimited text - find the actual position in the original line
|
||||
if ',' in line_text or '|' in line_text or ';' in line_text or '\t' in line_text:
|
||||
# Determine the separator being used
|
||||
separator = ','
|
||||
if '|' in line_text and line_text.count('|') > line_text.count(','):
|
||||
separator = '|'
|
||||
elif ';' in line_text and line_text.count(';') > line_text.count(','):
|
||||
separator = ';'
|
||||
elif '\t' in line_text:
|
||||
separator = '\t'
|
||||
|
||||
# Find the position by splitting and calculating
|
||||
if column_index == 0:
|
||||
return 0
|
||||
|
||||
# Count characters up to the target column
|
||||
parts = line_text.split(separator)
|
||||
position = 0
|
||||
for i in range(column_index):
|
||||
if i < len(parts):
|
||||
position += len(parts[i]) + 1 # +1 for separator
|
||||
return position
|
||||
else:
|
||||
# For space-aligned text, find the actual position of the column
|
||||
if column_index == 0:
|
||||
return 0
|
||||
|
||||
# Find the column text in the line
|
||||
target_text = columns[column_index].strip()
|
||||
search_start = 0
|
||||
for i in range(column_index):
|
||||
prev_text = columns[i].strip()
|
||||
found_pos = line_text.find(prev_text, search_start)
|
||||
if found_pos != -1:
|
||||
search_start = found_pos + len(prev_text)
|
||||
|
||||
column_pos = line_text.find(target_text, search_start)
|
||||
return column_pos if column_pos != -1 else search_start
|
||||
|
||||
def reset_table_mode(self):
|
||||
self.set_head_line()
|
||||
@ -34,11 +372,21 @@ class TableManager:
|
||||
self.coun_no_of_head_columns()
|
||||
|
||||
def coun_no_of_head_columns(self):
|
||||
pass
|
||||
if self.headLine and self.headColumnSep:
|
||||
self.noOfHeadLineColumns = len(self.headLine.split(self.headColumnSep))
|
||||
|
||||
def search_for_head_column_sep(self, headLine):
|
||||
if " " in headLine:
|
||||
return " "
|
||||
"""Find the most likely column separator in a header line"""
|
||||
separators = [',', '|', ';', '\t']
|
||||
|
||||
for sep in separators:
|
||||
if sep in headLine:
|
||||
return sep
|
||||
|
||||
# Check for multiple spaces (aligned columns)
|
||||
if ' ' in headLine:
|
||||
return ' '
|
||||
|
||||
return ""
|
||||
|
||||
def set_row_column_sep(self, columnSep=""):
|
||||
|
||||
Reference in New Issue
Block a user