Highlighting and Selecting text with Python curses

Neal picture Neal · Jul 24, 2011 · Viewed 8.2k times · Source

This is my first post to stack overflow. I've been lurking this site for information for years, and it's always helpful, so I thought that I would post my first question.

I've been searching for some similar examples, but can't seem to find anything.

Ultimately, I'm trying to write a simple text ui for finding false positives and false negatives in a text extraction program. The false positive module is a simple yes/no selection, and displaying colored text and using getch() is easy enough. The False negative portion however, is a becoming difficult.

So this is what I want to do:

  1. Display a string onto the screen (forget scrolling for now.... ugh) It will only be a string with no newlines.
  2. The user sees the text, and pushes 'h' to enter highlight mode.
  3. The user can then control the cursor to move it over a portion of the text (still displayed), and select 'v' to begin highlighting (I trying to make this as close to vi as I can)
  4. The user then uses control keys (arrows, hjkl) to move the cursor from a starting point to and end point, highlighting the text on the way. This highlighted portion should be a word which is a false negative
  5. The user presses some key ('y' probably), and the selected text is colored, or stays highlighted, and the highlighted text is save to some variable that I'll handle later.
  6. The user exits highlight mode and proceeds.

Any ideas to even START? I'm trying simple things like keeping text on the screen and moving the cursor around, but to no avail.

I'm aware of the curses.textpad.TextBox() module, but it performs editing like insertion and deletion, which I don't want to do. Perhaps there is a way to disable it.

I have other questions, but I'll keep this specific for now.

Thanks!!

Neal

Edit: To be more specific, I'm not looking for help writing the whole program, just help moving the cursor over displayed text, highlighting it, and selecting it and saVing it in a variable.

Answer

Neal picture Neal · Jul 27, 2011

I would like to update this question in case anybody else is searching the web for this and stumbles upon this question.

Okay, so the answer was actually quite simple and required reading all the functions listed on the python curses documentation.

What I did was make a 3 state machine: State 1: Normal mode (displays text only), State 2: highlight mode, allow the cursor to move around the window, and State 3: highlighter mode, which gives only limited movement from left to right over texts and highlights the text as it's moving.

To accomplish this task, it just takes some basic curses function calls.

I made separate windows, but I'll just assume a single window for explaination.

Do display text in the window, stick with the:

window.addstr()
window.refresh()

For moving the cursor around:

#get current cursor position
curr_y, curr_x = window.getyx()

# depending on direction, update the cursor with
next_y, next_x = get_next_direction()

# place cursor in new position
window.move(next_y, next_x)

window.refresh()

Once the cursor is over the starting point for highlighting, press 'v' enter the highlighter state, and limit the movement of the cursor, change the attribute of the selected text:

# get current cursor position
curr_y, curr_x = window.getyx()

# save position for later use
start_y = curr_y; start_x = curr_x

# loop until 'v' is pressed again
while highlight_state:
    # change the attribute of the current character, for 1 character only, to reverse
    window.chgat(curr_y,curr_x, 1, curses.A_REVERSE)
    curr_y, curr_x = get_next_direction()

# save end state, this is buggy obviously, but you get the point
end_y = curr_y; end_x = curr_X

Now extract that information from start to end

# get integer representation of char at positiong

outstr = ''
#from start to end
char_as_int = windows.inch(y,x)
char = char_as_int & 0000FF
attr = char_as_int & FFFF00 #not useful here, but maybe later

outstr += char

That's it! I also tried another way to save the highlighted material which was basically transformat the x,y coordinates to the index of the string that was being display, but that let to issue in string representation (newlines, tabs, and so on), plus was just harder to do.

If anyone has comments of a more efficient/ cleaner method, please respond!