怎么在python中實(shí)現(xiàn)導(dǎo)入一個需要傳參的模塊

執(zhí)手不憶殤 2021-08-18 10:36:51 瀏覽數(shù) (2759)

反饋

我們在使用python模塊的時候有沒有想過使用這些模塊的方法需不需要傳參？需要傳參的模塊要怎么傳參又該如何實(shí)現(xiàn)？今天小編就帶來一篇怎么在python中實(shí)現(xiàn)導(dǎo)入一個需要傳參的模塊的文章，希望可以幫到各位需要的小伙伴。

最近跑實(shí)驗，遇到了一個問題：

由于實(shí)驗數(shù)據(jù)集比較多，每次跑完一個數(shù)據(jù)集就需要手動更改文件路徑，再將文件傳到服務(wù)器，再運(yùn)行實(shí)驗，這樣的話效率很低，必須要專門看著這個實(shí)驗，啥時候跑完就手動修改運(yùn)行下一個實(shí)驗。我個人無法忍受這樣低效率，就想能不能有什么解決的辦法。

我們期望的解決辦法是通過命令行傳參來解決這個問題，因為接下來是需要編寫shell腳本來批量運(yùn)行實(shí)驗，如果用輸入語句的方式顯得太笨拙。

在編寫實(shí)驗代碼的時候，我將所有的參數(shù)集中到一個py文件中，這樣便于后期的維護(hù)，現(xiàn)在的問題就是需要通過命令行傳參的方式改變該文件中的某些值。

關(guān)于傳參，python中提供了argparse這一模塊。通過這個模塊就能很快的實(shí)現(xiàn)命令行傳參的功能。

下面舉個例子：

文件test1.py

import argparse
from argparse import RawTextHelpFormatter
              
parse = argparse.ArgumentParser(description="The parameters for the feature select method", formatter_class=RawTextHelpFormatter)
parse.add_argument('name', type = str)
 
args = parse.parse_args() 
name = args.name

文件test3.py

import test1 
str = test1.name + 'asdasdasd'

文件test2.py

import test3 
print test3.str

運(yùn)行：

我們可以通過改變命令行的參數(shù)來改變這個name的值，其實(shí)關(guān)于這一點(diǎn)也比較好理解，我們可以想成python將這三份代碼拼接在一起，再一起執(zhí)行。

這樣就實(shí)現(xiàn)了我剛才想要的功能。

補(bǔ)充：python調(diào)用模塊時傳參出錯的解決方案

首先定義了一個Login模塊

傳參是兩個：username,password

#coding=utf-8
def login(self,username,password):
    driver=self.driver
    self.driver.implicitly_wait(30)#不加等待會出錯
    driver.switch_to.frame(0)
    driver.find_element_by_class_name("dlemail").clear()
    driver.find_element_by_class_name('dlemail').send_keys("username")
    driver.find_element_by_class_name('dlpwd').clear()
    driver.find_element_by_class_name('dlpwd').send_keys("password")
    driver.find_element_by_id("dologin").click()
def logout(self):
    driver=self.driver
    driver.find_element_by_link_text(u"退出").click()
    driver.quit()

在Test.login中調(diào)用該模塊：

#coding=utf-8
from selenium import webdriver
import unittest,time
from public import Login
class TestLogin(unittest.TestCase):
    def setUp(self):
        self.driver=webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.126.com/"
        self.verificationErrors = []
    def test_login(self):
        driver=self.driver
        driver.get(self.base_url)
        Login.login(self,"jinbian3333","jinbian76")#傳參是這兩個
        text=driver.find_element_by_id("spnUid").text
        self.assertEqual(text,"jinbian3333@126.com")
        Login.logout(self)
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([],self.verificationErrors)
if __name__ == '__main__':
    unittest.main()