在python中實現導入一個需要傳參的模塊

Posted on by

最近跑實驗，遇到瞭一個問題：

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

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

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

關於傳參，python中提供瞭argparse這一模塊。通過這個模塊 就能很快的實現命令行傳參的功能。

下面舉個例子：

文件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

運行：

我們可以通過改變命令行的參數來改變這個name的值，其實關於這一點也比較好理解，我們可以想成python將這三份代碼拼接在一起，再一起執行。

這樣就實現瞭我剛才想要的功能。

補充：python調用模塊時傳參出錯的解決方案

首先定義瞭一個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中調用該模塊：

#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,"[email protected]")
        Login.logout(self)
    def tearDown(self):
        self.driver.quit()
        self.assertEqual([],self.verificationErrors)
if __name__ == '__main__':
    unittest.main()

可是運行後發現出錯：

明明傳參是“jinbian3333”,”jinbian76″，為什麼登錄信息是圖中呢？

回頭查看被調用模塊，

按我理解就是 加瞭” ” 就不是變量瞭，是常量瞭， 你調用瞭該模塊，即使傳參 它還是按照模塊本身的設置：“username””password”運行

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：