thrift/tutorial/py.twisted/PythonClient.py
2016-01-13 20:25:19 +09:00

80 lines
2.3 KiB
Python
Executable File

#!/usr/bin/env python
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
#
import glob
import sys
sys.path.append('gen-py.twisted')
sys.path.insert(0, glob.glob('../../lib/py/build/lib*')[0])
from tutorial import Calculator
from tutorial.ttypes import InvalidOperation, Operation, Work
from twisted.internet.defer import inlineCallbacks
from twisted.internet import reactor
from twisted.internet.protocol import ClientCreator
from thrift.transport import TTwisted
from thrift.protocol import TBinaryProtocol
@inlineCallbacks
def main(client):
yield client.ping()
print('ping()')
sum = yield client.add(1, 1)
print(('1+1=%d' % (sum)))
work = Work()
work.op = Operation.DIVIDE
work.num1 = 1
work.num2 = 0
try:
quotient = yield client.calculate(1, work)
print('Whoa? You know how to divide by zero?')
print('FYI the answer is %d' % quotient)
except InvalidOperation as e:
print(('InvalidOperation: %r' % e))
work.op = Operation.SUBTRACT
work.num1 = 15
work.num2 = 10
diff = yield client.calculate(1, work)
print(('15-10=%d' % (diff)))
log = yield client.getStruct(1)
print(('Check log: %s' % (log.value)))
reactor.stop()
if __name__ == '__main__':
d = ClientCreator(reactor,
TTwisted.ThriftClientProtocol,
Calculator.Client,
TBinaryProtocol.TBinaryProtocolFactory(),
).connectTCP("127.0.0.1", 9090)
d.addCallback(lambda conn: conn.client)
d.addCallback(main)
reactor.run()