包装python os.system方法

Question

我有以下perl模块，用于在perl脚本中包装核心：：system：

package system_wrapper;

sub check_system {
    my ($cmd) = @_;

    my $err = CORE::system($cmd);
    if ($err != 0) {
        print "Error occured when executing: $cmd. Exiting.\n";
        exit(-1);
    }
}
*CORE::GLOBAL::system = \&check_system;

1;
__END__

我正试图在蟒蛇身上取得同样的成就。我想不出如何使用装饰器将描述的这里语法扩展到这个os方法。

我希望对包装方法的调用与未包装方法完全相同。

即status = os.system("mycmd" + " myarg")

Answer 1

你可以只是猴子补丁os.system。将真正的os.system重命名为其他东西，然后使用它创建一个函数并将其分配给os.system

def my_os_system(cmd):
    err = os._system(cmd)
    if err != 0:
        print "Error occured when executing: %s. Exiting." % cmd
        sys.exit(-1)

os._system = os.system
os.system = my_os_system