将函数/方法移动到类

时间:2017-05-17 17:14:52

标签: python refactoring pycharm automated-refactoring

使用pycharm,我希望将方法重构为一个类。 (Staticmethod会这样做) 电流:

import math

class Solver(object):
   def __init__(self, a, b, c):
       self.a = a
       self.b = b
       self.c = c


def demo(b, a, c):
   d = b ** 2 - 4 * a * c
   if d >= 0:
       disc = math.sqrt(d)
       root1 = (- b + disc) / (2 * a)
       root2 = (- b - disc) / (2 * a)
       print(root1, root2)
       return root1, root2
   else:
       raise Exception


s = Solver(2, 123, 0.025)
demo(s.b, s.a, s.c)

所需:

import math

class Solver(object):
     def __init__(self,a,b,c):
            self.a = a
            self.b = b
            self.c = c

    def demo(self, a, b, c):
            d = self.b ** 2 - 4 * self.a * self.c
            if d >= 0:
                disc = math.sqrt(d)
                root1 = (- self.b + disc) / (2 * self.a)
                root2 = (- self.b - disc) / (2 * self.a)
                print(root1, root2)
                return root1, root2
            else:
                raise Exception

Solver(2, 123, 0.025).demo()

我基本上试图获得相反的功能: "将功能/方法移至顶级"

如下所述: https://www.jetbrains.com/help/pycharm/2017.1/move-refactorings.html

我不介意选择没有init参数的课程。

1 个答案:

答案 0 :(得分:1)

默认情况下没有这样的选项:PyCharm非常擅长重构类和方法,但不能对独立函数做很多事情。 虽然,有一个解决方案可以解决你的问题:正则表达式!

基本上,你要做的是:

  1. 将函数更改为有界方法(您已在示例中执行此操作)
  2. 用重构的方法替换所有出现的旧方法调用。
  3. 这是正则表达式,可以让你为上述例子做到这一点:

    ([\w]+)[ \t]+=[ \t](Solver[ \t]*\(([\d.]+)[ \t]*,[ \t]*([\d.]+)[ \t]*,[ \t]*([\d.]+)[ \t]*\))\n\r?demo[ \t]*\(\1\.b[ \t]*,[ \t]*\1\.a[ \t]*,[ \t]*\1\.c[ \t]*\)
    

    这是替代品:

    $2\.demo()
    

    现在您可以在PyCharm中选择Edit -> Find -> Replace in Path,检查正则表达式选项,将第一个正则表达式粘贴到第一个字段,然后粘贴到下一个字段。我已经在本地测试了一个文件并且运行良好。这里是regex101 example所以你可以玩它并测试它。

    如果您使用该方法有很多用法,这将非常有用,否则手动执行此操作可能会更快。