Скрученный клиент Python - PullRequest
Новая
       18

Скрученный клиент Python

4 голосов
/ 25 декабря 2010

У меня есть этот простой Twisted Client, который подключается к Twisted серверу и запрашивает индекс. Если вы видите фн. connectionMade() в class SpellClient, query жестко запрограммирован. Сделал это в целях тестирования. Как передать этот запрос извне этому классу?

код - 

from twisted.internet import reactor
from twisted.internet import protocol

# a client protocol
class SpellClient(protocol.Protocol):
    """Once connected, send a message, then print the result."""

    def connectionMade(self):
        query = 'abased'
        self.transport.write(query)

    def dataReceived(self, data):
        "As soon as any data is received, write it back."
        print "Server said:", data
        self.transport.loseConnection()

    def connectionLost(self, reason):
        print "connection lost"

class SpellFactory(protocol.ClientFactory):
    protocol = SpellClient

    def clientConnectionFailed(self, connector, reason):
        print "Connection failed - goodbye!"
        reactor.stop()

    def clientConnectionLost(self, connector, reason):
        print "Connection lost - goodbye!"
        reactor.stop()

# this connects the protocol to a server runing on port 8000
def main():
    f = SpellFactory()
    reactor.connectTCP("localhost", 8090, f)
    reactor.run()

# this only runs if the module was *not* imported
if __name__ == '__main__':
    main()

1 Ответ

5 голосов
/ 25 декабря 2010

Протоколы, такие как SpellClient, имеют доступ к своей фабрике как к собственной.
... так что было бы несколько способов сделать это, но одним из них было бы создание другого метода в SpellFactory, такого как setQuery, и последующий доступ к нему с клиента ... 

#...in SpellFactory:  
def setQuery(self, query):
    self.query = query


#...and in SpellClient:
def connectionMade(self):
    self.transport.write(self.factory.query)

... так в основном: 

f = SpellFactory()
f.setQuery('some query')
...

... или вы можете просто создать метод _ init _ для SpellFactory и передать его туда.

...