from __future__ import absolute_import

import code

import logging

import sys

from optparse import Option

import curtsies

import curtsies.window

import curtsies.input

import curtsies.events

from bpython.curtsiesfrontend.repl import Repl

from bpython.curtsiesfrontend.coderunner import SystemExitFromCodeGreenlet

from bpython import args as bpargs

from bpython.translations import _

from bpython.importcompletion import find_iterator

from bpython.curtsiesfrontend import events as bpythonevents

logger = logging.getLogger(__name__)

repl = None # global for `from bpython.curtsies import repl`

# WARNING Will be a problem if more than one repl is ever instantiated this way

def main(args=None, locals_=None, banner=None):

config, options, exec_args = bpargs.parse(args, (

'curtsies options', None, [

Option('--log', '-L', action='store_true',

help=_("log debug messages to bpython.log")),

Option('--type', '-t', action='store_true',

help=_("enter lines of file as though interactively typed")),

]))

if options.log:

handler = logging.FileHandler(filename='bpython.log')

logging.getLogger('curtsies').setLevel(logging.WARNING)

logging.getLogger('curtsies').addHandler(handler)

logging.getLogger('curtsies').propagate = False

logging.getLogger('bpython').setLevel(logging.WARNING)

logging.getLogger('bpython').addHandler(handler)

logging.getLogger('bpython').propagate = False

else:

logging.getLogger('bpython').setLevel(logging.ERROR)

logging.getLogger('curtsies').setLevel(logging.ERROR)

interp = None

paste = None

if exec_args:

if not options:

raise ValueError("don't pass in exec_args without options")

exit_value = 0

if options.type:

paste = curtsies.events.PasteEvent()

sourcecode = open(exec_args[0]).read()

paste.events.extend(sourcecode)

else:

try:

interp = code.InteractiveInterpreter(locals=locals_)

bpargs.exec_code(interp, exec_args)

except SystemExit, e:

exit_value = e.args

if not options.interactive:

raise SystemExit(exit_value)

else:

sys.path.insert(0, '') # expected for interactive sessions (vanilla python does it)

print(bpargs.version_banner())

mainloop(config, locals_, banner, interp, paste, interactive=(not exec_args))

def mainloop(config, locals_, banner, interp=None, paste=None, interactive=True):

with curtsies.input.Input(keynames='curtsies', sigint_event=True) as input_generator:

with curtsies.window.CursorAwareWindow(

sys.stdout,

sys.stdin,

keep_last_line=True,

hide_cursor=False,

extra_bytes_callback=input_generator.unget_bytes) as window:

request_refresh = input_generator.event_trigger(bpythonevents.RefreshRequestEvent)

schedule_refresh = input_generator.scheduled_event_trigger(bpythonevents.ScheduledRefreshRequestEvent)

request_reload = input_generator.threadsafe_event_trigger(bpythonevents.ReloadEvent)

def on_suspend():

window.__exit__(None, None, None)

input_generator.__exit__(None, None, None)

def after_suspend():

input_generator.__enter__()

window.__enter__()

global repl # global for easy introspection `from bpython.curtsies import repl`

with Repl(config=config,

locals_=locals_,

request_refresh=request_refresh,

schedule_refresh=schedule_refresh,

request_reload=request_reload,

get_term_hw=window.get_term_hw,

get_cursor_vertical_diff=window.get_cursor_vertical_diff,

banner=banner,

interp=interp,

interactive=interactive,

orig_tcattrs=input_generator.original_stty,

on_suspend=on_suspend,

after_suspend=after_suspend) as repl:

repl.height, repl.width = window.t.height, window.t.width

def process_event(e):

"""If None is passed in, just paint the screen"""

try:

if e is not None:

repl.process_event(e)

except (SystemExitFromCodeGreenlet, SystemExit) as err:

array, cursor_pos = repl.paint(about_to_exit=True, user_quit=isinstance(err, SystemExitFromCodeGreenlet))

scrolled = window.render_to_terminal(array, cursor_pos)

repl.scroll_offset += scrolled

raise

else:

array, cursor_pos = repl.paint()

scrolled = window.render_to_terminal(array, cursor_pos)

repl.scroll_offset += scrolled

if paste:

process_event(paste)

process_event(None) # do a display before waiting for first event

for _ in find_iterator:

e = input_generator.send(0)

if e is not None:

process_event(e)

for e in input_generator:

process_event(e)

if __name__ == '__main__':

sys.exit(main())