Selenium2Library代码在PyCharm中的完成

Question

我正在尝试使用Python为机器人框架创建一个基本的库扩展，并使用PyCharm作为编辑器。对于直接导入的库，代码完成工作正常，但在本例中，我通过一种方法间接导入Selenium2Library：

def get_current_browser():
    browser = BuiltIn().get_library_instance('Selenium2Library')._current_browser()
    return browser

我从其他方法中调用的方法如下

driver = get_current_browser()

这成功地抓取了机器人框架中的webdriver实例，并允许我随心所欲地进行操作，但当我编辑一个“驱动程序”变量时，我没有得到代码提示。有没有办法在这种情况下得到提示？

下面是完整的代码：

from robot.libraries.BuiltIn import BuiltIn
from Selenium2Library.keywords.keywordgroup import KeywordGroup
import logging


def get_current_browser():
    browser = BuiltIn().get_library_instance('Selenium2Library')._current_browser()
    return browser


class MyLibrary(KeywordGroup):

    def get_title_via_python(self):
        driver = get_current_browser()
        title = driver.title
        logging.warn("checking title %s" % title)
        return title

Answer 1

尝试向函数中添加一个docstring以帮助PyCharm。

from selenium.webdriver import Remote # Remote imported only for code completion


def get_current_browser():
    """
    :rtype: Remote
    """
    browser = BuiltIn().get_library_instance('Selenium2Library')._current_browser()
    return browser

更多信息来自http://www.jetbrains.com/pycharm/webhelp/type-hinting-in-pycharm.html