Selenium2Library代码在PyCharm中完成

时间:2014-07-04 22:41:09

标签: python selenium pycharm robotframework

我尝试使用Python为Robot Framework创建基本库扩展,并且我使用PyCharm作为编辑器。对于直接导入的库,代码完成工作正常,但在这种情况下,我通过方法间接导入Selenium2Library:

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

我用其他方法用

之类的方法调用它
driver = get_current_browser()

这成功地抓住了Robot Framework中的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

1 个答案:

答案 0 :(得分: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