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):

    17

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

    18

        self.returncode = None

    19

        self.stdout = None

    20

        self.stdin = None

    21

        # TODO: stderr

    22

        file_actions = posix_spawn.FileActions()

    23

        if stdout is not None:

    24

            # child's stdout, child 2 parent pipe

    25

            c2pread, c2pwrite = os.pipe()

    26

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

    27

            file_actions.add_dup2(c2pwrite, 1)

    28

            # close other end

    29

            file_actions.add_close(c2pread)

    30

        if stdin is not None:

    31

            # child's stdin, parent to child pipe

    32

            p2cread, p2cwrite = os.pipe()

    33

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

    34

            file_actions.add_dup2(p2cread, 0)

    35

            # close other end

    36

            file_actions.add_close(p2cwrite)

    37

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

    38

        if stdout is not None:

    39

            self.stdout = os.fdopen(c2pread)

    40

            os.close(c2pwrite)

    41

        if stdin is not None:

    42

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

    43

            os.close(p2cread)

    44

    45

    def _wait(self):

    46

        if self.returncode is None:

    47

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

    48

    49

    def communicate(self):

    50

        if self.stdin is not None:

    51

            self.stdin.close()

    52

            self.stdin = None

    53

        if self.stdout is not None:

    54

            stdoutdata = self.stdout.read()

    55

        else:

    56

            stdoutdata = ""

    57

    58

        # TODO

    59

        stderrdata = ""

    60

    61

        self._wait()

    62

        if self.stdout is not None:

    63

            self.stdout.close()

    64

            self.stdout = None

    65

    66

        return (stdoutdata, stderrdata)

    67

    68

    def wait(self):

    69

        if self.stdin is not None:

    70

            self.stdin.close()

    71

            self.stdin = None

    72

        self._wait()

    73

        if self.stdout is not None:

    74

            self.stdout.close()

    75

            self.stdout = None

    76

        return self.returncode

    77

    78

    def poll(self):

    79

        if self.returncode is None:

    80

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

    81

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

    82

                self.returncode = ret

    83

    84

                if self.stdin is not None:

    85

                    self.stdin.close()

    86

                    self.stdin = None

    87

                if self.stdout is not None:

    88

                    self.stdout.close()

    89

                    self.stdout = None

    90

    91

        return self.returncode

    92

    93

    def kill(self):

    94

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

    95

    96

        if self.stdin is not None:

    97

            self.stdin.close()

    98

            self.stdin = None

    99

        if self.stdout is not None:

   100

            self.stdout.close()

   101

            self.stdout = None

   102

   103

   104

def call(*args):

   105

    cmd = []

   106

    if isinstance(args[0], str):

   107

        if len(args) == 1:

   108

            # splitting of arguments that cares about 

   109

            # use of simple and double quotes

   110

            cmd = shlex.split(args[0])

   111

        else:

   112

            cmd = args

   113

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

   114

        cmd = args[0]

   115

    else:

   116

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

   117

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

   118

    return os.waitpid(pid, 0)

   119

   120

   121

if __name__ == '__main__':

   122

    # unit test

   123

   124

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

   125

    p.stdin.write("blah")

   126

    p.stdin.close()

   127

    print(p.stdout.read())

   128

    p.wait()

   129

   130

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

   131

    p.stdin.write("blah")

   132

    print(p.communicate())

   133

   134

    call("echo blah0")

   135

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

   136

    call("echo", "blah2")