pythonVSCode/pythonFiles/rope/refactor/method_object.py at master · privat1/pythonVSCode · GitHub

History

90 lines (80 loc) · 3.84 KB

Raw

import warnings

from rope.base import libutils

from rope.base import pyobjects, exceptions, change, evaluate, codeanalyze

from rope.refactor import sourceutils, occurrences, rename

class MethodObject(object):

def __init__(self, project, resource, offset):

self.project = project

this_pymodule = self.project.get_pymodule(resource)

pyname = evaluate.eval_location(this_pymodule, offset)

if pyname is None or not isinstance(pyname.get_object(),

pyobjects.PyFunction):

raise exceptions.RefactoringError(

'Replace method with method object refactoring should be '

'performed on a function.')

self.pyfunction = pyname.get_object()

self.pymodule = self.pyfunction.get_module()

self.resource = self.pymodule.get_resource()

def get_new_class(self, name):

body = sourceutils.fix_indentation(

self._get_body(), sourceutils.get_indent(self.project) * 2)

return 'class %s(object):\n\n%s%sdef __call__(self):\n%s' % \

(name, self._get_init(),

' ' * sourceutils.get_indent(self.project), body)

def get_changes(self, classname=None, new_class_name=None):

if new_class_name is not None:

warnings.warn(

'new_class_name parameter is deprecated; use classname',

DeprecationWarning, stacklevel=2)

classname = new_class_name

collector = codeanalyze.ChangeCollector(self.pymodule.source_code)

start, end = sourceutils.get_body_region(self.pyfunction)

indents = sourceutils.get_indents(

self.pymodule.lines, self.pyfunction.get_scope().get_start()) + \

sourceutils.get_indent(self.project)

new_contents = ' ' * indents + 'return %s(%s)()\n' % \

(classname, ', '.join(self._get_parameter_names()))

collector.add_change(start, end, new_contents)

insertion = self._get_class_insertion_point()

collector.add_change(insertion, insertion,

'\n\n' + self.get_new_class(classname))

changes = change.ChangeSet(

'Replace method with method object refactoring')

changes.add_change(change.ChangeContents(self.resource,

collector.get_changed()))

return changes

def _get_class_insertion_point(self):

current = self.pyfunction

while current.parent != self.pymodule:

current = current.parent

end = self.pymodule.lines.get_line_end(current.get_scope().get_end())

return min(end + 1, len(self.pymodule.source_code))

def _get_body(self):

body = sourceutils.get_body(self.pyfunction)

for param in self._get_parameter_names():

body = param + ' = None\n' + body

pymod = libutils.get_string_module(

self.project, body, self.resource)

pyname = pymod[param]

finder = occurrences.create_finder(self.project, param, pyname)

result = rename.rename_in_module(finder, 'self.' + param,

pymodule=pymod)

body = result[result.index('\n') + 1:]

return body

def _get_init(self):

params = self._get_parameter_names()

indents = ' ' * sourceutils.get_indent(self.project)

if not params:

return ''

header = indents + 'def __init__(self'

body = ''

for arg in params:

new_name = arg

if arg == 'self':

new_name = 'host'

header += ', %s' % new_name

body += indents * 2 + 'self.%s = %s\n' % (arg, new_name)

header += '):'

return '%s\n%s\n' % (header, body)

def _get_parameter_names(self):

return self.pyfunction.get_param_names()

pythonVSCode/pythonFiles/rope/refactor/method_object.py at master · privat1/pythonVSCode · GitHub

Uh oh!

Expand file tree

Breadcrumbs

method_object.py

Latest commit

History

Breadcrumbs

method_object.py

File metadata and controls