1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# subset of subprocess built-in module using posix_spawn rather than fork.

     5

     6

from __future__ import print_function

     7

from __future__ import absolute_import

     8

import os

     9

import signal

    10

import shlex

    11

import posix_spawn

    12

    13

PIPE = "42"

    14

    15

    16

class Popen(object):

    18

        self.returncode = None

    19

        self.stdout = None

    21

        self.stdin = None

    22

        file_actions = posix_spawn.FileActions()

    23

        if stdout is not None:

    24

            # child's stdout, child 2 parent pipe

    32

            c2pread, c2pwrite = os.pipe()

    35

            # close other end

    36

            file_actions.add_close(c2pread)

    37

        if stdin is not None:

    38

            # child's stdin, parent to child pipe

    39

            p2cread, p2cwrite = os.pipe()

    40

            # attach child's stdin to reading en of p2c pipe

    41

            file_actions.add_dup2(p2cread, 0)

    42

            # close other end

    43

            file_actions.add_close(p2cwrite)

    44

        self.pid = posix_spawn.posix_spawnp(args[0], args, file_actions=file_actions)

    45

        if stdout is not None:

    50

            os.close(c2pwrite)

    51

        if stdin is not None:

    52

            self.stdin = os.fdopen(p2cwrite, 'w')

    53

            os.close(p2cread)

    54

    55

    def _wait(self):

    56

        if self.returncode is None:

    57

            self.returncode = os.waitpid(self.pid, 0)[1]

    58

    59

    def communicate(self):

    60

        if self.stdin is not None:

    61

            self.stdin.close()

    62

            self.stdin = None

    64

        if self.stdout is not None:

    65

            stdoutdata = self.stdout.read()

    66

        else:

    67

            stdoutdata = ""

    68

    73

    74

        self._wait()

    76

        if self.stdout is not None:

    77

            self.stdout.close()

    78

            self.stdout = None

    79

    84

        return (stdoutdata, stderrdata)

    85

    86

    def wait(self):

    87

        if self.stdin is not None:

    88

            self.stdin.close()

    89

            self.stdin = None

    91

        self._wait()

    93

        if self.stdout is not None:

    94

            self.stdout.close()

    95

            self.stdout = None

   101

        return self.returncode

   102

   103

    def poll(self):

   104

        if self.returncode is None:

   105

            pid, ret = os.waitpid(self.pid, os.WNOHANG)

   106

            if (pid, ret) != (0, 0):

   107

                self.returncode = ret

   108

   109

                if self.stdin is not None:

   110

                    self.stdin.close()

   111

                    self.stdin = None

   113

                if self.stdout is not None:

   114

                    self.stdout.close()

   115

                    self.stdout = None

   116

   121

        return self.returncode

   122

   123

    def kill(self):

   124

        os.kill(self.pid, signal.SIGKILL)

   125

   126

        if self.stdin is not None:

   127

            self.stdin.close()

   128

            self.stdin = None

   130

        if self.stdout is not None:

   131

            self.stdout.close()

   132

            self.stdout = None

   133

   138

   139

def call(*args):

   140

    cmd = []

   141

    if isinstance(args[0], str):

   142

        if len(args) == 1:

   143

            # splitting of arguments that cares about 

   144

            # use of simple and double quotes

   145

            cmd = shlex.split(args[0])

   146

        else:

   147

            cmd = args

   148

    elif isinstance(args[0], list) and len(args) == 1:

   149

        cmd = args[0]

   150

    else:

   151

        raise Exception("Wrong arguments passed to subprocess.call")

   152

    pid = posix_spawn.posix_spawnp(cmd[0], cmd)

   153

    return os.waitpid(pid, 0)

   154

   155

   156

if __name__ == '__main__':

   157

    # unit test

   158

   159

    p = Popen(["tr", "abc", "def"], stdin=PIPE, stdout=PIPE)

   160

    p.stdin.write("blah")

   161

    p.stdin.close()

   162

    print(p.stdout.read())

   163

    p.wait()

   164

   165

    p = Popen(["tr", "abc", "def"], stdin=PIPE, stdout=PIPE)

   166

    p.stdin.write("blah")

   167

    print(p.communicate())

   168

   169

    call("echo blah0")

   170

    call(["echo", "blah1"])

   171

    call("echo", "blah2")