[Python #4] Learning Python with Selenium - Function [CN]

今天研究函数。反复使用的代码绑在一个函数,需要的时候呼叫即可。
这样容易管理代码, 可读性更佳。
python的函数类似于java的函数，
比起java写法更简单，自由度也高于java。
最大的区别就是不用必须定义type。

下面比较下java和python的函数，传 list 参数并返回object对象。

---

java:

public Object a(List<Object> name){
...  ...
return object
}

---

python:

def a(name):
    ...  ...
    return object

用 python 这么简单就可以写一个函数。
python的自由度高于java。
如果java是iphone那python就是安卓。
但是官方好像不太推荐此类写法， 函数调用虽然没有问题，却会降低可读性。
再说，你不调用这个函数前无法知道参数和返回值的类型是什么。
幸好python提供叫做annotation的功能来修饰函数。
修饰的方法也非常简单，在函数第一行加入几个字符串即可。
参数后面加：type，括号后面加 ->type来修饰函数。如果没有返回值就输入->None
如下

def a(name: list) -> object:
    ...  ...
    return object

Source

from selenium import webdriver
import time
from selenium.common.exceptions import NoSuchElementException
import pprint

def login():
    """
    log in to steemit
    """
    # Click the login button
    chrome.find_element_by_xpath('//Header/descendant::a[text()="로그인"]').click()
    # Enter username
    chrome.find_element_by_xpath('//input[contains(@name,"username")]').send_keys('june0620')
    # Get posting key from file
    with open('C:/Users/USER/Desktop/python/june0620.txt') as pw:
        for postingKey in pw:
            # Enter posting key
            chrome.find_element_by_xpath('//input[contains(@name,"password")]').send_keys(postingKey)
            # Click the login button
            chrome.find_element_by_xpath('//button[text()="로그인"]').click()
    # Waiting for N seconds
    time.sleep(3)

def vote():
    pass

def is_nsfw(post: webdriver) -> bool:
    """
    Check if it is nsfw.
    :param post: Post element
    """
    try:
        post.find_element_by_xpath('descendant::span[text()="nsfw"]')
        return True
    except NoSuchElementException as err:
        return False

def posts_info_list(posts: webdriver) -> list:
    """
    Return the information in the posts.

    """
    info_list = []
    for post in posts:
        post_info = {
            "title": post.find_element_by_xpath('descendant::h2').text,
            'url': post.find_element_by_xpath('descendant::h2/a').get_attribute('href'),
            'username': post.find_element_by_xpath('descendant::span[@class="author"]/strong').text,
            'reputation': post.find_element_by_xpath('descendant::span[@class="Reputation"]').text,
            'post_date': post.find_element_by_xpath('descendant::span[@class="updated"]').text,
            'upvoting_status': post.find_element_by_xpath('descendant::span[@class="Voting__inner"]/span[1]').get_attribute('class'),
            'rewards': post.find_element_by_xpath('descendant::span[@class="FormattedAsset "]').text
        }
        info_list.append(post_info)
    return info_list

---

内容整理

Python:

1. 函数由def 开头，由 :结束。函数内部必须缩进。
2. 可以简单写 python 函数，但最好用annotation和docstring来修饰函数。在函数内部第一行用''' ''' 来添加docstring (图片绿色部分)
3. 空函数用pass关键词来创建，但只需临时使用这样的写法，因为函数不能为空(图片的 vote 函数)

pycharm:

1. 浏览函数的annotation和docstring 快捷键是 ctrl + Q 。

Reference:
[Python #3] Learning Python with Selenium - Dictionary [KR]