main_linux.py

#! /use/bin/python3

#######################################################################
# THE DISPLAY MODULE
#######################################################################

# This module sets up the GUI, using Python bindings for GTK3

#######################################################################
# External modules we use
#######################################################################

import curses
from curses import wrapper

import os

import sys

#######################################################################
# Modules of our own project
#######################################################################

# The initialization module
import init

# The query module
import query

# Global variables
import global_vars

#######################################################################
# Classes
#######################################################################

#######################################################################
# Functions
#######################################################################


def open_file(file_name):
    abs_name = global_vars._DOCS_PATH + file_name
    if os.path.exists(abs_name) and os.path.isfile(abs_name):
        if sys.platform == "win32":
            os.system(abs_name)
        elif sys.platform == "linux" or sys.platform == "linux2":
            os.system('%s %s' % (os.getenv('EDITOR'), abs_name))


def display_init(stdscr):
    # Start
    stdscr.clear()
    curses.noecho()
    curses.cbreak()
    stdscr.keypad(True)

    header = ["***********************************************************************",
              global_vars._NAME,
              "AUTHORS : " + ', '.join(global_vars._AUTHORS),
              "***********************************************************************"]

    instructions = ["INSTRUCTIONS :",
                    "1) Type to search",
                    "2) Use the arrow keys to select the required file",
                    "3) DO NO resize the window",
                    "4) Press Enter when a file is selected to open it.",
                    "5) Press Esc to exit."]

    directory = "DIRECTORY : {}".format(global_vars._DOCS_PATH)

    prompt = "Search: "
    str_1 = ""
    str_2 = ""

    stdscr.move(5, 5)

    query.query("Hello")

    search_linum = 14
    display_linum = 16

    res = []

    selected = 0

    while (True):
        y, x = stdscr.getmaxyx()
        if x < 80 and y < 30:
            curses.resizeterm(30, 80)
        elif x < 80:
            curses.resizeterm(y, 80)
        elif y < 30:
            curses.resizeterm(30, x)
        stdscr.refresh()

        stdscr.clear()

        stdscr.addstr(0, 5, header[0])
        stdscr.addstr(1, 5, header[1])
        stdscr.addstr(2, 5, header[2])
        stdscr.addstr(3, 5, header[3])

        stdscr.addstr(5, 5, instructions[0])
        stdscr.addstr(6, 5, instructions[1])
        stdscr.addstr(7, 5, instructions[2])
        stdscr.addstr(8, 5, instructions[3])
        stdscr.addstr(9, 5, instructions[4])
        stdscr.addstr(10, 5, instructions[5])

        stdscr.addstr(12, 5, directory)

        linum = display_linum

        # Display the documents. Highlight the one selected
        if len(res) > 0:
            for (doc, score) in res[0:selected]:
                stdscr.addstr(linum, 7, doc)
                linum += 1
            stdscr.addstr(linum, 7, res[selected][0], curses.A_REVERSE)
            linum += 1
            for (doc, score) in res[selected + 1:10]:
                stdscr.addstr(linum, 7, doc)
                linum += 1

        stdscr.addstr(search_linum, 5, prompt + str_1 + str_2)
        stdscr.move(search_linum, 5 + len(prompt) + len(str_1))
        stdscr.refresh()

        ch = stdscr.getch()

        # Deal with resize first
        if ch == curses.KEY_RESIZE:
            pass
        # Backspace
        elif ch == 127:
            str_1 = str_1[:-1]
        # Enter
        elif ch == 10:
            open_file(res[selected][0])
            stdscr.clear()
            curses.noecho()
            curses.cbreak()
            stdscr.keypad(True)
        # Excape
        elif ch == 27:
            break
        # Delete
        elif ch == curses.KEY_DC:
            str_2 = str_2[1:]
        # Left arrow key
        elif ch == curses.KEY_LEFT:
            if len(str_1) > 0:
                str_2 = str_1[-1] + str_2
                str_1 = str_1[:-1]
        # Right arrow key
        elif ch == curses.KEY_RIGHT:
            if len(str_2) > 0:
                str_1 = str_1 + str_2[0]
                str_2 = str_2[1:]
        # Down arrow key
        elif ch == curses.KEY_DOWN:
            if selected < 9 and selected < len(res) - 1:
                selected += 1
                continue
        # Up arrow key
        elif ch == curses.KEY_UP:
            if selected > 0:
                selected -= 1
                continue
        # Any other character
        else:
            str_1 += chr(ch)
        selected = 0
        res = query.query(str_1 + str_2)

    # End
    curses.nocbreak()
    stdscr.keypad(False)
    curses.echo()
    curses.endwin()


def default_main():
    while True:
        intext = input()
        if intext == "quit()":
            quit()
        docs = query.query(intext)
        for doc in docs[:10]:
            print(doc[0])


if __name__ == "__main__":
    init.init()
    if sys.platform == "linux" or sys.platform == "linux2":
        wrapper(display_init)
    elif sys.platform == "win32":
        default_main()