abstract:一、前言在開(kāi)發(fā)過(guò)程中,遇到了這樣一個(gè)情況:我們需要在腳本中通過(guò) suprocess.call 方法來(lái)啟動(dòng)另外一個(gè)腳本(腳本 B),當(dāng)然啦,還得傳遞一些參數(shù)。在這些參數(shù)中,有一個(gè)需要傳遞的是一個(gè)實(shí)例化后的對(duì)象。我們知道,通過(guò)命令行的方式傳遞參數(shù)是基于字符格式的,也就是說(shuō)腳本 B 只能接收到字符串格式的參數(shù),那么如何接收啟動(dòng)腳本傳遞過(guò)來(lái)的實(shí)例化后的對(duì)象呢?今天就來(lái)聊聊我使用的兩種笨
一、前言
在開(kāi)發(fā)過(guò)程中,遇到了這樣一個(gè)情況:我們需要在腳本中通過(guò) suprocess.call 方法來(lái)啟動(dòng)另外一個(gè)腳本(腳本 B),當(dāng)然啦,還得傳遞一些參數(shù)。在這些參數(shù)中,有一個(gè)需要傳遞的是一個(gè)實(shí)例化后的對(duì)象。我們知道,通過(guò)命令行的方式傳遞參數(shù)是基于字符格式的,也就是說(shuō)腳本 B 只能接收到字符串格式的參數(shù),那么如何接收啟動(dòng)腳本傳遞過(guò)來(lái)的實(shí)例化后的對(duì)象呢?
今天就來(lái)聊聊我使用的兩種笨方法:使用 eval 以及使用 pickle 和 base64 模塊。
方法一:使用 eval
其實(shí)在代碼中使用 eval 應(yīng)該不算是 good practice,不過(guò)既然可以暫時(shí)解決問(wèn)題,何不拿來(lái)試試?其實(shí)使用這種方法并不能在命令行中傳遞實(shí)例化后的對(duì)象,只是將實(shí)例化的過(guò)程放在腳本 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í)例化的過(guò)程延遲 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í)例。
看起來(lái)上述過(guò)程似乎挺麻煩的,但是通常只需要兩行關(guān)鍵代碼就可以解決問(wèn)題了,不過(guò)我們?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
下面,我們來(lái)看看怎么借助上述兩個(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))
更多關(guān)于利用Python命令行傳遞實(shí)例化對(duì)象的方法請(qǐng)關(guān)注PHP中文網(wǎng)(ipnx.cn)其他文章!