利用Python命令行传递实例化对象的方法

一、前言

在开发过程中，遇到了这样一个情况：我们需要在脚本中通过 suprocess.call方法来启动另外一个脚本（脚本 B），当然啦，还得传递一些参数。在这些参数中，有一个需要传递的是一个实例化后的对象。我们知道，通过命令行的方式传递参数是基于字符格式的，也就是说脚本 B 只能接收到字符串格式的参数，那么如何接收启动脚本传递过来的实例化后的对象呢？

今天就来聊聊我使用的两种笨方法：使用 eval以及使用 pickle和 base64模块。

方法一：使用 eval

其实在代码中使用 eval应该不算是 good practice，不过既然可以暂时解决问题，何不拿来试试？其实使用这种方法并不能在命令行中传递实例化后的对象，只是将实例化的过程放在脚本 B 中进行了。

以下是启动脚本：

    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():
     # 我们把实例化的过程延迟
     commands = ['python3', '/home/chris/Projects/Python/movie_wisdom/script.py',
        'Student()']
     subprocess.call(commands)
    if __name__ == '__main__':
     start_script()

以下是被启动的脚本，即脚本 B 代码：

    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 `方法将上一步的字节串转换成对象实例。

看起来上述过程似乎挺麻烦的，但是通常只需要两行关键代码就可以解决问题了，不过我们在这儿给封装到函数中了。

函数的代码编写如下：

    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对象。

启动代码改写成下面这样：

    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)

被启动的脚本代码改写如下：

    def main():
     student_obj = sys.argv[-1]
     # 载入 Student 对象实例
     print(pickle_loads_from_str(student_obj))

总结

以上就是这篇文章的全部内容了，其实关于这种需要在参数中传递实例化后的对象的方法不仅限于此，不过这应该算是比较特殊的一种应用情景了吧。如果大家有更好的方法，还望有大神指点。希望这篇文章对有需要的朋友们能有所帮助。