textadept-python/init.lua at default · paaguti/textadept-python · GitHub

History

160 lines (146 loc) · 4.9 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

local M = {}

--[[ This comment is for LuaDoc.

---

-- The python module for Textadept.

-- It provides utilities for editing Python code.

-- ### Key Bindings

-- + `Shift+Enter` (`⇧↩` | `S-Enter`)

-- Add ':' to the end of the current line and insert a newline.

module('_M.python')]]

-- Sets default buffer properties for Python files.

events.connect(events.LEXER_LOADED, function(name)

if name ~= 'python' then return end

buffer.use_tabs, buffer.tab_width = false, 4

end)

-- Autocompletion and documentation.

---

-- List of ctags files to use for autocompletion.

-- @class table

-- @name tags

M.tags = {_HOME .. '/modules/python/tags', _USERHOME .. '/modules/python/tags'}

-- LuaFormatter off

---

-- Map of expression patterns to their types.

-- Expressions are expected to match after the '=' sign of a statement.

-- @class table

-- @name expr_types

M.expr_types = {

['^[\'"]'] = 'str',

['^%('] = 'tuple',

['^%['] = 'list',

['^{'] = 'dict',

['^open%s*%b()%s*$'] = 'file',

['^%d+%f[^%d%.]'] = 'int',

['^%d+%.'] = 'float'

}

-- LuaFormatter on

local XPM = textadept.editing.XPM_IMAGES

local xpms = {c = XPM.CLASS, f = XPM.METHOD, m = XPM.VARIABLE, M = XPM.STRUCT, v = XPM.VARIABLE}

textadept.editing.autocompleters.python = function()

local list = {}

-- Retrieve the symbol behind the caret.

local line, pos = buffer:get_cur_line()

local symbol, op, part = line:sub(1, pos - 1):match('([%w_%.]-)(%.?)([%w_]*)$')

if symbol == '' and part == '' then return nil end -- nothing to complete

-- Attempt to identify the symbol type.

-- TODO: identify literals like "'foo'." and "[1, 2, 3].".

local assignment = '%f[%w_]' .. symbol:gsub('(%p)', '%%%1') .. '%s*=%s*(.*)$'

for i = buffer:line_from_position(buffer.current_pos) - 1, 1, -1 do

local expr = buffer:get_line(i):match(assignment)

if not expr then goto continue end

for patt, type in pairs(M.expr_types) do

if expr:find(patt) then

symbol = type

break

end

if expr:find('^[%u][%w_.]*%s*%b()%s*$') then

symbol = expr:match('^([%u][%w_.]+)%s*%b()%s*$') -- e.g. a = Foo()

break

end

::continue::

end

-- Search through ctags for completions for that symbol.

local name_patt = '^' .. part

local sep = string.char(buffer.auto_c_type_separator)

for _, filename in ipairs(M.tags) do

if not lfs.attributes(filename) then goto continue end

for line in io.lines(filename) do

local name = line:match('^%S+')

if not name:find(name_patt) or list[name] then goto continue end

local fields = line:match(';"\t(.*)$')

local k, class = fields:sub(1, 1), fields:match('class:(%S+)') or ''

if class == symbol then list[#list + 1], list[name] = name .. sep .. xpms[k], true end

::continue::

end

::continue::

end

return #part, list

end

textadept.editing.api_files.python = {

_HOME .. '/modules/python/api', _USERHOME .. '/modules/python/api'

}

-- Commands.

-- Indent on 'Enter' after a ':' or auto-indent on ':'.

events.connect(events.CHAR_ADDED, function(ch)

if buffer:get_lexer() ~= 'python' or (ch ~= 10 and ch ~= 58) then return end

local l = buffer:line_from_position(buffer.current_pos)

if l > 1 then

local line = buffer:get_line(l - (ch == 10 and 1 or 0))

if ch == 10 and line:find(':%s+$') then

buffer.line_indentation[l] = buffer.line_indentation[l - 1] + buffer.tab_width

buffer:goto_pos(buffer.line_indent_position[l])

elseif ch == 58 and

(line:find('^%s*else%s*:') or line:find('^%s*elif[^:]+:') or line:find('^%s*except[^:]*:') or

line:find('^%s*finally%s*:')) then

local try = not line:find('^%s*el')

for i = l - 1, 1, -1 do

line = buffer:get_line(i)

if buffer.line_indentation[i] <= buffer.line_indentation[l] and (not try and

(line:find('^%s*if[^:]+:%s+$') or line:find('^%s*while[^:]+:%s+$') or

line:find('^%s*for[^:]+:%s+$')) or line:find('^%s*try%s*:%s+$')) then

local pos, s = buffer.current_pos, buffer.line_indent_position[l]

buffer.line_indentation[l] = buffer.line_indentation[i]

buffer:goto_pos(pos + buffer.line_indent_position[l] - s)

break

end

end)

keys.python['shift+\n'] = function()

buffer:line_end()

buffer:add_text(':')

buffer:new_line()

end

-- Snippets.

local snip = snippets.python

snip['.'] = 'self.'

snip.__ = '__%1(init)__'

snip.def = table.concat({

"def %1(name)(%2(self%3(, ))%4(arg)):",

"\t%5('''%6(doc)''')",

"",

"\t%0"

}, '\n')

snip.ifmain = "if __name__ == '__main__':\n" ..

"\t%1(main())\n"

snip.class = table.concat({

"class %1(ClassName)%2((%3(object))):",

"\t%4('''%5(doc)'''",

"",

"\t)def __init__(self%6(, %7(arg))):",

"\t\t%8(super(%1, self).__init__())"

}, '\n')

snip.try = table.concat({

"try:",

"\t%0",

"except %2(Exception) as %3(e):",

"\t%4(pass)%5(",

"finally:",

"\t%6(pass))"

}, '\n')

return M

textadept-python/init.lua at default · paaguti/textadept-python · GitHub

Expand file tree

Breadcrumbs

init.lua

Latest commit

History

Breadcrumbs

init.lua

File metadata and controls