一、前言
在開發過程中,遇到了這樣一個情況:我們需要在腳本中通過 suprocess.call
方法來啟動另外一個腳本(腳本 B),當然啦,還得傳遞一些參數。在這些參數中,有一個需要傳遞的是一個實例化后的對象。我們知道,通過命令行的方式傳遞參數是基于字符格式的,也就是說腳本 B 只能接收到字符串格式的參數,那么如何接收啟動腳本傳遞過來的實例化后的對象呢?
今天就來聊聊我使用的兩種笨方法:使用 eval
以及使用 pickle
和 base64
模塊。
方法一:使用 eval
其實在代碼中使用 eval
應該不算是 good practice,不過既然可以暫時解決問題,何不拿來試試?其實使用這種方法并不能在命令行中傳遞實例化后的對象,只是將實例化的過程放在腳本 B 中進行了。
以下是啟動腳本:
import subprocessclass 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(): # 我們把實例化的過程延遲 commands = ['python3', '/home/chris/Projects/Python/movie_wisdom/script.py', 'Student()'] subprocess.call(commands)if __name__ == '__main__': start_script()
以下是被啟動的腳本,即腳本 B 代碼:
from starter import Studentdef main(): student_obj = sys.argv[-1] # 進行實例化,從而達到“傳遞”對象的目的 print(eval(student_obj))main()
方法二:使用 pickle 和 base64 模塊
這種方法采用的思路描述如下:
1、啟動腳本:pickle
模塊的 dumps
方法可以將一個 Python 對象序列化成字節串;
2、啟動腳本:base64
模塊的 encodebytes
方法可以將二進制的字節串編碼為字符串;
3、被啟動腳本:base64
模塊的 decodebytes
方法用于將使用 base64
編碼的字符串轉換成為 pickle
模塊 dumps
后的字節串;
4、被啟動腳本:pickle
模塊的 loads
方法將上一步的字節串轉換成對象實例。
看起來上述過程似乎挺麻煩的,但是通常只需要兩行關鍵代碼就可以解決問題了,不過我們在這兒給封裝到函數中了。
函數的代碼編寫如下:
def pickle_dumps_to_str(obj): try: return base64.encodebytes(pickle.dumps(obj)).decode() except pickle.PicklingError: passdef pickle_loads_from_str(obj_str): try: return pickle.loads(base64.decodebytes(obj_str.encode())) except pickle.UnpicklingError: pass
新聞熱點
疑難解答