← Back to branch summary

~ntt-pf-lab/nova/monkey_patch_notification

« back to all changes in this revision

Viewing changes to vendor/Twisted-10.0.0/twisted/internet/_win32stdio.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/Twisted-10.0.0/twisted/internet/_win32stdio.py
 
1
# -*- test-case-name: twisted.test.test_stdio -*-
 
2
 
 
3
"""
 
4
Windows-specific implementation of the L{twisted.internet.stdio} interface.
 
5
"""
 
6
 
 
7
import win32api
 
8
import os, msvcrt
 
9
 
 
10
from zope.interface import implements
 
11
 
 
12
from twisted.internet.interfaces import IHalfCloseableProtocol, ITransport, IAddress
 
13
from twisted.internet.interfaces import IConsumer, IPushProducer
 
14
 
 
15
from twisted.internet import _pollingfile, main
 
16
from twisted.python.failure import Failure
 
17
 
 
18
 
 
19
class Win32PipeAddress(object):
 
20
    implements(IAddress)
 
21
 
 
22
 
 
23
 
 
24
class StandardIO(_pollingfile._PollingTimer):
 
25
 
 
26
    implements(ITransport,
 
27
               IConsumer,
 
28
               IPushProducer)
 
29
 
 
30
    disconnecting = False
 
31
    disconnected = False
 
32
 
 
33
    def __init__(self, proto):
 
34
        """
 
35
        Start talking to standard IO with the given protocol.
 
36
 
 
37
        Also, put it stdin/stdout/stderr into binary mode.
 
38
        """
 
39
        from twisted.internet import reactor
 
40
 
 
41
        for stdfd in range(0, 1, 2):
 
42
            msvcrt.setmode(stdfd, os.O_BINARY)
 
43
 
 
44
        _pollingfile._PollingTimer.__init__(self, reactor)
 
45
        self.proto = proto
 
46
 
 
47
        hstdin = win32api.GetStdHandle(win32api.STD_INPUT_HANDLE)
 
48
        hstdout = win32api.GetStdHandle(win32api.STD_OUTPUT_HANDLE)
 
49
 
 
50
        self.stdin = _pollingfile._PollableReadPipe(
 
51
            hstdin, self.dataReceived, self.readConnectionLost)
 
52
 
 
53
        self.stdout = _pollingfile._PollableWritePipe(
 
54
            hstdout, self.writeConnectionLost)
 
55
 
 
56
        self._addPollableResource(self.stdin)
 
57
        self._addPollableResource(self.stdout)
 
58
 
 
59
        self.proto.makeConnection(self)
 
60
 
 
61
    def dataReceived(self, data):
 
62
        self.proto.dataReceived(data)
 
63
 
 
64
    def readConnectionLost(self):
 
65
        if IHalfCloseableProtocol.providedBy(self.proto):
 
66
            self.proto.readConnectionLost()
 
67
        self.checkConnLost()
 
68
 
 
69
    def writeConnectionLost(self):
 
70
        if IHalfCloseableProtocol.providedBy(self.proto):
 
71
            self.proto.writeConnectionLost()
 
72
        self.checkConnLost()
 
73
 
 
74
    connsLost = 0
 
75
 
 
76
    def checkConnLost(self):
 
77
        self.connsLost += 1
 
78
        if self.connsLost >= 2:
 
79
            self.disconnecting = True
 
80
            self.disconnected = True
 
81
            self.proto.connectionLost(Failure(main.CONNECTION_DONE))
 
82
 
 
83
    # ITransport
 
84
 
 
85
    def write(self, data):
 
86
        self.stdout.write(data)
 
87
 
 
88
    def writeSequence(self, seq):
 
89
        self.stdout.write(''.join(seq))
 
90
 
 
91
    def loseConnection(self):
 
92
        self.disconnecting = True
 
93
        self.stdin.close()
 
94
        self.stdout.close()
 
95
 
 
96
    def getPeer(self):
 
97
        return Win32PipeAddress()
 
98
 
 
99
    def getHost(self):
 
100
        return Win32PipeAddress()
 
101
 
 
102
    # IConsumer
 
103
 
 
104
    def registerProducer(self, producer, streaming):
 
105
        return self.stdout.registerProducer(producer, streaming)
 
106
 
 
107
    def unregisterProducer(self):
 
108
        return self.stdout.unregisterProducer()
 
109
 
 
110
    # def write() above
 
111
 
 
112
    # IProducer
 
113
 
 
114
    def stopProducing(self):
 
115
        self.stdin.stopProducing()
 
116
 
 
117
    # IPushProducer
 
118
 
 
119
    def pauseProducing(self):
 
120
        self.stdin.pauseProducing()
 
121
 
 
122
    def resumeProducing(self):
 
123
        self.stdin.resumeProducing()
 
124