快捷導(dǎo)航

利用Python命令行傳遞實(shí)例化對(duì)象的方法

更新時(shí)間：2016年11月02日 09:03:22 投稿：daisy

最近在工作中遇到了一個(gè)問題，需要接收啟動(dòng)腳本傳遞過來的實(shí)例化后的對(duì)象,通過在網(wǎng)上查找資料發(fā)現(xiàn)了兩個(gè)方法，文中通過實(shí)例代碼就給大家詳細(xì)介紹了這兩種方法，有需要的朋友們可以參考借鑒，下面來一起看看吧。

一、前言

在開發(fā)過程中，遇到了這樣一個(gè)情況：我們需要在腳本中通過 suprocess.call 方法來啟動(dòng)另外一個(gè)腳本（腳本 B），當(dāng)然啦，還得傳遞一些參數(shù)。在這些參數(shù)中，有一個(gè)需要傳遞的是一個(gè)實(shí)例化后的對(duì)象。我們知道，通過命令行的方式傳遞參數(shù)是基于字符格式的，也就是說腳本 B 只能接收到字符串格式的參數(shù)，那么如何接收啟動(dòng)腳本傳遞過來的實(shí)例化后的對(duì)象呢？

今天就來聊聊我使用的兩種笨方法：使用 eval 以及使用 pickle 和 base64 模塊。

方法一：使用 eval

其實(shí)在代碼中使用 eval 應(yīng)該不算是 good practice，不過既然可以暫時(shí)解決問題，何不拿來試試？其實(shí)使用這種方法并不能在命令行中傳遞實(shí)例化后的對(duì)象，只是將實(shí)例化的過程放在腳本 B 中進(jìn)行了。

以下是啟動(dòng)腳本：

import subprocess
class Student(object):
 def __init__(self):
  self.name = 'Chris'
  self.age = 30
 def __str__(self):
  return '\n'.join('{}:{}'.format(k_, v_) for k_, v_ in self.__dict__.items()
       if not k_.startswith('_'))
def start_script():
 # 我們把實(shí)例化的過程延遲
 commands = ['python3', '/home/chris/Projects/Python/movie_wisdom/script.py',
    'Student()']
 subprocess.call(commands)
if __name__ == '__main__':
 start_script()

以下是被啟動(dòng)的腳本，即腳本 B 代碼：

from starter import Student
def main():
 student_obj = sys.argv[-1]
 # 進(jìn)行實(shí)例化，從而達(dá)到“傳遞”對(duì)象的目的
 print(eval(student_obj))
main()

方法二：使用 pickle 和 base64 模塊

這種方法采用的思路描述如下：

1、啟動(dòng)腳本：pickle 模塊的 dumps 方法可以將一個(gè) Python 對(duì)象序列化成字節(jié)串；

2、啟動(dòng)腳本：base64 模塊的 encodebytes 方法可以將二進(jìn)制的字節(jié)串編碼為字符串；

3、被啟動(dòng)腳本：base64 模塊的 decodebytes 方法用于將使用 base64 編碼的字符串轉(zhuǎn)換成為 pickle 模塊 dumps 后的字節(jié)串；

4、被啟動(dòng)腳本：pickle 模塊的 loads 方法將上一步的字節(jié)串轉(zhuǎn)換成對(duì)象實(shí)例。

看起來上述過程似乎挺麻煩的，但是通常只需要兩行關(guān)鍵代碼就可以解決問題了，不過我們?cè)谶@兒給封裝到函數(shù)中了。

函數(shù)的代碼編寫如下：

def pickle_dumps_to_str(obj):
 try:
  return base64.encodebytes(pickle.dumps(obj)).decode()
 except pickle.PicklingError:
  pass
def pickle_loads_from_str(obj_str):
 try:
  return pickle.loads(base64.decodebytes(obj_str.encode()))
 except pickle.UnpicklingError:
  pass

下面，我們來看看怎么借助上述兩個(gè)函數(shù)在命令行中傳遞實(shí)例化后的 Student 對(duì)象。

啟動(dòng)代碼改寫成下面這樣：

def start_script():
 student = Student()
 student.name = 'Mary'
 # 此時(shí)傳遞的將是序列化后的 Student 對(duì)象實(shí)例（注意和使用 `eval` 的區(qū)別）
 commands = ['python3', '/home/chris/Projects/Python/movie_wisdom/script.py',
    pickle_dumps_to_str(student)]
 subprocess.call(commands)

被啟動(dòng)的腳本代碼改寫如下：

def main():
 student_obj = sys.argv[-1]
 # 載入 Student 對(duì)象實(shí)例
 print(pickle_loads_from_str(student_obj))

總結(jié)

以上就是這篇文章的全部?jī)?nèi)容了，其實(shí)關(guān)于這種需要在參數(shù)中傳遞實(shí)例化后的對(duì)象的方法不僅限于此，不過這應(yīng)該算是比較特殊的一種應(yīng)用情景了吧。如果大家有更好的方法，還望有大神指點(diǎn)。希望這篇文章對(duì)有需要的朋友們能有所幫助。

您可能感興趣的文章: