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

     7

     8

import os, sys

     9

import signal

    10

import shlex

    11

import posix_spawn

    12

    13

PIPE = "42"

    14

    15

fsencoding = sys.getfilesystemencoding()

    16

    17

class Popen(object):

    18

    def __init__(self, args, stdin=None, stdout=None):

    19

        self.returncode = None

    20

        self.stdout = None

    21

        self.stdin = None

    22

        # TODO: stderr

    23

        file_actions = posix_spawn.FileActions()

    24

        if stdout is not None:

    25

            # child's stdout, child 2 parent pipe

    26

            c2pread, c2pwrite = os.pipe()

    27

            # attach child's stdout to writing en of c2p pipe

    28

            file_actions.add_dup2(c2pwrite, 1)

    29

            # close other end

    30

            file_actions.add_close(c2pread)

    31

        if stdin is not None:

    32

            # child's stdin, parent to child pipe

    33

            p2cread, p2cwrite = os.pipe()

    34

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

    35

            file_actions.add_dup2(p2cread, 0)

    36

            # close other end

    37

            file_actions.add_close(p2cwrite)

    38

        args = [s.encode(fsencoding) for s in args if type(s)==str]

    39

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

    40

        if stdout is not None:

    41

            self.stdout = os.fdopen(c2pread)

    42

            os.close(c2pwrite)

    43

        if stdin is not None:

    44

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

    45

            os.close(p2cread)

    46

    47

    def _wait(self):

    48

        if self.returncode is None:

    49

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

    50

    51

    def communicate(self):

    52

        if self.stdin is not None:

    53

            self.stdin.close()

    54

            self.stdin = None

    55

        if self.stdout is not None:

    56

            stdoutdata = self.stdout.read()

    57

        else:

    58

            stdoutdata = ""

    59

    60

        # TODO

    61

        stderrdata = ""

    62

    63

        self._wait()

    64

        if self.stdout is not None:

    65

            self.stdout.close()

    66

            self.stdout = None

    67

    68

        return (stdoutdata, stderrdata)

    69

    70

    def wait(self):

    71

        if self.stdin is not None:

    72

            self.stdin.close()

    73

            self.stdin = None

    74

        self._wait()

    75

        if self.stdout is not None:

    76

            self.stdout.close()

    77

            self.stdout = None

    78

        return self.returncode

    79

    80

    def poll(self):

    81

        if self.returncode is None:

    82

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

    83

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

    84

                self.returncode = ret

    85

    86

                if self.stdin is not None:

    87

                    self.stdin.close()

    88

                    self.stdin = None

    89

                if self.stdout is not None:

    90

                    self.stdout.close()

    91

                    self.stdout = None

    92

    93

        return self.returncode

    94

    95

    def kill(self):

    96

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

    97

    98

        if self.stdin is not None:

    99

            self.stdin.close()

   100

            self.stdin = None

   101

        if self.stdout is not None:

   102

            self.stdout.close()

   103

            self.stdout = None

   104

   105

   106

def call(*args):

   107

    cmd = []

   108

    if isinstance(args[0], str):

   109

        if len(args) == 1:

   110

            # splitting of arguments that cares about 

   111

            # use of simple and double quotes

   112

            cmd = shlex.split(args[0])

   113

        else:

   114

            cmd = args

   115

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

   116

        cmd = args[0]

   117

    else:

   118

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

   119

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

   120

    return os.waitpid(pid, 0)

   121

   122

   123

if __name__ == '__main__':

   124

    # unit test

   125

   126

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

   127

    p.stdin.write("blah")

   128

    p.stdin.close()

   129

    print(p.stdout.read())

   130

    p.wait()

   131

   132

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

   133

    p.stdin.write("blah")

   134

    print(p.communicate())

   135

   136

    call("echo blah0")

   137

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

   138

    call("echo", "blah2")