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 posix_spawn

    11

    12

PIPE = "42"

    13

    14

    15

class Popen(object):

    16

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

    17

        self.returncode = None

    18

        self.stdout = None

    19

        self.stdin = None

    20

        # TODO: stderr

    21

        file_actions = posix_spawn.FileActions()

    22

        if stdout is not None:

    23

            # child's stdout, child 2 parent pipe

    24

            c2pread, c2pwrite = os.pipe()

    25

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

    26

            file_actions.add_dup2(c2pwrite, 1)

    27

            # close other end

    28

            file_actions.add_close(c2pread)

    29

        if stdin is not None:

    30

            # child's stdin, parent to child pipe

    31

            p2cread, p2cwrite = os.pipe()

    32

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

    33

            file_actions.add_dup2(p2cread, 0)

    34

            # close other end

    35

            file_actions.add_close(p2cwrite)

    36

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

    37

        if stdout is not None:

    38

            self.stdout = os.fdopen(c2pread)

    39

            os.close(c2pwrite)

    40

        if stdin is not None:

    41

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

    42

            os.close(p2cread)

    43

    44

    def _wait(self):

    45

        if self.returncode is None:

    46

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

    47

    48

    def communicate(self):

    49

        if self.stdin is not None:

    50

            self.stdin.close()

    51

            self.stdin = None

    52

        if self.stdout is not None:

    53

            stdoutdata = self.stdout.read()

    54

        else:

    55

            stdoutdata = ""

    56

    57

        # TODO

    58

        stderrdata = ""

    59

    60

        self._wait()

    61

        if self.stdout is not None:

    62

            self.stdout.close()

    63

            self.stdout = None

    64

    65

        return (stdoutdata, stderrdata)

    66

    67

    def wait(self):

    68

        if self.stdin is not None:

    69

            self.stdin.close()

    70

            self.stdin = None

    71

        self._wait()

    72

        if self.stdout is not None:

    73

            self.stdout.close()

    74

            self.stdout = None

    75

        return self.returncode

    76

    77

    def poll(self):

    78

        if self.returncode is None:

    79

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

    80

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

    81

                self.returncode = ret

    82

    83

                if self.stdin is not None:

    84

                    self.stdin.close()

    85

                    self.stdin = None

    86

                if self.stdout is not None:

    87

                    self.stdout.close()

    88

                    self.stdout = None

    89

    90

        return self.returncode

    91

    92

    def kill(self):

    93

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

    94

    95

        if self.stdin is not None:

    96

            self.stdin.close()

    97

            self.stdin = None

    98

        if self.stdout is not None:

    99

            self.stdout.close()

   100

            self.stdout = None

   101

   102

   103

def call(*args):

   104

    cmd = []

   105

    if isinstance(args[0], str):

   106

        if len(args) == 1:

   107

            # oversimplified splitting of arguments,

   108

            # TODO: care about use of simple and double quotes

   109

            cmd = args[0].split()

   110

        else:

   111

            cmd = args

   112

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

   113

        cmd = args[0]

   114

    else:

   115

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

   116

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

   117

    return os.waitpid(pid, 0)

   118

   119

   120

if __name__ == '__main__':

   121

    # unit test

   122

   123

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

   124

    p.stdin.write("blah")

   125

    p.stdin.close()

   126

    print(p.stdout.read())

   127

    p.wait()

   128

   129

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

   130

    p.stdin.write("blah")

   131

    print(p.communicate())

   132

   133

    call("echo blah0")

   134

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

   135

    call("echo", "blah2")