2023-03-24 06:01:57 +00:00
|
|
|
# Python imports
|
|
|
|
|
|
|
|
# Lib imports
|
|
|
|
import gi
|
|
|
|
gi.require_version('Gtk', '3.0')
|
|
|
|
gi.require_version('GtkSource', '4')
|
|
|
|
|
|
|
|
from gi.repository import Gtk
|
|
|
|
from gi.repository import GtkSource
|
|
|
|
from gi.repository import GObject
|
|
|
|
|
2023-03-25 06:00:39 +00:00
|
|
|
import jedi
|
2023-03-24 06:01:57 +00:00
|
|
|
from jedi.api import Script
|
|
|
|
|
|
|
|
# Application imports
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# FIXME: Find real icon names...
|
|
|
|
icon_names = {
|
|
|
|
'import': '',
|
|
|
|
'module': '',
|
|
|
|
'class': '',
|
|
|
|
'function': '',
|
|
|
|
'statement': '',
|
|
|
|
'param': ''
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
class Jedi:
|
2023-03-25 06:00:39 +00:00
|
|
|
def get_script(file, doc_text):
|
2023-03-24 06:01:57 +00:00
|
|
|
return Script(code = doc_text, path = file)
|
|
|
|
|
|
|
|
|
2023-03-25 06:00:39 +00:00
|
|
|
class PythonCompletionProvider(GObject.Object, GtkSource.CompletionProvider):
|
2023-03-24 06:01:57 +00:00
|
|
|
"""
|
2023-03-25 06:00:39 +00:00
|
|
|
This code is A python code completion plugin for Newton.
|
|
|
|
# NOTE: Some code pulled/referenced from here --> https://github.com/isamert/gedi
|
2023-03-24 06:01:57 +00:00
|
|
|
"""
|
2023-03-25 06:00:39 +00:00
|
|
|
__gtype_name__ = 'PythonProvider'
|
2023-03-24 06:01:57 +00:00
|
|
|
|
|
|
|
def __init__(self, file):
|
|
|
|
GObject.Object.__init__(self)
|
|
|
|
self._theme = Gtk.IconTheme.get_default()
|
|
|
|
self._file = file
|
|
|
|
|
|
|
|
def do_get_name(self):
|
2023-03-25 06:00:39 +00:00
|
|
|
return "Python Code Completion"
|
2023-03-24 06:01:57 +00:00
|
|
|
|
|
|
|
def get_iter_correctly(self, context):
|
2023-03-25 06:00:39 +00:00
|
|
|
return context.get_iter()[1] if isinstance(context.get_iter(), tuple) else context.get_iter()
|
2023-03-24 06:01:57 +00:00
|
|
|
|
|
|
|
def do_match(self, context):
|
|
|
|
iter = self.get_iter_correctly(context)
|
|
|
|
iter.backward_char()
|
2023-03-25 06:00:39 +00:00
|
|
|
|
2023-03-24 06:01:57 +00:00
|
|
|
buffer = iter.get_buffer()
|
|
|
|
if buffer.get_context_classes_at_iter(iter) != ['no-spell-check']:
|
|
|
|
return False
|
2023-03-25 06:00:39 +00:00
|
|
|
|
2023-03-24 06:01:57 +00:00
|
|
|
ch = iter.get_char()
|
|
|
|
if not (ch in ('_', '.') or ch.isalnum()):
|
|
|
|
return False
|
|
|
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
def do_get_priority(self):
|
|
|
|
return 1
|
|
|
|
|
|
|
|
def do_get_activation(self):
|
|
|
|
return GtkSource.CompletionActivation.INTERACTIVE
|
|
|
|
|
|
|
|
def do_populate(self, context):
|
2023-03-25 06:00:39 +00:00
|
|
|
# TODO: Maybe convert async?
|
|
|
|
it = self.get_iter_correctly(context)
|
|
|
|
buffer = it.get_buffer()
|
2023-03-24 06:01:57 +00:00
|
|
|
proposals = []
|
|
|
|
|
2023-03-25 06:00:39 +00:00
|
|
|
doc_text = buffer.get_text(buffer.get_start_iter(), buffer.get_end_iter(), False)
|
|
|
|
iter_cursor = buffer.get_iter_at_mark(buffer.get_insert())
|
|
|
|
linenum = iter_cursor.get_line() + 1
|
|
|
|
charnum = iter_cursor.get_line_index()
|
2023-03-24 06:01:57 +00:00
|
|
|
|
2023-03-25 06:00:39 +00:00
|
|
|
def create_generator():
|
|
|
|
for completion in Jedi.get_script(self._file, doc_text).complete(line = linenum, column = None, fuzzy = False):
|
|
|
|
comp_item = GtkSource.CompletionItem.new()
|
|
|
|
comp_item.set_label(completion.name)
|
|
|
|
comp_item.set_text(completion.name)
|
|
|
|
comp_item.set_icon(self.get_icon_for_type(completion.type))
|
|
|
|
comp_item.set_info(completion.docstring())
|
|
|
|
yield comp_item
|
|
|
|
|
|
|
|
for item in create_generator():
|
|
|
|
proposals.append(item)
|
2023-03-24 06:01:57 +00:00
|
|
|
|
|
|
|
context.add_proposals(self, proposals, True)
|
|
|
|
|
|
|
|
def get_icon_for_type(self, _type):
|
|
|
|
try:
|
|
|
|
return self._theme.load_icon(icon_names[_type.lower()], 16, 0)
|
|
|
|
except:
|
|
|
|
try:
|
|
|
|
return self._theme.load_icon(Gtk.STOCK_ADD, 16, 0)
|
|
|
|
except:
|
|
|
|
return None
|