使用python動態特性時，讓pycharm自動補全

在函數中指定方法類型

class Element(object):
    """
    封裝element元素
    """

    def __new__(cls, *args):
        """
        :rtype: WebElement
        """
        args_copy = MyList(args)
        args_copy[0] = by[args_copy[0]]
        return args_copy

 

指定某一個變量類型， else分支上sele.driver為指定的類型

class APP(object):
    """
    APP頁面基類
    """
    __metaclass__ = APPMetaclass
    driver_share = {}
    driver = None
    command_executor = None
    desired_capabilities = None

    def __init__(self):
        if not self.driver_share.get('driver'):
            self.driver = Remote(self.command_executor, desired_capabilities=self.desired_capabilities)
            self.driver_share['driver'] = self.driver
        else:
            self.driver = self.driver_share['driver']  # type: Remote

        print id(self.driver)

 

這樣子，在使用過程中就可以自動補全了