關於命令列執行Python指令碼的傳參方式

2022-09-06 18:06:40

命令列執行Python指令碼的傳參

應用場景

在對ABAQUS進行二次開發時，需要將核心指令碼的外部資料傳遞到指令碼內部並執行
核心指令碼在執行時，可以呼叫所傳遞的變數引數
命令列執行或者使用者子程式執行

方式一

使用sys.args

簡單範例

import sys

def test_sys_args():
    if len(sys.argv) > 1:
        print(len(sys.argv) - 1)
        print(sys.argv)
    else:
        print('無引數輸入')
if __name__ == '__main__':
    test_sys_args()

執行

python test.py a 1 test

指令碼檔名後面的是需要傳遞的引數

其它引數在命令列中傳入時需要用空格分開

若引數中需要包含“”，則需要使用到跳脫字元進行跳脫

輸出結果

3
['test.py', 'a', '1', 'test']

3代表著傳遞引數的數量
sys.args實現從程式外部向程式傳遞引數，它的值是一個列表list，其中儲存了通過命令列傳遞的各個引數
輸出sys.argv[0],即為第一個引數，是指令碼本身
sys.argv[1]的輸出結果為a

所以在二次開發的模擬指令碼中，使用子程式進或命令列執行指令碼，且需要使用此種方式進行傳參時，需要將變數及其引數一併傳遞，具體使用方式如下

#執行指令碼
child_process.exec(command val1=1 val2=2 val3=3)

指令碼內部

sys.argv[1]輸出結果為val1=1

滿足指令碼對的變數需求，成功將指令碼外的引數傳遞到指令碼內

方式二

建立獨立指令碼引數檔案

使用node.js的文字操作fs模組
在python的執行目錄下建立單獨的變數資料txt資料夾
先讀取變數資料，然後將變數根據格式拼接，最後將其寫入單獨的文字檔案
在執行python指令碼時，只需在指令碼中執行目錄下的文字檔案，就完成了引數的傳遞工作

如下所示

node.js後端

var fs = require('fs')

fs.writeFile('test.txt', _registerMsg, function (err) {

        if (err) {
            return console.log(err);
        } else {
            // 變數檔案建立成功後，執行核心計算指令碼
            exec("abaqus cae nogui=abaqus.py", function (error, stdout, stderr) {
                if (stdout.length > 1) {
                    // 計算成功
                    console.log('you offer args:', stdout);
                } else {
                    // 計算失敗
                    console.log('you don't offer args');
                }
                if (error) {
                    console.info('stderr : ' + stderr);
                }
            })

        }
    })

指令碼所需引數已經提前拼接並寫入到_registerMsg變數中

python指令碼

import io
with io.open("test.txt", encoding='utf-8') as f:
    code = f.read()
exec(code)

指令碼只需開啟同目錄下的引數檔案並執行，即可將引數傳遞到指令碼中

兩種方式的優缺點

第一種方式能夠對每個引數進行單獨讀取，但是也需要進行單獨輸出
當需要傳遞引數較多時，每次執行指令碼都要進行輸入，工作量較大
第二種方式將引數進行了整合，只需調整指令碼中的變數，集合進行引數的修改
在面向較多的引數時便於對引數的操作，能夠提升效率

python-命令列傳參sys.argv實際運用

平常我們在用別人寫好的python包的時候，在cmd輸入xx -h就能檢視到幫助資訊，輸入xx -p 8080就能把引數傳入程式裡，看起來非常酷。

本篇就來講下如何在python程式碼里加入命令列引數，並且其它功能，能呼叫這個引數。

argv獲取引數

Python 中也可以所用 sys 的 sys.argv 來獲取命令列引數：

sys.argv 是命令列參數列。
len(sys.argv) 是命令列引數個數。

注：sys.argv[0] 表示指令碼名。

test.py程式碼如下

# -*- coding: UTF-8 -*-
import sys
 
print '引數個數為:', len(sys.argv), '個引數。'
print '參數列:', str(sys.argv)

執行以上程式碼，輸出結果為：

$ python test.py arg1 arg2 arg3
引數個數為: 4 個引數。
參數列: ['test.py', 'arg1', 'arg2', 'arg3']

getopt模組

getopt模組是專門處理命令列引數的模組，用於獲取命令列選項和引數，也就是sys.argv。命令列選項使得程式的引數更加靈活。支援短選項模式（-）和長選項模式（--）。

該模組提供了兩個方法及一個例外處理（Exception getopt.GetoptError）來解析命令列引數。

getopt.getopt 方法用於解析命令列參數列，語法格式如下：

getopt.getopt(args, options[, long_options])

引數說明：

args: 要解析的命令列參數列。
options: 以列表的格式定義，options後的冒號(:)表示該選項必須有附加的引數，不帶冒號表示該選項不附加引數。
long_options: 以字串的格式定義，long_options 後的等號(=)表示如果設定該選項，必須有附加的引數，否則就不附加引數。
該方法返回值由兩個元素組成: 第一個是 (option, value) 元組的列表。第二個是參數列，包含那些沒有'-'或'--'的引數。

範例

假定我們建立這樣一個指令碼，可以通過命令列向指令碼檔案傳遞兩個檔名，同時我們通過另外一個選項檢視指令碼的使用。指令碼使用方法如下：

$ test.py -i -o

test.py 檔案程式碼如下所示：

# -*- coding: UTF-8 -*-
 
import sys, getopt
 
def main(argv):
   inputfile = ''
   outputfile = ''
   try:
      opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])
   except getopt.GetoptError:
      print 'test.py -i <inputfile> -o <outputfile>'
      sys.exit(2)
   for opt, arg in opts:
      if opt == '-h':
         print 'test.py -i <inputfile> -o <outputfile>'
         sys.exit()
      elif opt in ("-i", "--ifile"):
         inputfile = arg
      elif opt in ("-o", "--ofile"):
         outputfile = arg
   print '輸入的檔案為：', inputfile
   print '輸出的檔案為：', outputfile
 
if __name__ == "__main__":
   main(sys.argv[1:])

執行以上程式碼，輸出結果為：

$ python test.py -h
usage: test.py -i <inputfile> -o <outputfile>

$ python test.py -i inputfile -o outputfile
輸入的檔案為： inputfile
輸出的檔案為： outputfile

實際場景運用

結合selenium測試，比如我想測試chrome瀏覽器，那就在命令列輸入“chrome”引數，想測試firefox瀏覽器的時候，就在命令列輸入“firefox”引數，這樣就能靈活切換不同瀏覽器之間的測試了

# 儲存為run.py
 
# coding:utf-8
import sys, getopt
from selenium import webdriver
import time
 
def main(argv):
    '''
    命令列傳參
    上海-悠悠部落格：https://www.cnblogs.com/yoyoketang/
    '''
    name = "firefox" # 給個預設值
 
    try:
        # 這裡的 h 就表示該選項無引數，n:表示 n 選項後需要有引數
        opts, args = getopt.getopt(argv, "hn:", ["name="])
    except getopt.GetoptError:
        print('Error: test_yoyo.py -n <browsername>')
        print('   or: test_yoyo.py --name=<browsername>')
        sys.exit(2)
 
    for opt, arg in opts:
        if opt == "-h":
            print('test_yoyo.py -n <browsername>')
            print('or: test_yoyo.py --name=<browsername>')
            sys.exit()
        elif opt in ("-n", "--name"):
            name = arg
 
    print('run browser name : %s' % name)
    return name
 
def browser(n=None):
    '''
    啟動瀏覽器, n是瀏覽器名稱，支援瀏覽器：chrome ,firefox
    上海-悠悠部落格：https://www.cnblogs.com/yoyoketang/
    '''
    if n == None:
        name = main(sys.argv[1:])
    else:
        name = n
    if name == "firefox":
        print("當前執行瀏覽器：%s" % name)
        return webdriver.Firefox()
    elif name == "chrome":
        print("當前執行瀏覽器：%s" % name)
        return webdriver.Chrome()
    else:
        print("支援瀏覽器：chrome,firefox")
 
if __name__ == "__main__":
    driver = browser()
    driver.get("https://www.cnblogs.com/yoyoketang/")
    t = driver.title
    print(t)
    time.sleep(10)
    driver.quit()

cmd執行情況

C:Usersadmin>d:
 
D:>cd lianxi
 
D:lianxi>python run.py -n chrome
Input name : chrome
當前執行瀏覽器：chrome
 
DevTools listening on ws://127.0.0.1:54248/devtools/browser/595fe8cf-524d-4599-9
540-2502f6a6f2ca
上海-悠悠 - 部落格園
 
D:lianxi>python run.py -n firefox
Input name : firefox
當前執行瀏覽器：firefox
上海-悠悠 - 部落格園

備註：python2在cmd執行時，中文會顯示亂碼，用python3就不會有亂碼了

以上為個人經驗，希望能給大家一個參考，也希望大家多多支援it145.com。