一、前言
在开发过程中,遇到了这样一个情况:我们需要在脚本中通过 suprocess.call
方法来启动另外一个脚本(脚本 B),当然啦,还得传递一些参数。在这些参数中,有一个需要传递的是一个实例化后的对象。我们知道,通过命令行的方式传递参数是基于字符格式的,也就是说脚本 B 只能接收到字符串格式的参数,那么如何接收启动脚本传递过来的实例化后的对象呢?
今天就来聊聊我使用的两种笨方法:使用 eval
以及使用 pickle
和 base64
模块。
方法一:使用 eval
其实在代码中使用 eval
应该不算是 good practice,不过既然可以暂时解决问题,何不拿来试试?其实使用这种方法并不能在命令行中传递实例化后的对象,只是将实例化的过程放在脚本 B 中进行了。
以下是启动脚本:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
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():
# 我们把实例化的过程延迟
'Student()' ]
subprocess.call(commands)
if __name__ = = '__main__' :
start_script()
|
以下是被启动的脚本,即脚本 B 代码:
1
2
3
4
5
6
|
from starter import Student
def 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
方法将上一步的字节串转换成对象实例。
看起来上述过程似乎挺麻烦的,但是通常只需要两行关键代码就可以解决问题了,不过我们在这儿给封装到函数中了。
函数的代码编写如下:
1
2
3
4
5
6
7
8
9
10
|
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
|
下面,我们来看看怎么借助上述两个函数在命令行中传递实例化后的 Student
对象。
启动代码改写成下面这样:
1
2
3
4
5
6
7
|
def start_script():
student = Student()
student.name = 'Mary'
# 此时传递的将是序列化后的 Student 对象实例(注意和使用 `eval` 的区别)
commands = [ 'python3' , '/home/chris/Projects/Python/movie_wisdom/script.py' ,
pickle_dumps_to_str(student)]
subprocess.call(commands)
|
被启动的脚本代码改写如下:
1
2
3
4
|
def main():
student_obj = sys.argv[ - 1 ]
# 载入 Student 对象实例
print (pickle_loads_from_str(student_obj))
|
总结
以上就是这篇文章的全部内容了,其实关于这种需要在参数中传递实例化后的对象的方法不仅限于此,不过这应该算是比较特殊的一种应用情景了吧。如果大家有更好的方法,还望有大神指点。希望这篇文章对有需要的朋友们能有所帮助。
原文链接:http://blog.chriscabin.com/coding-life/python/python-in-real-world/1507.html