如何用Python编写适当的函数文件

Question

我想写一个包含我们想在项目中使用的函数的Python文件。我们正在为Instagram开发一个硒网刮擦机器人。现在我们在脚本中编写所有的函数，但是我们想制作一个“函数”文件，我们将导入并用于我们的脚本。但问题是，当我想使用像driver.find_element_by_xpath（cookies_button_xpath）这样的网络驱动程序函数时，VS代码不使用自动完成。函数文件（还没有完成）是这样的：

import time

from selenium.webdriver.common.keys import Keys
from selenium import webdriver

# set constants for functions to run
WEBSITE_PRE_FIX = 'https://www.instagram.com/'
FORBIDDEN_CAPTION_WORDS = ['link in bio','buy now','limited time']

def open_ig(driver: webdriver):
    # opens the website and waits till it is loaded
    driver.get(WEBSITE_PRE_FIX)
    time.sleep(2)

    # accept cookies
    cookies_button_xpath = "/html/body/div[4]/div/div/button[1]"
    driver.find_element_by_xpath(cookies_button_xpath).click()

def login(driver: webdriver, username, password):
    time.sleep(2)
    
    # fill in user name and password and log in
    username_box_xpath = '/html/body/div[1]/section/main/article/div[2]/div[1]/div/form/div/div[1]/div/label/input'
    username_element = driver.find_element_by_xpath(username_box_xpath)
    username_element.send_keys(username)

    password_box_xpath = '/html/body/div[1]/section/main/article/div[2]/div[1]/div/form/div/div[2]/div/label/input'
    password_element = driver.find_element_by_xpath(password_box_xpath)
    password_element.send_keys(password)
    password_element.send_keys(Keys.ENTER)

    # click on do not save username and password + do not turn on notifications
    time.sleep(3)
    dont_save_username_button_password_xpath = '/html/body/div[1]/section/main/div/div/div/div/button'
    dont_save_username_button_element = driver.find_element_by_xpath(dont_save_username_button_password_xpath)
    dont_save_username_button_element.click()

因此，代码确实可以工作（就像它运行并做我想要的那样），但我想知道我们是否可以用另一种方式编写函数文件，以便像颜色过滤器中的自动完成这样的工作。我不完全确定这是否可能。如果有任何其他方法来编写函数文件，欢迎提供所有建议。




Answer 1

你试过把函数文件写成一个简单的类吗？

class FunctionsFile():
    def __init__(self):
        self.website_pre_fix = 'https://www.instagram.com/'
        self.forbidden_capture_words = ['link in bio','buy now','limited time']

    def open_ig(self, driver: webdriver):
        # opens the website and waits till it is loaded
        driver.get(WEBSITE_PRE_FIX)
        time.sleep(2)

        # accept cookies
        cookies_button_xpath = "/html/body/div[4]/div/div/button[1]"
        driver.find_element_by_xpath(cookies_button_xpath).click()

    def login(self, driver: webdriver, username, password):
        time.sleep(2)
    
        # fill in user name and password and log in
        username_box_xpath = '/html/body/div[1]/section/main/article/div[2]/div[1]/div/form/div/div[1]/div/label/input'
        username_element = driver.find_element_by_xpath(username_box_xpath)
        username_element.send_keys(username)

        password_box_xpath = '/html/body/div[1]/section/main/article/div[2]/div[1]/div/form/div/div[2]/div/label/input'
        password_element = driver.find_element_by_xpath(password_box_xpath)
        password_element.send_keys(password)
        password_element.send_keys(Keys.ENTER)

        # click on do not save username and password + do not turn on notifications
        time.sleep(3)
        dont_save_username_button_password_xpath = '/html/body/div[1]/section/main/div/div/div/div/button'
        dont_save_username_button_element = driver.find_element_by_xpath(dont_save_username_button_password_xpath)
        dont_save_username_button_element.click()

然后可以在任何文件中实例化该类。如果在同一目录中：

from FunctionsFile import FunctionsFile

funcs = FunctionsFile()
funcs.open_ig(driver)

应该使用标准VS代码配色方案和自动补全。（我想无论如何）。