auser-poolparty 1.3.1 → 1.3.2
Sign up to get free protection for your applications and to get access to all the features.
- data/VERSION.yml +1 -1
- data/bin/cloud-compile +6 -1
- data/config/jeweler.rb +1 -3
- data/lib/cloud_providers/cloud_provider_instance.rb +11 -1
- data/lib/dependency_resolvers/chef/resources/chef_recipe.rb +51 -0
- data/lib/dependency_resolvers/chef.rb +12 -3
- data/lib/mixins/callbacks.rb +6 -2
- data/lib/poolparty/plugins/apache2/passenger_site.rb +2 -2
- data/test/fixtures/chef/recipes/sudo/attributes/sudoers.rb +30 -0
- data/test/fixtures/chef/recipes/sudo/recipes/default.rb +33 -0
- data/test/fixtures/chef/recipes/sudo/templates/default/sudoers.erb +22 -0
- data/test/lib/dependency_resolvers/chef/resources/chef_recipe_test.rb +21 -0
- data/test/lib/dependency_resolvers/chef_test.rb +8 -1
- data/vendor/gems/thrift/CHANGELOG +1 -0
- data/vendor/gems/thrift/Makefile.am +25 -6
- data/vendor/gems/thrift/Manifest +81 -0
- data/vendor/gems/thrift/README +17 -111
- data/vendor/gems/thrift/Rakefile +104 -0
- data/vendor/gems/thrift/{lib/hs/Setup.lhs → benchmark/Benchmark.thrift} +5 -4
- data/vendor/gems/thrift/benchmark/benchmark.rb +271 -0
- data/vendor/gems/thrift/benchmark/client.rb +74 -0
- data/vendor/gems/thrift/benchmark/server.rb +82 -0
- data/vendor/gems/thrift/benchmark/thin_server.rb +44 -0
- data/vendor/gems/thrift/ext/binary_protocol_accelerated.c +474 -0
- data/vendor/gems/thrift/{lib/cocoa/src/protocol/TProtocolException.m → ext/binary_protocol_accelerated.h} +1 -4
- data/vendor/gems/thrift/ext/compact_protocol.c +665 -0
- data/vendor/gems/thrift/{lib/cocoa/src/protocol/TProtocolException.h → ext/compact_protocol.h} +1 -6
- data/vendor/gems/thrift/ext/constants.h +95 -0
- data/vendor/gems/thrift/{contrib/fb303/Makefile.am → ext/extconf.rb} +8 -13
- data/vendor/gems/thrift/{compiler/cpp/src/platform.h → ext/macros.h} +18 -13
- data/vendor/gems/thrift/ext/memory_buffer.c +76 -0
- data/vendor/gems/thrift/{lib/csharp/src/Protocol/TMessageType.cs → ext/memory_buffer.h} +2 -13
- data/vendor/gems/thrift/ext/protocol.c +185 -0
- data/vendor/gems/thrift/{lib/csharp/src/TProcessor.cs → ext/protocol.h} +2 -11
- data/vendor/gems/thrift/ext/struct.c +606 -0
- data/vendor/gems/thrift/ext/struct.h +67 -0
- data/vendor/gems/thrift/ext/thrift_native.c +194 -0
- data/vendor/gems/thrift/lib/thrift/client.rb +62 -0
- data/vendor/gems/thrift/{contrib/fb303/py/fb303_scripts/__init__.py → lib/thrift/core_ext/fixnum.rb} +13 -4
- data/vendor/gems/thrift/{contrib/fb303/global_footer.mk → lib/thrift/core_ext.rb} +4 -2
- data/vendor/gems/thrift/lib/thrift/exceptions.rb +82 -0
- data/vendor/gems/thrift/lib/thrift/processor.rb +57 -0
- data/vendor/gems/thrift/lib/thrift/protocol/base_protocol.rb +290 -0
- data/vendor/gems/thrift/lib/thrift/protocol/binary_protocol.rb +225 -0
- data/vendor/gems/thrift/{cleanup.sh → lib/thrift/protocol/binary_protocol_accelerated.rb} +19 -42
- data/vendor/gems/thrift/lib/thrift/protocol/compact_protocol.rb +422 -0
- data/vendor/gems/thrift/lib/thrift/serializer/deserializer.rb +33 -0
- data/vendor/gems/thrift/{bootstrap.sh → lib/thrift/serializer/serializer.rb} +17 -18
- data/vendor/gems/thrift/lib/thrift/server/base_server.rb +31 -0
- data/vendor/gems/thrift/lib/thrift/server/mongrel_http_server.rb +58 -0
- data/vendor/gems/thrift/lib/thrift/server/nonblocking_server.rb +296 -0
- data/vendor/gems/thrift/lib/thrift/server/simple_server.rb +43 -0
- data/vendor/gems/thrift/lib/thrift/server/thread_pool_server.rb +75 -0
- data/vendor/gems/thrift/lib/thrift/server/threaded_server.rb +47 -0
- data/vendor/gems/thrift/lib/thrift/struct.rb +298 -0
- data/vendor/gems/thrift/{contrib/fb303/bootstrap.sh → lib/thrift/thrift_native.rb} +5 -7
- data/vendor/gems/thrift/lib/{erl/Makefile → thrift/transport/base_server_transport.rb} +20 -20
- data/vendor/gems/thrift/lib/thrift/transport/base_transport.rb +70 -0
- data/vendor/gems/thrift/lib/thrift/transport/buffered_transport.rb +77 -0
- data/vendor/gems/thrift/lib/thrift/transport/framed_transport.rb +90 -0
- data/vendor/gems/thrift/lib/thrift/transport/http_client_transport.rb +45 -0
- data/vendor/gems/thrift/lib/thrift/transport/io_stream_transport.rb +39 -0
- data/vendor/gems/thrift/lib/thrift/transport/memory_buffer_transport.rb +96 -0
- data/vendor/gems/thrift/lib/thrift/transport/server_socket.rb +63 -0
- data/vendor/gems/thrift/lib/thrift/transport/socket.rb +136 -0
- data/vendor/gems/thrift/lib/{java/Makefile.am → thrift/transport/unix_server_socket.rb} +39 -17
- data/vendor/gems/thrift/{contrib/fb303/global_header.mk → lib/thrift/transport/unix_socket.rb} +23 -21
- data/vendor/gems/thrift/lib/thrift/types.rb +101 -0
- data/vendor/gems/thrift/lib/thrift.rb +59 -0
- data/vendor/gems/thrift/script/proto_benchmark.rb +121 -0
- data/vendor/gems/thrift/{lib/Makefile.am → script/read_struct.rb} +24 -36
- data/vendor/gems/thrift/{contrib/fb303/py/setup.py → script/write_struct.rb} +8 -5
- data/vendor/gems/thrift/setup.rb +1585 -0
- data/vendor/gems/thrift/spec/ThriftSpec.thrift +84 -0
- data/vendor/gems/thrift/spec/base_protocol_spec.rb +160 -0
- data/vendor/gems/thrift/spec/base_transport_spec.rb +351 -0
- data/vendor/gems/thrift/{contrib/fb303/py/Makefile.am → spec/binary_protocol_accelerated_spec.rb} +22 -25
- data/vendor/gems/thrift/spec/binary_protocol_spec.rb +63 -0
- data/vendor/gems/thrift/spec/binary_protocol_spec_shared.rb +375 -0
- data/vendor/gems/thrift/spec/client_spec.rb +100 -0
- data/vendor/gems/thrift/spec/compact_protocol_spec.rb +117 -0
- data/vendor/gems/thrift/spec/exception_spec.rb +142 -0
- data/vendor/gems/thrift/spec/http_client_spec.rb +49 -0
- data/vendor/gems/thrift/spec/mongrel_http_server_spec.rb +117 -0
- data/vendor/gems/thrift/spec/nonblocking_server_spec.rb +265 -0
- data/vendor/gems/thrift/spec/processor_spec.rb +83 -0
- data/vendor/gems/thrift/spec/serializer_spec.rb +69 -0
- data/vendor/gems/thrift/spec/server_socket_spec.rb +80 -0
- data/vendor/gems/thrift/spec/server_spec.rb +160 -0
- data/vendor/gems/thrift/spec/socket_spec.rb +61 -0
- data/vendor/gems/thrift/spec/socket_spec_shared.rb +104 -0
- data/vendor/gems/thrift/spec/spec_helper.rb +60 -0
- data/vendor/gems/thrift/spec/struct_spec.rb +252 -0
- data/vendor/gems/thrift/spec/types_spec.rb +116 -0
- data/vendor/gems/thrift/spec/unix_socket_spec.rb +108 -0
- data/vendor/gems/trollop/FAQ.txt +84 -0
- data/vendor/gems/trollop/History.txt +101 -0
- data/vendor/gems/trollop/Manifest.txt +7 -0
- data/vendor/gems/trollop/README.txt +40 -0
- data/vendor/gems/trollop/Rakefile +36 -0
- data/vendor/gems/trollop/lib/trollop.rb +739 -0
- data/vendor/gems/trollop/release-script.txt +13 -0
- data/vendor/gems/trollop/test/test_trollop.rb +1048 -0
- data/vendor/gems/trollop/www/index.html +172 -0
- metadata +123 -347
- data/vendor/gems/thrift/CHANGES +0 -35
- data/vendor/gems/thrift/CONTRIBUTORS +0 -77
- data/vendor/gems/thrift/DISCLAIMER +0 -6
- data/vendor/gems/thrift/LICENSE +0 -202
- data/vendor/gems/thrift/NEWS +0 -79
- data/vendor/gems/thrift/NOTICE +0 -26
- data/vendor/gems/thrift/aclocal/ax_boost_base.m4 +0 -198
- data/vendor/gems/thrift/aclocal/ax_javac_and_java.m4 +0 -107
- data/vendor/gems/thrift/aclocal/ax_lib_event.m4 +0 -194
- data/vendor/gems/thrift/aclocal/ax_lib_zlib.m4 +0 -173
- data/vendor/gems/thrift/aclocal/ax_signed_right_shift.m4 +0 -127
- data/vendor/gems/thrift/aclocal/ax_thrift_internal.m4 +0 -39
- data/vendor/gems/thrift/compiler/cpp/Makefile.am +0 -136
- data/vendor/gems/thrift/compiler/cpp/README +0 -39
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_cocoa_generator.cc +0 -2331
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_cpp_generator.cc +0 -3003
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_csharp_generator.cc +0 -1700
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_erl_generator.cc +0 -932
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_generator.cc +0 -173
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_generator.h +0 -321
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_hs_generator.cc +0 -1445
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_html_generator.cc +0 -637
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_java_generator.cc +0 -3069
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_ocaml_generator.cc +0 -1673
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_oop_generator.h +0 -77
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_perl_generator.cc +0 -1812
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_php_generator.cc +0 -2281
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_py_generator.cc +0 -2310
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_rb_generator.cc +0 -1114
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_st_generator.cc +0 -1071
- data/vendor/gems/thrift/compiler/cpp/src/generate/t_xsd_generator.cc +0 -354
- data/vendor/gems/thrift/compiler/cpp/src/globals.h +0 -117
- data/vendor/gems/thrift/compiler/cpp/src/main.cc +0 -1207
- data/vendor/gems/thrift/compiler/cpp/src/main.h +0 -103
- data/vendor/gems/thrift/compiler/cpp/src/md5.c +0 -381
- data/vendor/gems/thrift/compiler/cpp/src/md5.h +0 -91
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_base_type.h +0 -137
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_const.h +0 -59
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_const_value.h +0 -121
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_container.h +0 -56
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_doc.h +0 -51
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_enum.h +0 -59
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_enum_value.h +0 -64
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_field.h +0 -150
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_function.h +0 -93
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_list.h +0 -56
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_map.h +0 -64
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_program.h +0 -223
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_scope.h +0 -86
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_service.h +0 -68
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_set.h +0 -55
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_struct.h +0 -127
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_type.h +0 -176
- data/vendor/gems/thrift/compiler/cpp/src/parse/t_typedef.h +0 -70
- data/vendor/gems/thrift/compiler/cpp/src/thriftl.ll +0 -303
- data/vendor/gems/thrift/compiler/cpp/src/thrifty.yy +0 -1140
- data/vendor/gems/thrift/configure.ac +0 -255
- data/vendor/gems/thrift/contrib/fb303/LICENSE +0 -16
- data/vendor/gems/thrift/contrib/fb303/README +0 -37
- data/vendor/gems/thrift/contrib/fb303/acinclude.m4 +0 -258
- data/vendor/gems/thrift/contrib/fb303/aclocal/ax_boost_base.m4 +0 -198
- data/vendor/gems/thrift/contrib/fb303/configure.ac +0 -115
- data/vendor/gems/thrift/contrib/fb303/cpp/FacebookBase.cpp +0 -124
- data/vendor/gems/thrift/contrib/fb303/cpp/FacebookBase.h +0 -103
- data/vendor/gems/thrift/contrib/fb303/cpp/Makefile.am +0 -84
- data/vendor/gems/thrift/contrib/fb303/cpp/ServiceTracker.cpp +0 -481
- data/vendor/gems/thrift/contrib/fb303/cpp/ServiceTracker.h +0 -215
- data/vendor/gems/thrift/contrib/fb303/if/fb303.thrift +0 -112
- data/vendor/gems/thrift/contrib/fb303/java/FacebookBase.java +0 -103
- data/vendor/gems/thrift/contrib/fb303/java/build.xml +0 -84
- data/vendor/gems/thrift/contrib/fb303/php/FacebookBase.php +0 -89
- data/vendor/gems/thrift/contrib/fb303/py/fb303/FacebookBase.py +0 -82
- data/vendor/gems/thrift/contrib/fb303/py/fb303_scripts/fb303_simple_mgmt.py +0 -195
- data/vendor/gems/thrift/contrib/thrift.el +0 -126
- data/vendor/gems/thrift/contrib/thrift.spec +0 -206
- data/vendor/gems/thrift/contrib/thrift.vim +0 -91
- data/vendor/gems/thrift/contrib/thrift_dump.cpp +0 -91
- data/vendor/gems/thrift/doc/lgpl-2.1.txt +0 -504
- data/vendor/gems/thrift/doc/otp-base-license.txt +0 -20
- data/vendor/gems/thrift/doc/thrift.bnf +0 -96
- data/vendor/gems/thrift/doc/thrift.tex +0 -1057
- data/vendor/gems/thrift/lib/cocoa/README +0 -21
- data/vendor/gems/thrift/lib/cocoa/src/TApplicationException.h +0 -44
- data/vendor/gems/thrift/lib/cocoa/src/TApplicationException.m +0 -130
- data/vendor/gems/thrift/lib/cocoa/src/TException.h +0 -34
- data/vendor/gems/thrift/lib/cocoa/src/TException.m +0 -64
- data/vendor/gems/thrift/lib/cocoa/src/TProcessor.h +0 -29
- data/vendor/gems/thrift/lib/cocoa/src/TProcessorFactory.h +0 -27
- data/vendor/gems/thrift/lib/cocoa/src/TSharedProcessorFactory.h +0 -27
- data/vendor/gems/thrift/lib/cocoa/src/TSharedProcessorFactory.m +0 -51
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TBinaryProtocol.h +0 -51
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TBinaryProtocol.m +0 -477
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TProtocol.h +0 -148
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TProtocolFactory.h +0 -29
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TProtocolUtil.h +0 -29
- data/vendor/gems/thrift/lib/cocoa/src/protocol/TProtocolUtil.m +0 -104
- data/vendor/gems/thrift/lib/cocoa/src/server/TSocketServer.h +0 -50
- data/vendor/gems/thrift/lib/cocoa/src/server/TSocketServer.m +0 -153
- data/vendor/gems/thrift/lib/cocoa/src/transport/THTTPClient.h +0 -42
- data/vendor/gems/thrift/lib/cocoa/src/transport/THTTPClient.m +0 -159
- data/vendor/gems/thrift/lib/cocoa/src/transport/TNSFileHandleTransport.h +0 -35
- data/vendor/gems/thrift/lib/cocoa/src/transport/TNSFileHandleTransport.m +0 -91
- data/vendor/gems/thrift/lib/cocoa/src/transport/TNSStreamTransport.h +0 -38
- data/vendor/gems/thrift/lib/cocoa/src/transport/TNSStreamTransport.m +0 -89
- data/vendor/gems/thrift/lib/cocoa/src/transport/TSocketClient.h +0 -32
- data/vendor/gems/thrift/lib/cocoa/src/transport/TSocketClient.m +0 -58
- data/vendor/gems/thrift/lib/cocoa/src/transport/TTransport.h +0 -36
- data/vendor/gems/thrift/lib/cocoa/src/transport/TTransportException.h +0 -30
- data/vendor/gems/thrift/lib/cocoa/src/transport/TTransportException.m +0 -43
- data/vendor/gems/thrift/lib/cpp/Makefile.am +0 -158
- data/vendor/gems/thrift/lib/cpp/README +0 -67
- data/vendor/gems/thrift/lib/cpp/src/TLogging.h +0 -163
- data/vendor/gems/thrift/lib/cpp/src/TProcessor.h +0 -53
- data/vendor/gems/thrift/lib/cpp/src/TReflectionLocal.h +0 -96
- data/vendor/gems/thrift/lib/cpp/src/Thrift.cpp +0 -148
- data/vendor/gems/thrift/lib/cpp/src/Thrift.h +0 -191
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Exception.h +0 -60
- data/vendor/gems/thrift/lib/cpp/src/concurrency/FunctionRunner.h +0 -77
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Monitor.cpp +0 -137
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Monitor.h +0 -84
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Mutex.cpp +0 -160
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Mutex.h +0 -114
- data/vendor/gems/thrift/lib/cpp/src/concurrency/PosixThreadFactory.cpp +0 -314
- data/vendor/gems/thrift/lib/cpp/src/concurrency/PosixThreadFactory.h +0 -130
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Thread.h +0 -125
- data/vendor/gems/thrift/lib/cpp/src/concurrency/ThreadManager.cpp +0 -493
- data/vendor/gems/thrift/lib/cpp/src/concurrency/ThreadManager.h +0 -169
- data/vendor/gems/thrift/lib/cpp/src/concurrency/TimerManager.cpp +0 -284
- data/vendor/gems/thrift/lib/cpp/src/concurrency/TimerManager.h +0 -122
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Util.cpp +0 -55
- data/vendor/gems/thrift/lib/cpp/src/concurrency/Util.h +0 -100
- data/vendor/gems/thrift/lib/cpp/src/concurrency/test/Tests.cpp +0 -155
- data/vendor/gems/thrift/lib/cpp/src/concurrency/test/ThreadFactoryTests.h +0 -354
- data/vendor/gems/thrift/lib/cpp/src/concurrency/test/ThreadManagerTests.h +0 -379
- data/vendor/gems/thrift/lib/cpp/src/concurrency/test/TimerManagerTests.h +0 -155
- data/vendor/gems/thrift/lib/cpp/src/processor/PeekProcessor.cpp +0 -122
- data/vendor/gems/thrift/lib/cpp/src/processor/PeekProcessor.h +0 -77
- data/vendor/gems/thrift/lib/cpp/src/processor/StatsProcessor.h +0 -264
- data/vendor/gems/thrift/lib/cpp/src/protocol/TBase64Utils.cpp +0 -79
- data/vendor/gems/thrift/lib/cpp/src/protocol/TBase64Utils.h +0 -42
- data/vendor/gems/thrift/lib/cpp/src/protocol/TBinaryProtocol.cpp +0 -394
- data/vendor/gems/thrift/lib/cpp/src/protocol/TBinaryProtocol.h +0 -254
- data/vendor/gems/thrift/lib/cpp/src/protocol/TCompactProtocol.cpp +0 -736
- data/vendor/gems/thrift/lib/cpp/src/protocol/TCompactProtocol.h +0 -279
- data/vendor/gems/thrift/lib/cpp/src/protocol/TDebugProtocol.cpp +0 -346
- data/vendor/gems/thrift/lib/cpp/src/protocol/TDebugProtocol.h +0 -225
- data/vendor/gems/thrift/lib/cpp/src/protocol/TDenseProtocol.cpp +0 -762
- data/vendor/gems/thrift/lib/cpp/src/protocol/TDenseProtocol.h +0 -253
- data/vendor/gems/thrift/lib/cpp/src/protocol/TJSONProtocol.cpp +0 -998
- data/vendor/gems/thrift/lib/cpp/src/protocol/TJSONProtocol.h +0 -340
- data/vendor/gems/thrift/lib/cpp/src/protocol/TOneWayProtocol.h +0 -304
- data/vendor/gems/thrift/lib/cpp/src/protocol/TProtocol.h +0 -438
- data/vendor/gems/thrift/lib/cpp/src/protocol/TProtocolException.h +0 -104
- data/vendor/gems/thrift/lib/cpp/src/protocol/TProtocolTap.h +0 -187
- data/vendor/gems/thrift/lib/cpp/src/server/TNonblockingServer.cpp +0 -750
- data/vendor/gems/thrift/lib/cpp/src/server/TNonblockingServer.h +0 -435
- data/vendor/gems/thrift/lib/cpp/src/server/TServer.cpp +0 -38
- data/vendor/gems/thrift/lib/cpp/src/server/TServer.h +0 -213
- data/vendor/gems/thrift/lib/cpp/src/server/TSimpleServer.cpp +0 -118
- data/vendor/gems/thrift/lib/cpp/src/server/TSimpleServer.h +0 -70
- data/vendor/gems/thrift/lib/cpp/src/server/TThreadPoolServer.cpp +0 -217
- data/vendor/gems/thrift/lib/cpp/src/server/TThreadPoolServer.h +0 -79
- data/vendor/gems/thrift/lib/cpp/src/server/TThreadedServer.cpp +0 -243
- data/vendor/gems/thrift/lib/cpp/src/server/TThreadedServer.h +0 -74
- data/vendor/gems/thrift/lib/cpp/src/transport/TBufferTransports.cpp +0 -370
- data/vendor/gems/thrift/lib/cpp/src/transport/TBufferTransports.h +0 -667
- data/vendor/gems/thrift/lib/cpp/src/transport/TFDTransport.cpp +0 -77
- data/vendor/gems/thrift/lib/cpp/src/transport/TFDTransport.h +0 -73
- data/vendor/gems/thrift/lib/cpp/src/transport/TFileTransport.cpp +0 -953
- data/vendor/gems/thrift/lib/cpp/src/transport/TFileTransport.h +0 -442
- data/vendor/gems/thrift/lib/cpp/src/transport/THttpClient.cpp +0 -348
- data/vendor/gems/thrift/lib/cpp/src/transport/THttpClient.h +0 -111
- data/vendor/gems/thrift/lib/cpp/src/transport/TServerSocket.cpp +0 -368
- data/vendor/gems/thrift/lib/cpp/src/transport/TServerSocket.h +0 -76
- data/vendor/gems/thrift/lib/cpp/src/transport/TServerTransport.h +0 -92
- data/vendor/gems/thrift/lib/cpp/src/transport/TShortReadTransport.h +0 -96
- data/vendor/gems/thrift/lib/cpp/src/transport/TSimpleFileTransport.cpp +0 -54
- data/vendor/gems/thrift/lib/cpp/src/transport/TSimpleFileTransport.h +0 -41
- data/vendor/gems/thrift/lib/cpp/src/transport/TSocket.cpp +0 -591
- data/vendor/gems/thrift/lib/cpp/src/transport/TSocket.h +0 -242
- data/vendor/gems/thrift/lib/cpp/src/transport/TSocketPool.cpp +0 -235
- data/vendor/gems/thrift/lib/cpp/src/transport/TSocketPool.h +0 -191
- data/vendor/gems/thrift/lib/cpp/src/transport/TTransport.h +0 -224
- data/vendor/gems/thrift/lib/cpp/src/transport/TTransportException.cpp +0 -31
- data/vendor/gems/thrift/lib/cpp/src/transport/TTransportException.h +0 -117
- data/vendor/gems/thrift/lib/cpp/src/transport/TTransportUtils.cpp +0 -178
- data/vendor/gems/thrift/lib/cpp/src/transport/TTransportUtils.h +0 -287
- data/vendor/gems/thrift/lib/cpp/src/transport/TZlibTransport.cpp +0 -299
- data/vendor/gems/thrift/lib/cpp/src/transport/TZlibTransport.h +0 -219
- data/vendor/gems/thrift/lib/cpp/thrift-nb.pc.in +0 -30
- data/vendor/gems/thrift/lib/cpp/thrift-z.pc.in +0 -30
- data/vendor/gems/thrift/lib/cpp/thrift.pc.in +0 -29
- data/vendor/gems/thrift/lib/csharp/Makefile.am +0 -70
- data/vendor/gems/thrift/lib/csharp/README +0 -26
- data/vendor/gems/thrift/lib/csharp/ThriftMSBuildTask/Properties/AssemblyInfo.cs +0 -55
- data/vendor/gems/thrift/lib/csharp/ThriftMSBuildTask/ThriftBuild.cs +0 -242
- data/vendor/gems/thrift/lib/csharp/ThriftMSBuildTask/ThriftMSBuildTask.csproj +0 -62
- data/vendor/gems/thrift/lib/csharp/src/Collections/THashSet.cs +0 -142
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TBase.cs +0 -34
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TBinaryProtocol.cs +0 -392
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TField.cs +0 -58
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TList.cs +0 -50
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TMap.cs +0 -58
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TMessage.cs +0 -58
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TProtocol.cs +0 -87
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TProtocolException.cs +0 -61
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TProtocolFactory.cs +0 -29
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TProtocolUtil.cs +0 -94
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TSet.cs +0 -50
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TStruct.cs +0 -42
- data/vendor/gems/thrift/lib/csharp/src/Protocol/TType.cs +0 -40
- data/vendor/gems/thrift/lib/csharp/src/Server/TServer.cs +0 -135
- data/vendor/gems/thrift/lib/csharp/src/Server/TSimpleServer.cs +0 -148
- data/vendor/gems/thrift/lib/csharp/src/Server/TThreadPoolServer.cs +0 -186
- data/vendor/gems/thrift/lib/csharp/src/Server/TThreadedServer.cs +0 -234
- data/vendor/gems/thrift/lib/csharp/src/TApplicationException.cs +0 -131
- data/vendor/gems/thrift/lib/csharp/src/Thrift.csproj +0 -73
- data/vendor/gems/thrift/lib/csharp/src/Thrift.sln +0 -35
- data/vendor/gems/thrift/lib/csharp/src/Transport/TBufferedTransport.cs +0 -100
- data/vendor/gems/thrift/lib/csharp/src/Transport/TServerSocket.cs +0 -157
- data/vendor/gems/thrift/lib/csharp/src/Transport/TServerTransport.cs +0 -39
- data/vendor/gems/thrift/lib/csharp/src/Transport/TSocket.cs +0 -144
- data/vendor/gems/thrift/lib/csharp/src/Transport/TStreamTransport.cs +0 -103
- data/vendor/gems/thrift/lib/csharp/src/Transport/TTransport.cs +0 -66
- data/vendor/gems/thrift/lib/csharp/src/Transport/TTransportException.cs +0 -64
- data/vendor/gems/thrift/lib/csharp/src/Transport/TTransportFactory.cs +0 -38
- data/vendor/gems/thrift/lib/erl/README +0 -56
- data/vendor/gems/thrift/lib/erl/build/beamver +0 -59
- data/vendor/gems/thrift/lib/erl/build/buildtargets.mk +0 -15
- data/vendor/gems/thrift/lib/erl/build/colors.mk +0 -24
- data/vendor/gems/thrift/lib/erl/build/docs.mk +0 -12
- data/vendor/gems/thrift/lib/erl/build/mime.types +0 -98
- data/vendor/gems/thrift/lib/erl/build/otp.mk +0 -146
- data/vendor/gems/thrift/lib/erl/build/otp_subdir.mk +0 -85
- data/vendor/gems/thrift/lib/erl/build/raw_test.mk +0 -29
- data/vendor/gems/thrift/lib/erl/include/thrift_constants.hrl +0 -54
- data/vendor/gems/thrift/lib/erl/include/thrift_protocol.hrl +0 -31
- data/vendor/gems/thrift/lib/erl/src/Makefile +0 -116
- data/vendor/gems/thrift/lib/erl/src/test_handler.erl +0 -26
- data/vendor/gems/thrift/lib/erl/src/test_service.erl +0 -29
- data/vendor/gems/thrift/lib/erl/src/thrift.app.src +0 -44
- data/vendor/gems/thrift/lib/erl/src/thrift.appup.src +0 -1
- data/vendor/gems/thrift/lib/erl/src/thrift_base64_transport.erl +0 -64
- data/vendor/gems/thrift/lib/erl/src/thrift_binary_protocol.erl +0 -325
- data/vendor/gems/thrift/lib/erl/src/thrift_buffered_transport.erl +0 -180
- data/vendor/gems/thrift/lib/erl/src/thrift_client.erl +0 -384
- data/vendor/gems/thrift/lib/erl/src/thrift_disk_log_transport.erl +0 -118
- data/vendor/gems/thrift/lib/erl/src/thrift_file_transport.erl +0 -87
- data/vendor/gems/thrift/lib/erl/src/thrift_framed_transport.erl +0 -208
- data/vendor/gems/thrift/lib/erl/src/thrift_http_transport.erl +0 -199
- data/vendor/gems/thrift/lib/erl/src/thrift_memory_buffer.erl +0 -164
- data/vendor/gems/thrift/lib/erl/src/thrift_processor.erl +0 -188
- data/vendor/gems/thrift/lib/erl/src/thrift_protocol.erl +0 -356
- data/vendor/gems/thrift/lib/erl/src/thrift_server.erl +0 -183
- data/vendor/gems/thrift/lib/erl/src/thrift_service.erl +0 -25
- data/vendor/gems/thrift/lib/erl/src/thrift_socket_server.erl +0 -249
- data/vendor/gems/thrift/lib/erl/src/thrift_socket_transport.erl +0 -119
- data/vendor/gems/thrift/lib/erl/src/thrift_transport.erl +0 -57
- data/vendor/gems/thrift/lib/erl/vsn.mk +0 -1
- data/vendor/gems/thrift/lib/hs/README +0 -82
- data/vendor/gems/thrift/lib/hs/TODO +0 -2
- data/vendor/gems/thrift/lib/hs/Thrift.cabal +0 -20
- data/vendor/gems/thrift/lib/hs/src/Thrift/Protocol/Binary.hs +0 -147
- data/vendor/gems/thrift/lib/hs/src/Thrift/Protocol.hs +0 -191
- data/vendor/gems/thrift/lib/hs/src/Thrift/Server.hs +0 -65
- data/vendor/gems/thrift/lib/hs/src/Thrift/Transport/Handle.hs +0 -58
- data/vendor/gems/thrift/lib/hs/src/Thrift/Transport.hs +0 -60
- data/vendor/gems/thrift/lib/hs/src/Thrift.hs +0 -111
- data/vendor/gems/thrift/lib/java/README +0 -43
- data/vendor/gems/thrift/lib/java/build.xml +0 -195
- data/vendor/gems/thrift/lib/java/ivy.xml +0 -8
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/IntRangeSet.java +0 -171
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TApplicationException.java +0 -123
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TBase.java +0 -66
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TBaseHelper.java +0 -102
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TByteArrayOutputStream.java +0 -46
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TDeserializer.java +0 -94
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TException.java +0 -45
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TFieldRequirementType.java +0 -30
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TProcessor.java +0 -32
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TProcessorFactory.java +0 -39
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/TSerializer.java +0 -110
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/FieldMetaData.java +0 -69
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/FieldValueMetaData.java +0 -42
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/ListMetaData.java +0 -29
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/MapMetaData.java +0 -31
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/SetMetaData.java +0 -29
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/meta_data/StructMetaData.java +0 -31
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TBase64Utils.java +0 -128
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TBinaryProtocol.java +0 -331
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TCompactProtocol.java +0 -741
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TField.java +0 -48
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TJSONProtocol.java +0 -927
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TList.java +0 -38
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TMap.java +0 -40
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TMessage.java +0 -48
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TMessageType.java +0 -31
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TProtocol.java +0 -146
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TProtocolException.java +0 -81
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TProtocolFactory.java +0 -30
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TProtocolUtil.java +0 -158
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TSet.java +0 -42
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TSimpleJSONProtocol.java +0 -384
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TStruct.java +0 -36
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/protocol/TType.java +0 -40
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/server/THsHaServer.java +0 -304
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/server/TNonblockingServer.java +0 -772
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/server/TServer.java +0 -126
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/server/TSimpleServer.java +0 -145
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/server/TThreadPoolServer.java +0 -271
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TFramedTransport.java +0 -126
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/THttpClient.java +0 -157
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TIOStreamTransport.java +0 -159
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TMemoryBuffer.java +0 -98
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TNonblockingServerSocket.java +0 -160
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TNonblockingServerTransport.java +0 -31
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TNonblockingSocket.java +0 -213
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TNonblockingTransport.java +0 -31
- data/vendor/gems/thrift/lib/java/src/org/apache/thrift/transport/TServerSocket.java +0 -145
@@ -1,2310 +0,0 @@
|
|
1
|
-
/*
|
2
|
-
* Licensed to the Apache Software Foundation (ASF) under one
|
3
|
-
* or more contributor license agreements. See the NOTICE file
|
4
|
-
* distributed with this work for additional information
|
5
|
-
* regarding copyright ownership. The ASF licenses this file
|
6
|
-
* to you under the Apache License, Version 2.0 (the
|
7
|
-
* "License"); you may not use this file except in compliance
|
8
|
-
* with the License. You may obtain a copy of the License at
|
9
|
-
*
|
10
|
-
* http://www.apache.org/licenses/LICENSE-2.0
|
11
|
-
*
|
12
|
-
* Unless required by applicable law or agreed to in writing,
|
13
|
-
* software distributed under the License is distributed on an
|
14
|
-
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
15
|
-
* KIND, either express or implied. See the License for the
|
16
|
-
* specific language governing permissions and limitations
|
17
|
-
* under the License.
|
18
|
-
*/
|
19
|
-
|
20
|
-
#include <string>
|
21
|
-
#include <fstream>
|
22
|
-
#include <iostream>
|
23
|
-
#include <vector>
|
24
|
-
|
25
|
-
#include <stdlib.h>
|
26
|
-
#include <sys/stat.h>
|
27
|
-
#include <sys/types.h>
|
28
|
-
#include <sstream>
|
29
|
-
#include <algorithm>
|
30
|
-
#include "t_generator.h"
|
31
|
-
#include "platform.h"
|
32
|
-
using namespace std;
|
33
|
-
|
34
|
-
|
35
|
-
/**
|
36
|
-
* Python code generator.
|
37
|
-
*
|
38
|
-
*/
|
39
|
-
class t_py_generator : public t_generator {
|
40
|
-
public:
|
41
|
-
t_py_generator(
|
42
|
-
t_program* program,
|
43
|
-
const std::map<std::string, std::string>& parsed_options,
|
44
|
-
const std::string& option_string)
|
45
|
-
: t_generator(program)
|
46
|
-
{
|
47
|
-
std::map<std::string, std::string>::const_iterator iter;
|
48
|
-
|
49
|
-
iter = parsed_options.find("new_style");
|
50
|
-
gen_newstyle_ = (iter != parsed_options.end());
|
51
|
-
|
52
|
-
iter = parsed_options.find("twisted");
|
53
|
-
gen_twisted_ = (iter != parsed_options.end());
|
54
|
-
|
55
|
-
out_dir_base_ = "gen-py";
|
56
|
-
}
|
57
|
-
|
58
|
-
/**
|
59
|
-
* Init and close methods
|
60
|
-
*/
|
61
|
-
|
62
|
-
void init_generator();
|
63
|
-
void close_generator();
|
64
|
-
|
65
|
-
/**
|
66
|
-
* Program-level generation functions
|
67
|
-
*/
|
68
|
-
|
69
|
-
void generate_typedef (t_typedef* ttypedef);
|
70
|
-
void generate_enum (t_enum* tenum);
|
71
|
-
void generate_const (t_const* tconst);
|
72
|
-
void generate_struct (t_struct* tstruct);
|
73
|
-
void generate_xception (t_struct* txception);
|
74
|
-
void generate_service (t_service* tservice);
|
75
|
-
|
76
|
-
std::string render_const_value(t_type* type, t_const_value* value);
|
77
|
-
|
78
|
-
/**
|
79
|
-
* Struct generation code
|
80
|
-
*/
|
81
|
-
|
82
|
-
void generate_py_struct(t_struct* tstruct, bool is_exception);
|
83
|
-
void generate_py_struct_definition(std::ofstream& out, t_struct* tstruct, bool is_xception=false, bool is_result=false);
|
84
|
-
void generate_py_struct_reader(std::ofstream& out, t_struct* tstruct);
|
85
|
-
void generate_py_struct_writer(std::ofstream& out, t_struct* tstruct);
|
86
|
-
void generate_py_function_helpers(t_function* tfunction);
|
87
|
-
|
88
|
-
/**
|
89
|
-
* Service-level generation functions
|
90
|
-
*/
|
91
|
-
|
92
|
-
void generate_service_helpers (t_service* tservice);
|
93
|
-
void generate_service_interface (t_service* tservice);
|
94
|
-
void generate_service_client (t_service* tservice);
|
95
|
-
void generate_service_remote (t_service* tservice);
|
96
|
-
void generate_service_server (t_service* tservice);
|
97
|
-
void generate_process_function (t_service* tservice, t_function* tfunction);
|
98
|
-
|
99
|
-
/**
|
100
|
-
* Serialization constructs
|
101
|
-
*/
|
102
|
-
|
103
|
-
void generate_deserialize_field (std::ofstream &out,
|
104
|
-
t_field* tfield,
|
105
|
-
std::string prefix="",
|
106
|
-
bool inclass=false);
|
107
|
-
|
108
|
-
void generate_deserialize_struct (std::ofstream &out,
|
109
|
-
t_struct* tstruct,
|
110
|
-
std::string prefix="");
|
111
|
-
|
112
|
-
void generate_deserialize_container (std::ofstream &out,
|
113
|
-
t_type* ttype,
|
114
|
-
std::string prefix="");
|
115
|
-
|
116
|
-
void generate_deserialize_set_element (std::ofstream &out,
|
117
|
-
t_set* tset,
|
118
|
-
std::string prefix="");
|
119
|
-
|
120
|
-
void generate_deserialize_map_element (std::ofstream &out,
|
121
|
-
t_map* tmap,
|
122
|
-
std::string prefix="");
|
123
|
-
|
124
|
-
void generate_deserialize_list_element (std::ofstream &out,
|
125
|
-
t_list* tlist,
|
126
|
-
std::string prefix="");
|
127
|
-
|
128
|
-
void generate_serialize_field (std::ofstream &out,
|
129
|
-
t_field* tfield,
|
130
|
-
std::string prefix="");
|
131
|
-
|
132
|
-
void generate_serialize_struct (std::ofstream &out,
|
133
|
-
t_struct* tstruct,
|
134
|
-
std::string prefix="");
|
135
|
-
|
136
|
-
void generate_serialize_container (std::ofstream &out,
|
137
|
-
t_type* ttype,
|
138
|
-
std::string prefix="");
|
139
|
-
|
140
|
-
void generate_serialize_map_element (std::ofstream &out,
|
141
|
-
t_map* tmap,
|
142
|
-
std::string kiter,
|
143
|
-
std::string viter);
|
144
|
-
|
145
|
-
void generate_serialize_set_element (std::ofstream &out,
|
146
|
-
t_set* tmap,
|
147
|
-
std::string iter);
|
148
|
-
|
149
|
-
void generate_serialize_list_element (std::ofstream &out,
|
150
|
-
t_list* tlist,
|
151
|
-
std::string iter);
|
152
|
-
|
153
|
-
void generate_python_docstring (std::ofstream& out,
|
154
|
-
t_struct* tstruct);
|
155
|
-
|
156
|
-
void generate_python_docstring (std::ofstream& out,
|
157
|
-
t_function* tfunction);
|
158
|
-
|
159
|
-
void generate_python_docstring (std::ofstream& out,
|
160
|
-
t_doc* tdoc,
|
161
|
-
t_struct* tstruct,
|
162
|
-
const char* subheader);
|
163
|
-
|
164
|
-
void generate_python_docstring (std::ofstream& out,
|
165
|
-
t_doc* tdoc);
|
166
|
-
|
167
|
-
/**
|
168
|
-
* Helper rendering functions
|
169
|
-
*/
|
170
|
-
|
171
|
-
std::string py_autogen_comment();
|
172
|
-
std::string py_imports();
|
173
|
-
std::string render_includes();
|
174
|
-
std::string render_fastbinary_includes();
|
175
|
-
std::string declare_argument(t_field* tfield);
|
176
|
-
std::string render_field_default_value(t_field* tfield);
|
177
|
-
std::string type_name(t_type* ttype);
|
178
|
-
std::string function_signature(t_function* tfunction, std::string prefix="");
|
179
|
-
std::string function_signature_if(t_function* tfunction, std::string prefix="");
|
180
|
-
std::string argument_list(t_struct* tstruct);
|
181
|
-
std::string type_to_enum(t_type* ttype);
|
182
|
-
std::string type_to_spec_args(t_type* ttype);
|
183
|
-
|
184
|
-
static std::string get_real_py_module(const t_program* program) {
|
185
|
-
std::string real_module = program->get_namespace("py");
|
186
|
-
if (real_module.empty()) {
|
187
|
-
return program->get_name();
|
188
|
-
}
|
189
|
-
return real_module;
|
190
|
-
}
|
191
|
-
|
192
|
-
private:
|
193
|
-
|
194
|
-
/**
|
195
|
-
* True iff we should generate new-style classes.
|
196
|
-
*/
|
197
|
-
bool gen_newstyle_;
|
198
|
-
|
199
|
-
/**
|
200
|
-
* True iff we should generate Twisted-friendly RPC services.
|
201
|
-
*/
|
202
|
-
bool gen_twisted_;
|
203
|
-
|
204
|
-
/**
|
205
|
-
* File streams
|
206
|
-
*/
|
207
|
-
|
208
|
-
std::ofstream f_types_;
|
209
|
-
std::ofstream f_consts_;
|
210
|
-
std::ofstream f_service_;
|
211
|
-
|
212
|
-
std::string package_dir_;
|
213
|
-
|
214
|
-
};
|
215
|
-
|
216
|
-
|
217
|
-
/**
|
218
|
-
* Prepares for file generation by opening up the necessary file output
|
219
|
-
* streams.
|
220
|
-
*
|
221
|
-
* @param tprogram The program to generate
|
222
|
-
*/
|
223
|
-
void t_py_generator::init_generator() {
|
224
|
-
// Make output directory
|
225
|
-
string module = get_real_py_module(program_);
|
226
|
-
package_dir_ = get_out_dir();
|
227
|
-
while (true) {
|
228
|
-
// TODO: Do better error checking here.
|
229
|
-
MKDIR(package_dir_.c_str());
|
230
|
-
std::ofstream init_py((package_dir_+"/__init__.py").c_str());
|
231
|
-
init_py.close();
|
232
|
-
if (module.empty()) {
|
233
|
-
break;
|
234
|
-
}
|
235
|
-
string::size_type pos = module.find('.');
|
236
|
-
if (pos == string::npos) {
|
237
|
-
package_dir_ += "/";
|
238
|
-
package_dir_ += module;
|
239
|
-
module.clear();
|
240
|
-
} else {
|
241
|
-
package_dir_ += "/";
|
242
|
-
package_dir_ += module.substr(0, pos);
|
243
|
-
module.erase(0, pos+1);
|
244
|
-
}
|
245
|
-
}
|
246
|
-
|
247
|
-
// Make output file
|
248
|
-
string f_types_name = package_dir_+"/"+"ttypes.py";
|
249
|
-
f_types_.open(f_types_name.c_str());
|
250
|
-
|
251
|
-
string f_consts_name = package_dir_+"/"+"constants.py";
|
252
|
-
f_consts_.open(f_consts_name.c_str());
|
253
|
-
|
254
|
-
string f_init_name = package_dir_+"/__init__.py";
|
255
|
-
ofstream f_init;
|
256
|
-
f_init.open(f_init_name.c_str());
|
257
|
-
f_init <<
|
258
|
-
"__all__ = ['ttypes', 'constants'";
|
259
|
-
vector<t_service*> services = program_->get_services();
|
260
|
-
vector<t_service*>::iterator sv_iter;
|
261
|
-
for (sv_iter = services.begin(); sv_iter != services.end(); ++sv_iter) {
|
262
|
-
f_init << ", '" << (*sv_iter)->get_name() << "'";
|
263
|
-
}
|
264
|
-
f_init << "]" << endl;
|
265
|
-
f_init.close();
|
266
|
-
|
267
|
-
// Print header
|
268
|
-
f_types_ <<
|
269
|
-
py_autogen_comment() << endl <<
|
270
|
-
py_imports() << endl <<
|
271
|
-
render_includes() << endl <<
|
272
|
-
render_fastbinary_includes() <<
|
273
|
-
endl << endl;
|
274
|
-
|
275
|
-
f_consts_ <<
|
276
|
-
py_autogen_comment() << endl <<
|
277
|
-
py_imports() << endl <<
|
278
|
-
"from ttypes import *" << endl <<
|
279
|
-
endl;
|
280
|
-
}
|
281
|
-
|
282
|
-
/**
|
283
|
-
* Renders all the imports necessary for including another Thrift program
|
284
|
-
*/
|
285
|
-
string t_py_generator::render_includes() {
|
286
|
-
const vector<t_program*>& includes = program_->get_includes();
|
287
|
-
string result = "";
|
288
|
-
for (size_t i = 0; i < includes.size(); ++i) {
|
289
|
-
result += "import " + get_real_py_module(includes[i]) + ".ttypes\n";
|
290
|
-
}
|
291
|
-
if (includes.size() > 0) {
|
292
|
-
result += "\n";
|
293
|
-
}
|
294
|
-
return result;
|
295
|
-
}
|
296
|
-
|
297
|
-
/**
|
298
|
-
* Renders all the imports necessary to use the accelerated TBinaryProtocol
|
299
|
-
*/
|
300
|
-
string t_py_generator::render_fastbinary_includes() {
|
301
|
-
return
|
302
|
-
"from thrift.transport import TTransport\n"
|
303
|
-
"from thrift.protocol import TBinaryProtocol\n"
|
304
|
-
"try:\n"
|
305
|
-
" from thrift.protocol import fastbinary\n"
|
306
|
-
"except:\n"
|
307
|
-
" fastbinary = None\n";
|
308
|
-
}
|
309
|
-
|
310
|
-
/**
|
311
|
-
* Autogen'd comment
|
312
|
-
*/
|
313
|
-
string t_py_generator::py_autogen_comment() {
|
314
|
-
return
|
315
|
-
std::string("#\n") +
|
316
|
-
"# Autogenerated by Thrift\n" +
|
317
|
-
"#\n" +
|
318
|
-
"# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING\n" +
|
319
|
-
"#\n";
|
320
|
-
}
|
321
|
-
|
322
|
-
/**
|
323
|
-
* Prints standard thrift imports
|
324
|
-
*/
|
325
|
-
string t_py_generator::py_imports() {
|
326
|
-
return
|
327
|
-
string("from thrift.Thrift import *");
|
328
|
-
}
|
329
|
-
|
330
|
-
/**
|
331
|
-
* Closes the type files
|
332
|
-
*/
|
333
|
-
void t_py_generator::close_generator() {
|
334
|
-
// Close types file
|
335
|
-
f_types_.close();
|
336
|
-
f_consts_.close();
|
337
|
-
}
|
338
|
-
|
339
|
-
/**
|
340
|
-
* Generates a typedef. This is not done in Python, types are all implicit.
|
341
|
-
*
|
342
|
-
* @param ttypedef The type definition
|
343
|
-
*/
|
344
|
-
void t_py_generator::generate_typedef(t_typedef* ttypedef) {}
|
345
|
-
|
346
|
-
/**
|
347
|
-
* Generates code for an enumerated type. Done using a class to scope
|
348
|
-
* the values.
|
349
|
-
*
|
350
|
-
* @param tenum The enumeration
|
351
|
-
*/
|
352
|
-
void t_py_generator::generate_enum(t_enum* tenum) {
|
353
|
-
f_types_ <<
|
354
|
-
"class " << tenum->get_name() <<
|
355
|
-
(gen_newstyle_ ? "(object)" : "") <<
|
356
|
-
":" << endl;
|
357
|
-
indent_up();
|
358
|
-
generate_python_docstring(f_types_, tenum);
|
359
|
-
|
360
|
-
vector<t_enum_value*> constants = tenum->get_constants();
|
361
|
-
vector<t_enum_value*>::iterator c_iter;
|
362
|
-
int value = -1;
|
363
|
-
for (c_iter = constants.begin(); c_iter != constants.end(); ++c_iter) {
|
364
|
-
if ((*c_iter)->has_value()) {
|
365
|
-
value = (*c_iter)->get_value();
|
366
|
-
} else {
|
367
|
-
++value;
|
368
|
-
}
|
369
|
-
|
370
|
-
f_types_ <<
|
371
|
-
indent() << (*c_iter)->get_name() << " = " << value << endl;
|
372
|
-
}
|
373
|
-
|
374
|
-
indent_down();
|
375
|
-
f_types_ << endl;
|
376
|
-
}
|
377
|
-
|
378
|
-
/**
|
379
|
-
* Generate a constant value
|
380
|
-
*/
|
381
|
-
void t_py_generator::generate_const(t_const* tconst) {
|
382
|
-
t_type* type = tconst->get_type();
|
383
|
-
string name = tconst->get_name();
|
384
|
-
t_const_value* value = tconst->get_value();
|
385
|
-
|
386
|
-
indent(f_consts_) << name << " = " << render_const_value(type, value);
|
387
|
-
f_consts_ << endl << endl;
|
388
|
-
}
|
389
|
-
|
390
|
-
/**
|
391
|
-
* Prints the value of a constant with the given type. Note that type checking
|
392
|
-
* is NOT performed in this function as it is always run beforehand using the
|
393
|
-
* validate_types method in main.cc
|
394
|
-
*/
|
395
|
-
string t_py_generator::render_const_value(t_type* type, t_const_value* value) {
|
396
|
-
type = get_true_type(type);
|
397
|
-
std::ostringstream out;
|
398
|
-
|
399
|
-
if (type->is_base_type()) {
|
400
|
-
t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
|
401
|
-
switch (tbase) {
|
402
|
-
case t_base_type::TYPE_STRING:
|
403
|
-
out << '"' << get_escaped_string(value) << '"';
|
404
|
-
break;
|
405
|
-
case t_base_type::TYPE_BOOL:
|
406
|
-
out << (value->get_integer() > 0 ? "True" : "False");
|
407
|
-
break;
|
408
|
-
case t_base_type::TYPE_BYTE:
|
409
|
-
case t_base_type::TYPE_I16:
|
410
|
-
case t_base_type::TYPE_I32:
|
411
|
-
case t_base_type::TYPE_I64:
|
412
|
-
out << value->get_integer();
|
413
|
-
break;
|
414
|
-
case t_base_type::TYPE_DOUBLE:
|
415
|
-
if (value->get_type() == t_const_value::CV_INTEGER) {
|
416
|
-
out << value->get_integer();
|
417
|
-
} else {
|
418
|
-
out << value->get_double();
|
419
|
-
}
|
420
|
-
break;
|
421
|
-
default:
|
422
|
-
throw "compiler error: no const of base type " + t_base_type::t_base_name(tbase);
|
423
|
-
}
|
424
|
-
} else if (type->is_enum()) {
|
425
|
-
indent(out) << value->get_integer();
|
426
|
-
} else if (type->is_struct() || type->is_xception()) {
|
427
|
-
out << type->get_name() << "(**{" << endl;
|
428
|
-
indent_up();
|
429
|
-
const vector<t_field*>& fields = ((t_struct*)type)->get_members();
|
430
|
-
vector<t_field*>::const_iterator f_iter;
|
431
|
-
const map<t_const_value*, t_const_value*>& val = value->get_map();
|
432
|
-
map<t_const_value*, t_const_value*>::const_iterator v_iter;
|
433
|
-
for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
|
434
|
-
t_type* field_type = NULL;
|
435
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
436
|
-
if ((*f_iter)->get_name() == v_iter->first->get_string()) {
|
437
|
-
field_type = (*f_iter)->get_type();
|
438
|
-
}
|
439
|
-
}
|
440
|
-
if (field_type == NULL) {
|
441
|
-
throw "type error: " + type->get_name() + " has no field " + v_iter->first->get_string();
|
442
|
-
}
|
443
|
-
out << indent();
|
444
|
-
out << render_const_value(g_type_string, v_iter->first);
|
445
|
-
out << " : ";
|
446
|
-
out << render_const_value(field_type, v_iter->second);
|
447
|
-
out << "," << endl;
|
448
|
-
}
|
449
|
-
indent_down();
|
450
|
-
indent(out) << "})";
|
451
|
-
} else if (type->is_map()) {
|
452
|
-
t_type* ktype = ((t_map*)type)->get_key_type();
|
453
|
-
t_type* vtype = ((t_map*)type)->get_val_type();
|
454
|
-
out << "{" << endl;
|
455
|
-
indent_up();
|
456
|
-
const map<t_const_value*, t_const_value*>& val = value->get_map();
|
457
|
-
map<t_const_value*, t_const_value*>::const_iterator v_iter;
|
458
|
-
for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
|
459
|
-
out << indent();
|
460
|
-
out << render_const_value(ktype, v_iter->first);
|
461
|
-
out << " : ";
|
462
|
-
out << render_const_value(vtype, v_iter->second);
|
463
|
-
out << "," << endl;
|
464
|
-
}
|
465
|
-
indent_down();
|
466
|
-
indent(out) << "}";
|
467
|
-
} else if (type->is_list() || type->is_set()) {
|
468
|
-
t_type* etype;
|
469
|
-
if (type->is_list()) {
|
470
|
-
etype = ((t_list*)type)->get_elem_type();
|
471
|
-
} else {
|
472
|
-
etype = ((t_set*)type)->get_elem_type();
|
473
|
-
}
|
474
|
-
if (type->is_set()) {
|
475
|
-
out << "set(";
|
476
|
-
}
|
477
|
-
out << "[" << endl;
|
478
|
-
indent_up();
|
479
|
-
const vector<t_const_value*>& val = value->get_list();
|
480
|
-
vector<t_const_value*>::const_iterator v_iter;
|
481
|
-
for (v_iter = val.begin(); v_iter != val.end(); ++v_iter) {
|
482
|
-
out << indent();
|
483
|
-
out << render_const_value(etype, *v_iter);
|
484
|
-
out << "," << endl;
|
485
|
-
}
|
486
|
-
indent_down();
|
487
|
-
indent(out) << "]";
|
488
|
-
if (type->is_set()) {
|
489
|
-
out << ")";
|
490
|
-
}
|
491
|
-
} else {
|
492
|
-
throw "CANNOT GENERATE CONSTANT FOR TYPE: " + type->get_name();
|
493
|
-
}
|
494
|
-
|
495
|
-
return out.str();
|
496
|
-
}
|
497
|
-
|
498
|
-
/**
|
499
|
-
* Generates a python struct
|
500
|
-
*/
|
501
|
-
void t_py_generator::generate_struct(t_struct* tstruct) {
|
502
|
-
generate_py_struct(tstruct, false);
|
503
|
-
}
|
504
|
-
|
505
|
-
/**
|
506
|
-
* Generates a struct definition for a thrift exception. Basically the same
|
507
|
-
* as a struct but extends the Exception class.
|
508
|
-
*
|
509
|
-
* @param txception The struct definition
|
510
|
-
*/
|
511
|
-
void t_py_generator::generate_xception(t_struct* txception) {
|
512
|
-
generate_py_struct(txception, true);
|
513
|
-
}
|
514
|
-
|
515
|
-
/**
|
516
|
-
* Generates a python struct
|
517
|
-
*/
|
518
|
-
void t_py_generator::generate_py_struct(t_struct* tstruct,
|
519
|
-
bool is_exception) {
|
520
|
-
generate_py_struct_definition(f_types_, tstruct, is_exception);
|
521
|
-
}
|
522
|
-
|
523
|
-
/**
|
524
|
-
* Generates a struct definition for a thrift data type.
|
525
|
-
*
|
526
|
-
* @param tstruct The struct definition
|
527
|
-
*/
|
528
|
-
void t_py_generator::generate_py_struct_definition(ofstream& out,
|
529
|
-
t_struct* tstruct,
|
530
|
-
bool is_exception,
|
531
|
-
bool is_result) {
|
532
|
-
|
533
|
-
const vector<t_field*>& members = tstruct->get_members();
|
534
|
-
const vector<t_field*>& sorted_members = tstruct->get_sorted_members();
|
535
|
-
vector<t_field*>::const_iterator m_iter;
|
536
|
-
|
537
|
-
out <<
|
538
|
-
"class " << tstruct->get_name();
|
539
|
-
if (is_exception) {
|
540
|
-
out << "(Exception)";
|
541
|
-
} else if (gen_newstyle_) {
|
542
|
-
out << "(object)";
|
543
|
-
}
|
544
|
-
out <<
|
545
|
-
":" << endl;
|
546
|
-
indent_up();
|
547
|
-
generate_python_docstring(out, tstruct);
|
548
|
-
|
549
|
-
out << endl;
|
550
|
-
|
551
|
-
/*
|
552
|
-
Here we generate the structure specification for the fastbinary codec.
|
553
|
-
These specifications have the following structure:
|
554
|
-
thrift_spec -> tuple of item_spec
|
555
|
-
item_spec -> None | (tag, type_enum, name, spec_args, default)
|
556
|
-
tag -> integer
|
557
|
-
type_enum -> TType.I32 | TType.STRING | TType.STRUCT | ...
|
558
|
-
name -> string_literal
|
559
|
-
default -> None # Handled by __init__
|
560
|
-
spec_args -> None # For simple types
|
561
|
-
| (type_enum, spec_args) # Value type for list/set
|
562
|
-
| (type_enum, spec_args, type_enum, spec_args)
|
563
|
-
# Key and value for map
|
564
|
-
| (class_name, spec_args_ptr) # For struct/exception
|
565
|
-
class_name -> identifier # Basically a pointer to the class
|
566
|
-
spec_args_ptr -> expression # just class_name.spec_args
|
567
|
-
|
568
|
-
TODO(dreiss): Consider making this work for structs with negative tags.
|
569
|
-
*/
|
570
|
-
|
571
|
-
// TODO(dreiss): Look into generating an empty tuple instead of None
|
572
|
-
// for structures with no members.
|
573
|
-
// TODO(dreiss): Test encoding of structs where some inner structs
|
574
|
-
// don't have thrift_spec.
|
575
|
-
if (sorted_members.empty() || (sorted_members[0]->get_key() >= 0)) {
|
576
|
-
indent(out) << "thrift_spec = (" << endl;
|
577
|
-
indent_up();
|
578
|
-
|
579
|
-
int sorted_keys_pos = 0;
|
580
|
-
for (m_iter = sorted_members.begin(); m_iter != sorted_members.end(); ++m_iter) {
|
581
|
-
|
582
|
-
for (; sorted_keys_pos != (*m_iter)->get_key(); sorted_keys_pos++) {
|
583
|
-
indent(out) << "None, # " << sorted_keys_pos << endl;
|
584
|
-
}
|
585
|
-
|
586
|
-
indent(out) << "(" << (*m_iter)->get_key() << ", "
|
587
|
-
<< type_to_enum((*m_iter)->get_type()) << ", "
|
588
|
-
<< "'" << (*m_iter)->get_name() << "'" << ", "
|
589
|
-
<< type_to_spec_args((*m_iter)->get_type()) << ", "
|
590
|
-
<< render_field_default_value(*m_iter) << ", "
|
591
|
-
<< "),"
|
592
|
-
<< " # " << sorted_keys_pos
|
593
|
-
<< endl;
|
594
|
-
|
595
|
-
sorted_keys_pos ++;
|
596
|
-
}
|
597
|
-
|
598
|
-
indent_down();
|
599
|
-
indent(out) << ")" << endl << endl;
|
600
|
-
} else {
|
601
|
-
indent(out) << "thrift_spec = None" << endl;
|
602
|
-
}
|
603
|
-
|
604
|
-
|
605
|
-
if (members.size() > 0) {
|
606
|
-
out <<
|
607
|
-
indent() << "def __init__(self,";
|
608
|
-
|
609
|
-
for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
|
610
|
-
// This fills in default values, as opposed to nulls
|
611
|
-
out << " " << declare_argument(*m_iter) << ",";
|
612
|
-
}
|
613
|
-
|
614
|
-
out << "):" << endl;
|
615
|
-
|
616
|
-
indent_up();
|
617
|
-
|
618
|
-
for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
|
619
|
-
// Initialize fields
|
620
|
-
t_type* type = (*m_iter)->get_type();
|
621
|
-
if (!type->is_base_type() && !type->is_enum() && (*m_iter)->get_value() != NULL) {
|
622
|
-
indent(out) <<
|
623
|
-
"if " << (*m_iter)->get_name() << " is " << "self.thrift_spec[" <<
|
624
|
-
(*m_iter)->get_key() << "][4]:" << endl;
|
625
|
-
indent(out) << " " << (*m_iter)->get_name() << " = " <<
|
626
|
-
render_field_default_value(*m_iter) << endl;
|
627
|
-
}
|
628
|
-
indent(out) <<
|
629
|
-
"self." << (*m_iter)->get_name() << " = " << (*m_iter)->get_name() << endl;
|
630
|
-
}
|
631
|
-
|
632
|
-
indent_down();
|
633
|
-
|
634
|
-
out << endl;
|
635
|
-
}
|
636
|
-
|
637
|
-
generate_py_struct_reader(out, tstruct);
|
638
|
-
generate_py_struct_writer(out, tstruct);
|
639
|
-
|
640
|
-
// For exceptions only, generate a __str__ method. This is
|
641
|
-
// because when raised exceptions are printed to the console, __repr__
|
642
|
-
// isn't used. See python bug #5882
|
643
|
-
if (is_exception) {
|
644
|
-
out <<
|
645
|
-
indent() << "def __str__(self):" << endl <<
|
646
|
-
indent() << " return repr(self)" << endl <<
|
647
|
-
endl;
|
648
|
-
}
|
649
|
-
|
650
|
-
// Printing utilities so that on the command line thrift
|
651
|
-
// structs look pretty like dictionaries
|
652
|
-
out <<
|
653
|
-
indent() << "def __repr__(self):" << endl <<
|
654
|
-
indent() << " L = ['%s=%r' % (key, value)" << endl <<
|
655
|
-
indent() << " for key, value in self.__dict__.iteritems()]" << endl <<
|
656
|
-
indent() << " return '%s(%s)' % (self.__class__.__name__, ', '.join(L))" << endl <<
|
657
|
-
endl;
|
658
|
-
|
659
|
-
// Equality and inequality methods that compare by value
|
660
|
-
out <<
|
661
|
-
indent() << "def __eq__(self, other):" << endl;
|
662
|
-
indent_up();
|
663
|
-
out <<
|
664
|
-
indent() << "return isinstance(other, self.__class__) and "
|
665
|
-
"self.__dict__ == other.__dict__" << endl;
|
666
|
-
indent_down();
|
667
|
-
out << endl;
|
668
|
-
|
669
|
-
out <<
|
670
|
-
indent() << "def __ne__(self, other):" << endl;
|
671
|
-
indent_up();
|
672
|
-
out <<
|
673
|
-
indent() << "return not (self == other)" << endl;
|
674
|
-
indent_down();
|
675
|
-
out << endl;
|
676
|
-
|
677
|
-
indent_down();
|
678
|
-
}
|
679
|
-
|
680
|
-
/**
|
681
|
-
* Generates the read method for a struct
|
682
|
-
*/
|
683
|
-
void t_py_generator::generate_py_struct_reader(ofstream& out,
|
684
|
-
t_struct* tstruct) {
|
685
|
-
const vector<t_field*>& fields = tstruct->get_members();
|
686
|
-
vector<t_field*>::const_iterator f_iter;
|
687
|
-
|
688
|
-
indent(out) <<
|
689
|
-
"def read(self, iprot):" << endl;
|
690
|
-
indent_up();
|
691
|
-
|
692
|
-
indent(out) <<
|
693
|
-
"if iprot.__class__ == TBinaryProtocol.TBinaryProtocolAccelerated "
|
694
|
-
"and isinstance(iprot.trans, TTransport.CReadableTransport) "
|
695
|
-
"and self.thrift_spec is not None "
|
696
|
-
"and fastbinary is not None:" << endl;
|
697
|
-
indent_up();
|
698
|
-
|
699
|
-
indent(out) <<
|
700
|
-
"fastbinary.decode_binary(self, iprot.trans, (self.__class__, self.thrift_spec))" << endl;
|
701
|
-
indent(out) <<
|
702
|
-
"return" << endl;
|
703
|
-
indent_down();
|
704
|
-
|
705
|
-
indent(out) <<
|
706
|
-
"iprot.readStructBegin()" << endl;
|
707
|
-
|
708
|
-
// Loop over reading in fields
|
709
|
-
indent(out) <<
|
710
|
-
"while True:" << endl;
|
711
|
-
indent_up();
|
712
|
-
|
713
|
-
// Read beginning field marker
|
714
|
-
indent(out) <<
|
715
|
-
"(fname, ftype, fid) = iprot.readFieldBegin()" << endl;
|
716
|
-
|
717
|
-
// Check for field STOP marker and break
|
718
|
-
indent(out) <<
|
719
|
-
"if ftype == TType.STOP:" << endl;
|
720
|
-
indent_up();
|
721
|
-
indent(out) <<
|
722
|
-
"break" << endl;
|
723
|
-
indent_down();
|
724
|
-
|
725
|
-
// Switch statement on the field we are reading
|
726
|
-
bool first = true;
|
727
|
-
|
728
|
-
// Generate deserialization code for known cases
|
729
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
730
|
-
if (first) {
|
731
|
-
first = false;
|
732
|
-
out <<
|
733
|
-
indent() << "if ";
|
734
|
-
} else {
|
735
|
-
out <<
|
736
|
-
indent() << "elif ";
|
737
|
-
}
|
738
|
-
out << "fid == " << (*f_iter)->get_key() << ":" << endl;
|
739
|
-
indent_up();
|
740
|
-
indent(out) << "if ftype == " << type_to_enum((*f_iter)->get_type()) << ":" << endl;
|
741
|
-
indent_up();
|
742
|
-
generate_deserialize_field(out, *f_iter, "self.");
|
743
|
-
indent_down();
|
744
|
-
out <<
|
745
|
-
indent() << "else:" << endl <<
|
746
|
-
indent() << " iprot.skip(ftype)" << endl;
|
747
|
-
indent_down();
|
748
|
-
}
|
749
|
-
|
750
|
-
// In the default case we skip the field
|
751
|
-
out <<
|
752
|
-
indent() << "else:" << endl <<
|
753
|
-
indent() << " iprot.skip(ftype)" << endl;
|
754
|
-
|
755
|
-
// Read field end marker
|
756
|
-
indent(out) <<
|
757
|
-
"iprot.readFieldEnd()" << endl;
|
758
|
-
|
759
|
-
indent_down();
|
760
|
-
|
761
|
-
indent(out) <<
|
762
|
-
"iprot.readStructEnd()" << endl;
|
763
|
-
|
764
|
-
indent_down();
|
765
|
-
out << endl;
|
766
|
-
}
|
767
|
-
|
768
|
-
void t_py_generator::generate_py_struct_writer(ofstream& out,
|
769
|
-
t_struct* tstruct) {
|
770
|
-
string name = tstruct->get_name();
|
771
|
-
const vector<t_field*>& fields = tstruct->get_sorted_members();
|
772
|
-
vector<t_field*>::const_iterator f_iter;
|
773
|
-
|
774
|
-
indent(out) <<
|
775
|
-
"def write(self, oprot):" << endl;
|
776
|
-
indent_up();
|
777
|
-
|
778
|
-
indent(out) <<
|
779
|
-
"if oprot.__class__ == TBinaryProtocol.TBinaryProtocolAccelerated "
|
780
|
-
"and self.thrift_spec is not None "
|
781
|
-
"and fastbinary is not None:" << endl;
|
782
|
-
indent_up();
|
783
|
-
|
784
|
-
indent(out) <<
|
785
|
-
"oprot.trans.write(fastbinary.encode_binary(self, (self.__class__, self.thrift_spec)))" << endl;
|
786
|
-
indent(out) <<
|
787
|
-
"return" << endl;
|
788
|
-
indent_down();
|
789
|
-
|
790
|
-
indent(out) <<
|
791
|
-
"oprot.writeStructBegin('" << name << "')" << endl;
|
792
|
-
|
793
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
794
|
-
// Write field header
|
795
|
-
indent(out) <<
|
796
|
-
"if self." << (*f_iter)->get_name() << " != None:" << endl;
|
797
|
-
indent_up();
|
798
|
-
indent(out) <<
|
799
|
-
"oprot.writeFieldBegin(" <<
|
800
|
-
"'" << (*f_iter)->get_name() << "', " <<
|
801
|
-
type_to_enum((*f_iter)->get_type()) << ", " <<
|
802
|
-
(*f_iter)->get_key() << ")" << endl;
|
803
|
-
|
804
|
-
// Write field contents
|
805
|
-
generate_serialize_field(out, *f_iter, "self.");
|
806
|
-
|
807
|
-
// Write field closer
|
808
|
-
indent(out) <<
|
809
|
-
"oprot.writeFieldEnd()" << endl;
|
810
|
-
|
811
|
-
indent_down();
|
812
|
-
}
|
813
|
-
|
814
|
-
// Write the struct map
|
815
|
-
out <<
|
816
|
-
indent() << "oprot.writeFieldStop()" << endl <<
|
817
|
-
indent() << "oprot.writeStructEnd()" << endl;
|
818
|
-
|
819
|
-
indent_down();
|
820
|
-
out <<
|
821
|
-
endl;
|
822
|
-
}
|
823
|
-
|
824
|
-
/**
|
825
|
-
* Generates a thrift service.
|
826
|
-
*
|
827
|
-
* @param tservice The service definition
|
828
|
-
*/
|
829
|
-
void t_py_generator::generate_service(t_service* tservice) {
|
830
|
-
string f_service_name = package_dir_+"/"+service_name_+".py";
|
831
|
-
f_service_.open(f_service_name.c_str());
|
832
|
-
|
833
|
-
f_service_ <<
|
834
|
-
py_autogen_comment() << endl <<
|
835
|
-
py_imports() << endl;
|
836
|
-
|
837
|
-
if (tservice->get_extends() != NULL) {
|
838
|
-
f_service_ <<
|
839
|
-
"import " << get_real_py_module(tservice->get_extends()->get_program()) <<
|
840
|
-
"." << tservice->get_extends()->get_name() << endl;
|
841
|
-
}
|
842
|
-
|
843
|
-
f_service_ <<
|
844
|
-
"from ttypes import *" << endl <<
|
845
|
-
"from thrift.Thrift import TProcessor" << endl <<
|
846
|
-
render_fastbinary_includes() << endl;
|
847
|
-
|
848
|
-
if (gen_twisted_) {
|
849
|
-
f_service_ <<
|
850
|
-
"from zope.interface import Interface, implements" << endl <<
|
851
|
-
"from twisted.internet import defer" << endl <<
|
852
|
-
"from thrift.transport import TTwisted" << endl;
|
853
|
-
}
|
854
|
-
|
855
|
-
f_service_ << endl;
|
856
|
-
|
857
|
-
// Generate the three main parts of the service (well, two for now in PHP)
|
858
|
-
generate_service_interface(tservice);
|
859
|
-
generate_service_client(tservice);
|
860
|
-
generate_service_server(tservice);
|
861
|
-
generate_service_helpers(tservice);
|
862
|
-
generate_service_remote(tservice);
|
863
|
-
|
864
|
-
// Close service file
|
865
|
-
f_service_ << endl;
|
866
|
-
f_service_.close();
|
867
|
-
}
|
868
|
-
|
869
|
-
/**
|
870
|
-
* Generates helper functions for a service.
|
871
|
-
*
|
872
|
-
* @param tservice The service to generate a header definition for
|
873
|
-
*/
|
874
|
-
void t_py_generator::generate_service_helpers(t_service* tservice) {
|
875
|
-
vector<t_function*> functions = tservice->get_functions();
|
876
|
-
vector<t_function*>::iterator f_iter;
|
877
|
-
|
878
|
-
f_service_ <<
|
879
|
-
"# HELPER FUNCTIONS AND STRUCTURES" << endl << endl;
|
880
|
-
|
881
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
882
|
-
t_struct* ts = (*f_iter)->get_arglist();
|
883
|
-
generate_py_struct_definition(f_service_, ts, false);
|
884
|
-
generate_py_function_helpers(*f_iter);
|
885
|
-
}
|
886
|
-
}
|
887
|
-
|
888
|
-
/**
|
889
|
-
* Generates a struct and helpers for a function.
|
890
|
-
*
|
891
|
-
* @param tfunction The function
|
892
|
-
*/
|
893
|
-
void t_py_generator::generate_py_function_helpers(t_function* tfunction) {
|
894
|
-
if (!tfunction->is_oneway()) {
|
895
|
-
t_struct result(program_, tfunction->get_name() + "_result");
|
896
|
-
t_field success(tfunction->get_returntype(), "success", 0);
|
897
|
-
if (!tfunction->get_returntype()->is_void()) {
|
898
|
-
result.append(&success);
|
899
|
-
}
|
900
|
-
|
901
|
-
t_struct* xs = tfunction->get_xceptions();
|
902
|
-
const vector<t_field*>& fields = xs->get_members();
|
903
|
-
vector<t_field*>::const_iterator f_iter;
|
904
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
905
|
-
result.append(*f_iter);
|
906
|
-
}
|
907
|
-
generate_py_struct_definition(f_service_, &result, false, true);
|
908
|
-
}
|
909
|
-
}
|
910
|
-
|
911
|
-
/**
|
912
|
-
* Generates a service interface definition.
|
913
|
-
*
|
914
|
-
* @param tservice The service to generate a header definition for
|
915
|
-
*/
|
916
|
-
void t_py_generator::generate_service_interface(t_service* tservice) {
|
917
|
-
string extends = "";
|
918
|
-
string extends_if = "";
|
919
|
-
if (tservice->get_extends() != NULL) {
|
920
|
-
extends = type_name(tservice->get_extends());
|
921
|
-
extends_if = "(" + extends + ".Iface)";
|
922
|
-
} else {
|
923
|
-
if (gen_twisted_) {
|
924
|
-
extends_if = "(Interface)";
|
925
|
-
}
|
926
|
-
}
|
927
|
-
|
928
|
-
f_service_ <<
|
929
|
-
"class Iface" << extends_if << ":" << endl;
|
930
|
-
indent_up();
|
931
|
-
generate_python_docstring(f_service_, tservice);
|
932
|
-
vector<t_function*> functions = tservice->get_functions();
|
933
|
-
if (functions.empty()) {
|
934
|
-
f_service_ <<
|
935
|
-
indent() << "pass" << endl;
|
936
|
-
} else {
|
937
|
-
vector<t_function*>::iterator f_iter;
|
938
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
939
|
-
f_service_ <<
|
940
|
-
indent() << "def " << function_signature_if(*f_iter) << ":" << endl;
|
941
|
-
indent_up();
|
942
|
-
generate_python_docstring(f_service_, (*f_iter));
|
943
|
-
f_service_ <<
|
944
|
-
indent() << "pass" << endl << endl;
|
945
|
-
indent_down();
|
946
|
-
}
|
947
|
-
}
|
948
|
-
|
949
|
-
indent_down();
|
950
|
-
f_service_ <<
|
951
|
-
endl;
|
952
|
-
}
|
953
|
-
|
954
|
-
/**
|
955
|
-
* Generates a service client definition.
|
956
|
-
*
|
957
|
-
* @param tservice The service to generate a server for.
|
958
|
-
*/
|
959
|
-
void t_py_generator::generate_service_client(t_service* tservice) {
|
960
|
-
string extends = "";
|
961
|
-
string extends_client = "";
|
962
|
-
if (tservice->get_extends() != NULL) {
|
963
|
-
extends = type_name(tservice->get_extends());
|
964
|
-
if (gen_twisted_) {
|
965
|
-
extends_client = "(" + extends + ".Client)";
|
966
|
-
} else {
|
967
|
-
extends_client = extends + ".Client, ";
|
968
|
-
}
|
969
|
-
} else {
|
970
|
-
if (gen_twisted_ && gen_newstyle_) {
|
971
|
-
extends_client = "(object)";
|
972
|
-
}
|
973
|
-
}
|
974
|
-
|
975
|
-
if (gen_twisted_) {
|
976
|
-
f_service_ <<
|
977
|
-
"class Client" << extends_client << ":" << endl <<
|
978
|
-
" implements(Iface)" << endl << endl;
|
979
|
-
} else {
|
980
|
-
f_service_ <<
|
981
|
-
"class Client(" << extends_client << "Iface):" << endl;
|
982
|
-
}
|
983
|
-
indent_up();
|
984
|
-
generate_python_docstring(f_service_, tservice);
|
985
|
-
|
986
|
-
// Constructor function
|
987
|
-
if (gen_twisted_) {
|
988
|
-
f_service_ <<
|
989
|
-
indent() << "def __init__(self, transport, oprot_factory):" << endl;
|
990
|
-
} else {
|
991
|
-
f_service_ <<
|
992
|
-
indent() << "def __init__(self, iprot, oprot=None):" << endl;
|
993
|
-
}
|
994
|
-
if (extends.empty()) {
|
995
|
-
if (gen_twisted_) {
|
996
|
-
f_service_ <<
|
997
|
-
indent() << " self._transport = transport" << endl <<
|
998
|
-
indent() << " self._oprot_factory = oprot_factory" << endl <<
|
999
|
-
indent() << " self._seqid = 0" << endl <<
|
1000
|
-
indent() << " self._reqs = {}" << endl <<
|
1001
|
-
endl;
|
1002
|
-
} else {
|
1003
|
-
f_service_ <<
|
1004
|
-
indent() << " self._iprot = self._oprot = iprot" << endl <<
|
1005
|
-
indent() << " if oprot != None:" << endl <<
|
1006
|
-
indent() << " self._oprot = oprot" << endl <<
|
1007
|
-
indent() << " self._seqid = 0" << endl <<
|
1008
|
-
endl;
|
1009
|
-
}
|
1010
|
-
} else {
|
1011
|
-
if (gen_twisted_) {
|
1012
|
-
f_service_ <<
|
1013
|
-
indent() << " " << extends << ".Client.__init__(self, transport, oprot_factory)" << endl <<
|
1014
|
-
endl;
|
1015
|
-
} else {
|
1016
|
-
f_service_ <<
|
1017
|
-
indent() << " " << extends << ".Client.__init__(self, iprot, oprot)" << endl <<
|
1018
|
-
endl;
|
1019
|
-
}
|
1020
|
-
}
|
1021
|
-
|
1022
|
-
// Generate client method implementations
|
1023
|
-
vector<t_function*> functions = tservice->get_functions();
|
1024
|
-
vector<t_function*>::const_iterator f_iter;
|
1025
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
1026
|
-
t_struct* arg_struct = (*f_iter)->get_arglist();
|
1027
|
-
const vector<t_field*>& fields = arg_struct->get_members();
|
1028
|
-
vector<t_field*>::const_iterator fld_iter;
|
1029
|
-
string funname = (*f_iter)->get_name();
|
1030
|
-
|
1031
|
-
// Open function
|
1032
|
-
indent(f_service_) <<
|
1033
|
-
"def " << function_signature(*f_iter) << ":" << endl;
|
1034
|
-
indent_up();
|
1035
|
-
generate_python_docstring(f_service_, (*f_iter));
|
1036
|
-
if (gen_twisted_) {
|
1037
|
-
indent(f_service_) << "self._seqid += 1" << endl;
|
1038
|
-
if (!(*f_iter)->is_oneway()) {
|
1039
|
-
indent(f_service_) <<
|
1040
|
-
"d = self._reqs[self._seqid] = defer.Deferred()" << endl;
|
1041
|
-
}
|
1042
|
-
}
|
1043
|
-
|
1044
|
-
indent(f_service_) <<
|
1045
|
-
"self.send_" << funname << "(";
|
1046
|
-
|
1047
|
-
bool first = true;
|
1048
|
-
for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) {
|
1049
|
-
if (first) {
|
1050
|
-
first = false;
|
1051
|
-
} else {
|
1052
|
-
f_service_ << ", ";
|
1053
|
-
}
|
1054
|
-
f_service_ << (*fld_iter)->get_name();
|
1055
|
-
}
|
1056
|
-
f_service_ << ")" << endl;
|
1057
|
-
|
1058
|
-
if (!(*f_iter)->is_oneway()) {
|
1059
|
-
f_service_ << indent();
|
1060
|
-
if (gen_twisted_) {
|
1061
|
-
f_service_ << "return d" << endl;
|
1062
|
-
} else {
|
1063
|
-
if (!(*f_iter)->get_returntype()->is_void()) {
|
1064
|
-
f_service_ << "return ";
|
1065
|
-
}
|
1066
|
-
f_service_ <<
|
1067
|
-
"self.recv_" << funname << "()" << endl;
|
1068
|
-
}
|
1069
|
-
} else {
|
1070
|
-
if (gen_twisted_) {
|
1071
|
-
f_service_ <<
|
1072
|
-
indent() << "return defer.succeed(None)" << endl;
|
1073
|
-
}
|
1074
|
-
}
|
1075
|
-
indent_down();
|
1076
|
-
f_service_ << endl;
|
1077
|
-
|
1078
|
-
indent(f_service_) <<
|
1079
|
-
"def send_" << function_signature(*f_iter) << ":" << endl;
|
1080
|
-
|
1081
|
-
indent_up();
|
1082
|
-
|
1083
|
-
std::string argsname = (*f_iter)->get_name() + "_args";
|
1084
|
-
|
1085
|
-
// Serialize the request header
|
1086
|
-
if (gen_twisted_) {
|
1087
|
-
f_service_ <<
|
1088
|
-
indent() << "oprot = self._oprot_factory.getProtocol(self._transport)" << endl <<
|
1089
|
-
indent() <<
|
1090
|
-
"oprot.writeMessageBegin('" << (*f_iter)->get_name() << "', TMessageType.CALL, self._seqid)"
|
1091
|
-
<< endl;
|
1092
|
-
} else {
|
1093
|
-
f_service_ <<
|
1094
|
-
indent() << "self._oprot.writeMessageBegin('" << (*f_iter)->get_name() << "', TMessageType.CALL, self._seqid)" << endl;
|
1095
|
-
}
|
1096
|
-
|
1097
|
-
f_service_ <<
|
1098
|
-
indent() << "args = " << argsname << "()" << endl;
|
1099
|
-
|
1100
|
-
for (fld_iter = fields.begin(); fld_iter != fields.end(); ++fld_iter) {
|
1101
|
-
f_service_ <<
|
1102
|
-
indent() << "args." << (*fld_iter)->get_name() << " = " << (*fld_iter)->get_name() << endl;
|
1103
|
-
}
|
1104
|
-
|
1105
|
-
// Write to the stream
|
1106
|
-
if (gen_twisted_) {
|
1107
|
-
f_service_ <<
|
1108
|
-
indent() << "args.write(oprot)" << endl <<
|
1109
|
-
indent() << "oprot.writeMessageEnd()" << endl <<
|
1110
|
-
indent() << "oprot.trans.flush()" << endl;
|
1111
|
-
} else {
|
1112
|
-
f_service_ <<
|
1113
|
-
indent() << "args.write(self._oprot)" << endl <<
|
1114
|
-
indent() << "self._oprot.writeMessageEnd()" << endl <<
|
1115
|
-
indent() << "self._oprot.trans.flush()" << endl;
|
1116
|
-
}
|
1117
|
-
|
1118
|
-
indent_down();
|
1119
|
-
|
1120
|
-
if (!(*f_iter)->is_oneway()) {
|
1121
|
-
std::string resultname = (*f_iter)->get_name() + "_result";
|
1122
|
-
// Open function
|
1123
|
-
f_service_ <<
|
1124
|
-
endl;
|
1125
|
-
if (gen_twisted_) {
|
1126
|
-
f_service_ <<
|
1127
|
-
indent() << "def recv_" << (*f_iter)->get_name() <<
|
1128
|
-
"(self, iprot, mtype, rseqid):" << endl;
|
1129
|
-
} else {
|
1130
|
-
t_struct noargs(program_);
|
1131
|
-
t_function recv_function((*f_iter)->get_returntype(),
|
1132
|
-
string("recv_") + (*f_iter)->get_name(),
|
1133
|
-
&noargs);
|
1134
|
-
f_service_ <<
|
1135
|
-
indent() << "def " << function_signature(&recv_function) << ":" << endl;
|
1136
|
-
}
|
1137
|
-
indent_up();
|
1138
|
-
|
1139
|
-
// TODO(mcslee): Validate message reply here, seq ids etc.
|
1140
|
-
|
1141
|
-
if (gen_twisted_) {
|
1142
|
-
f_service_ <<
|
1143
|
-
indent() << "d = self._reqs.pop(rseqid)" << endl;
|
1144
|
-
} else {
|
1145
|
-
f_service_ <<
|
1146
|
-
indent() << "(fname, mtype, rseqid) = self._iprot.readMessageBegin()" << endl;
|
1147
|
-
}
|
1148
|
-
|
1149
|
-
f_service_ <<
|
1150
|
-
indent() << "if mtype == TMessageType.EXCEPTION:" << endl <<
|
1151
|
-
indent() << " x = TApplicationException()" << endl;
|
1152
|
-
|
1153
|
-
if (gen_twisted_) {
|
1154
|
-
f_service_ <<
|
1155
|
-
indent() << " x.read(iprot)" << endl <<
|
1156
|
-
indent() << " iprot.readMessageEnd()" << endl <<
|
1157
|
-
indent() << " return d.errback(x)" << endl <<
|
1158
|
-
indent() << "result = " << resultname << "()" << endl <<
|
1159
|
-
indent() << "result.read(iprot)" << endl <<
|
1160
|
-
indent() << "iprot.readMessageEnd()" << endl;
|
1161
|
-
} else {
|
1162
|
-
f_service_ <<
|
1163
|
-
indent() << " x.read(self._iprot)" << endl <<
|
1164
|
-
indent() << " self._iprot.readMessageEnd()" << endl <<
|
1165
|
-
indent() << " raise x" << endl <<
|
1166
|
-
indent() << "result = " << resultname << "()" << endl <<
|
1167
|
-
indent() << "result.read(self._iprot)" << endl <<
|
1168
|
-
indent() << "self._iprot.readMessageEnd()" << endl;
|
1169
|
-
}
|
1170
|
-
|
1171
|
-
// Careful, only return _result if not a void function
|
1172
|
-
if (!(*f_iter)->get_returntype()->is_void()) {
|
1173
|
-
f_service_ <<
|
1174
|
-
indent() << "if result.success != None:" << endl;
|
1175
|
-
if (gen_twisted_) {
|
1176
|
-
f_service_ <<
|
1177
|
-
indent() << " return d.callback(result.success)" << endl;
|
1178
|
-
} else {
|
1179
|
-
f_service_ <<
|
1180
|
-
indent() << " return result.success" << endl;
|
1181
|
-
}
|
1182
|
-
}
|
1183
|
-
|
1184
|
-
t_struct* xs = (*f_iter)->get_xceptions();
|
1185
|
-
const std::vector<t_field*>& xceptions = xs->get_members();
|
1186
|
-
vector<t_field*>::const_iterator x_iter;
|
1187
|
-
for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
|
1188
|
-
f_service_ <<
|
1189
|
-
indent() << "if result." << (*x_iter)->get_name() << " != None:" << endl;
|
1190
|
-
if (gen_twisted_) {
|
1191
|
-
f_service_ <<
|
1192
|
-
indent() << " return d.errback(result." << (*x_iter)->get_name() << ")" << endl;
|
1193
|
-
|
1194
|
-
} else {
|
1195
|
-
f_service_ <<
|
1196
|
-
indent() << " raise result." << (*x_iter)->get_name() << "" << endl;
|
1197
|
-
}
|
1198
|
-
}
|
1199
|
-
|
1200
|
-
// Careful, only return _result if not a void function
|
1201
|
-
if ((*f_iter)->get_returntype()->is_void()) {
|
1202
|
-
if (gen_twisted_) {
|
1203
|
-
indent(f_service_) <<
|
1204
|
-
"return d.callback(None)" << endl;
|
1205
|
-
} else {
|
1206
|
-
indent(f_service_) <<
|
1207
|
-
"return" << endl;
|
1208
|
-
}
|
1209
|
-
} else {
|
1210
|
-
if (gen_twisted_) {
|
1211
|
-
f_service_ <<
|
1212
|
-
indent() << "return d.errback(TApplicationException(TApplicationException.MISSING_RESULT, \"" << (*f_iter)->get_name() << " failed: unknown result\"))" << endl;
|
1213
|
-
} else {
|
1214
|
-
f_service_ <<
|
1215
|
-
indent() << "raise TApplicationException(TApplicationException.MISSING_RESULT, \"" << (*f_iter)->get_name() << " failed: unknown result\");" << endl;
|
1216
|
-
}
|
1217
|
-
}
|
1218
|
-
|
1219
|
-
// Close function
|
1220
|
-
indent_down();
|
1221
|
-
f_service_ << endl;
|
1222
|
-
}
|
1223
|
-
}
|
1224
|
-
|
1225
|
-
indent_down();
|
1226
|
-
f_service_ <<
|
1227
|
-
endl;
|
1228
|
-
}
|
1229
|
-
|
1230
|
-
/**
|
1231
|
-
* Generates a command line tool for making remote requests
|
1232
|
-
*
|
1233
|
-
* @param tservice The service to generate a remote for.
|
1234
|
-
*/
|
1235
|
-
void t_py_generator::generate_service_remote(t_service* tservice) {
|
1236
|
-
vector<t_function*> functions = tservice->get_functions();
|
1237
|
-
vector<t_function*>::iterator f_iter;
|
1238
|
-
|
1239
|
-
string f_remote_name = package_dir_+"/"+service_name_+"-remote";
|
1240
|
-
ofstream f_remote;
|
1241
|
-
f_remote.open(f_remote_name.c_str());
|
1242
|
-
|
1243
|
-
f_remote <<
|
1244
|
-
"#!/usr/bin/env python" << endl <<
|
1245
|
-
py_autogen_comment() << endl <<
|
1246
|
-
"import sys" << endl <<
|
1247
|
-
"import pprint" << endl <<
|
1248
|
-
"from urlparse import urlparse" << endl <<
|
1249
|
-
"from thrift.transport import TTransport" << endl <<
|
1250
|
-
"from thrift.transport import TSocket" << endl <<
|
1251
|
-
"from thrift.transport import THttpClient" << endl <<
|
1252
|
-
"from thrift.protocol import TBinaryProtocol" << endl <<
|
1253
|
-
endl;
|
1254
|
-
|
1255
|
-
f_remote <<
|
1256
|
-
"import " << service_name_ << endl <<
|
1257
|
-
"from ttypes import *" << endl <<
|
1258
|
-
endl;
|
1259
|
-
|
1260
|
-
f_remote <<
|
1261
|
-
"if len(sys.argv) <= 1 or sys.argv[1] == '--help':" << endl <<
|
1262
|
-
" print ''" << endl <<
|
1263
|
-
" print 'Usage: ' + sys.argv[0] + ' [-h host:port] [-u url] [-f[ramed]] function [arg1 [arg2...]]'" << endl <<
|
1264
|
-
" print ''" << endl <<
|
1265
|
-
" print 'Functions:'" << endl;
|
1266
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
1267
|
-
f_remote <<
|
1268
|
-
" print ' " << (*f_iter)->get_returntype()->get_name() << " " << (*f_iter)->get_name() << "(";
|
1269
|
-
t_struct* arg_struct = (*f_iter)->get_arglist();
|
1270
|
-
const std::vector<t_field*>& args = arg_struct->get_members();
|
1271
|
-
vector<t_field*>::const_iterator a_iter;
|
1272
|
-
int num_args = args.size();
|
1273
|
-
bool first = true;
|
1274
|
-
for (int i = 0; i < num_args; ++i) {
|
1275
|
-
if (first) {
|
1276
|
-
first = false;
|
1277
|
-
} else {
|
1278
|
-
f_remote << ", ";
|
1279
|
-
}
|
1280
|
-
f_remote <<
|
1281
|
-
args[i]->get_type()->get_name() << " " << args[i]->get_name();
|
1282
|
-
}
|
1283
|
-
f_remote << ")'" << endl;
|
1284
|
-
}
|
1285
|
-
f_remote <<
|
1286
|
-
" print ''" << endl <<
|
1287
|
-
" sys.exit(0)" << endl <<
|
1288
|
-
endl;
|
1289
|
-
|
1290
|
-
f_remote <<
|
1291
|
-
"pp = pprint.PrettyPrinter(indent = 2)" << endl <<
|
1292
|
-
"host = 'localhost'" << endl <<
|
1293
|
-
"port = 9090" << endl <<
|
1294
|
-
"uri = ''" << endl <<
|
1295
|
-
"framed = False" << endl <<
|
1296
|
-
"http = False" << endl <<
|
1297
|
-
"argi = 1" << endl <<
|
1298
|
-
endl <<
|
1299
|
-
"if sys.argv[argi] == '-h':" << endl <<
|
1300
|
-
" parts = sys.argv[argi+1].split(':') " << endl <<
|
1301
|
-
" host = parts[0]" << endl <<
|
1302
|
-
" port = int(parts[1])" << endl <<
|
1303
|
-
" argi += 2" << endl <<
|
1304
|
-
endl <<
|
1305
|
-
"if sys.argv[argi] == '-u':" << endl <<
|
1306
|
-
" url = urlparse(sys.argv[argi+1])" << endl <<
|
1307
|
-
" parts = url[1].split(':') " << endl <<
|
1308
|
-
" host = parts[0]" << endl <<
|
1309
|
-
" if len(parts) > 1:" << endl <<
|
1310
|
-
" port = int(parts[1])" << endl <<
|
1311
|
-
" else:" << endl <<
|
1312
|
-
" port = 80" << endl <<
|
1313
|
-
" uri = url[2]" << endl <<
|
1314
|
-
" http = True" << endl <<
|
1315
|
-
" argi += 2" << endl <<
|
1316
|
-
endl <<
|
1317
|
-
"if sys.argv[argi] == '-f' or sys.argv[argi] == '-framed':" << endl <<
|
1318
|
-
" framed = True" << endl <<
|
1319
|
-
" argi += 1" << endl <<
|
1320
|
-
endl <<
|
1321
|
-
"cmd = sys.argv[argi]" << endl <<
|
1322
|
-
"args = sys.argv[argi+1:]" << endl <<
|
1323
|
-
endl <<
|
1324
|
-
"if http:" << endl <<
|
1325
|
-
" transport = THttpClient.THttpClient(host, port, uri)" << endl <<
|
1326
|
-
"else:" << endl <<
|
1327
|
-
" socket = TSocket.TSocket(host, port)" << endl <<
|
1328
|
-
" if framed:" << endl <<
|
1329
|
-
" transport = TTransport.TFramedTransport(socket)" << endl <<
|
1330
|
-
" else:" << endl <<
|
1331
|
-
" transport = TTransport.TBufferedTransport(socket)" << endl <<
|
1332
|
-
"protocol = TBinaryProtocol.TBinaryProtocol(transport)" << endl <<
|
1333
|
-
"client = " << service_name_ << ".Client(protocol)" << endl <<
|
1334
|
-
"transport.open()" << endl <<
|
1335
|
-
endl;
|
1336
|
-
|
1337
|
-
// Generate the dispatch methods
|
1338
|
-
bool first = true;
|
1339
|
-
|
1340
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
1341
|
-
if (first) {
|
1342
|
-
first = false;
|
1343
|
-
} else {
|
1344
|
-
f_remote << "el";
|
1345
|
-
}
|
1346
|
-
|
1347
|
-
t_struct* arg_struct = (*f_iter)->get_arglist();
|
1348
|
-
const std::vector<t_field*>& args = arg_struct->get_members();
|
1349
|
-
vector<t_field*>::const_iterator a_iter;
|
1350
|
-
int num_args = args.size();
|
1351
|
-
|
1352
|
-
f_remote <<
|
1353
|
-
"if cmd == '" << (*f_iter)->get_name() << "':" << endl <<
|
1354
|
-
" if len(args) != " << num_args << ":" << endl <<
|
1355
|
-
" print '" << (*f_iter)->get_name() << " requires " << num_args << " args'" << endl <<
|
1356
|
-
" sys.exit(1)" << endl <<
|
1357
|
-
" pp.pprint(client." << (*f_iter)->get_name() << "(";
|
1358
|
-
for (int i = 0; i < num_args; ++i) {
|
1359
|
-
if (args[i]->get_type()->is_string()) {
|
1360
|
-
f_remote << "args[" << i << "],";
|
1361
|
-
} else {
|
1362
|
-
f_remote << "eval(args[" << i << "]),";
|
1363
|
-
}
|
1364
|
-
}
|
1365
|
-
f_remote << "))" << endl;
|
1366
|
-
|
1367
|
-
f_remote << endl;
|
1368
|
-
}
|
1369
|
-
|
1370
|
-
f_remote << "transport.close()" << endl;
|
1371
|
-
|
1372
|
-
// Close service file
|
1373
|
-
f_remote.close();
|
1374
|
-
|
1375
|
-
// Make file executable, love that bitwise OR action
|
1376
|
-
chmod(f_remote_name.c_str(),
|
1377
|
-
S_IRUSR
|
1378
|
-
| S_IWUSR
|
1379
|
-
| S_IXUSR
|
1380
|
-
#ifndef MINGW
|
1381
|
-
| S_IRGRP
|
1382
|
-
| S_IXGRP
|
1383
|
-
| S_IROTH
|
1384
|
-
| S_IXOTH
|
1385
|
-
#endif
|
1386
|
-
);
|
1387
|
-
}
|
1388
|
-
|
1389
|
-
/**
|
1390
|
-
* Generates a service server definition.
|
1391
|
-
*
|
1392
|
-
* @param tservice The service to generate a server for.
|
1393
|
-
*/
|
1394
|
-
void t_py_generator::generate_service_server(t_service* tservice) {
|
1395
|
-
// Generate the dispatch methods
|
1396
|
-
vector<t_function*> functions = tservice->get_functions();
|
1397
|
-
vector<t_function*>::iterator f_iter;
|
1398
|
-
|
1399
|
-
string extends = "";
|
1400
|
-
string extends_processor = "";
|
1401
|
-
if (tservice->get_extends() != NULL) {
|
1402
|
-
extends = type_name(tservice->get_extends());
|
1403
|
-
extends_processor = extends + ".Processor, ";
|
1404
|
-
}
|
1405
|
-
|
1406
|
-
// Generate the header portion
|
1407
|
-
if (gen_twisted_) {
|
1408
|
-
f_service_ <<
|
1409
|
-
"class Processor(" << extends_processor << "TProcessor):" << endl <<
|
1410
|
-
" implements(Iface)" << endl << endl;
|
1411
|
-
} else {
|
1412
|
-
f_service_ <<
|
1413
|
-
"class Processor(" << extends_processor << "Iface, TProcessor):" << endl;
|
1414
|
-
}
|
1415
|
-
|
1416
|
-
indent_up();
|
1417
|
-
|
1418
|
-
indent(f_service_) <<
|
1419
|
-
"def __init__(self, handler):" << endl;
|
1420
|
-
indent_up();
|
1421
|
-
if (extends.empty()) {
|
1422
|
-
if (gen_twisted_) {
|
1423
|
-
f_service_ <<
|
1424
|
-
indent() << "self._handler = Iface(handler)" << endl;
|
1425
|
-
} else {
|
1426
|
-
f_service_ <<
|
1427
|
-
indent() << "self._handler = handler" << endl;
|
1428
|
-
}
|
1429
|
-
|
1430
|
-
f_service_ <<
|
1431
|
-
indent() << "self._processMap = {}" << endl;
|
1432
|
-
} else {
|
1433
|
-
if (gen_twisted_) {
|
1434
|
-
f_service_ <<
|
1435
|
-
indent() << extends << ".Processor.__init__(self, Iface(handler))" << endl;
|
1436
|
-
} else {
|
1437
|
-
f_service_ <<
|
1438
|
-
indent() << extends << ".Processor.__init__(self, handler)" << endl;
|
1439
|
-
}
|
1440
|
-
}
|
1441
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
1442
|
-
f_service_ <<
|
1443
|
-
indent() << "self._processMap[\"" << (*f_iter)->get_name() << "\"] = Processor.process_" << (*f_iter)->get_name() << endl;
|
1444
|
-
}
|
1445
|
-
indent_down();
|
1446
|
-
f_service_ << endl;
|
1447
|
-
|
1448
|
-
// Generate the server implementation
|
1449
|
-
indent(f_service_) <<
|
1450
|
-
"def process(self, iprot, oprot):" << endl;
|
1451
|
-
indent_up();
|
1452
|
-
|
1453
|
-
f_service_ <<
|
1454
|
-
indent() << "(name, type, seqid) = iprot.readMessageBegin()" << endl;
|
1455
|
-
|
1456
|
-
// TODO(mcslee): validate message
|
1457
|
-
|
1458
|
-
// HOT: dictionary function lookup
|
1459
|
-
f_service_ <<
|
1460
|
-
indent() << "if name not in self._processMap:" << endl <<
|
1461
|
-
indent() << " iprot.skip(TType.STRUCT)" << endl <<
|
1462
|
-
indent() << " iprot.readMessageEnd()" << endl <<
|
1463
|
-
indent() << " x = TApplicationException(TApplicationException.UNKNOWN_METHOD, 'Unknown function %s' % (name))" << endl <<
|
1464
|
-
indent() << " oprot.writeMessageBegin(name, TMessageType.EXCEPTION, seqid)" << endl <<
|
1465
|
-
indent() << " x.write(oprot)" << endl <<
|
1466
|
-
indent() << " oprot.writeMessageEnd()" << endl <<
|
1467
|
-
indent() << " oprot.trans.flush()" << endl;
|
1468
|
-
|
1469
|
-
if (gen_twisted_) {
|
1470
|
-
f_service_ <<
|
1471
|
-
indent() << " return defer.succeed(None)" << endl;
|
1472
|
-
} else {
|
1473
|
-
f_service_ <<
|
1474
|
-
indent() << " return" << endl;
|
1475
|
-
}
|
1476
|
-
|
1477
|
-
f_service_ <<
|
1478
|
-
indent() << "else:" << endl;
|
1479
|
-
|
1480
|
-
if (gen_twisted_) {
|
1481
|
-
f_service_ <<
|
1482
|
-
indent() << " return self._processMap[name](self, seqid, iprot, oprot)" << endl;
|
1483
|
-
} else {
|
1484
|
-
f_service_ <<
|
1485
|
-
indent() << " self._processMap[name](self, seqid, iprot, oprot)" << endl;
|
1486
|
-
|
1487
|
-
// Read end of args field, the T_STOP, and the struct close
|
1488
|
-
f_service_ <<
|
1489
|
-
indent() << "return True" << endl;
|
1490
|
-
}
|
1491
|
-
|
1492
|
-
indent_down();
|
1493
|
-
f_service_ << endl;
|
1494
|
-
|
1495
|
-
// Generate the process subfunctions
|
1496
|
-
for (f_iter = functions.begin(); f_iter != functions.end(); ++f_iter) {
|
1497
|
-
generate_process_function(tservice, *f_iter);
|
1498
|
-
}
|
1499
|
-
|
1500
|
-
indent_down();
|
1501
|
-
f_service_ << endl;
|
1502
|
-
}
|
1503
|
-
|
1504
|
-
/**
|
1505
|
-
* Generates a process function definition.
|
1506
|
-
*
|
1507
|
-
* @param tfunction The function to write a dispatcher for
|
1508
|
-
*/
|
1509
|
-
void t_py_generator::generate_process_function(t_service* tservice,
|
1510
|
-
t_function* tfunction) {
|
1511
|
-
// Open function
|
1512
|
-
indent(f_service_) <<
|
1513
|
-
"def process_" << tfunction->get_name() <<
|
1514
|
-
"(self, seqid, iprot, oprot):" << endl;
|
1515
|
-
indent_up();
|
1516
|
-
|
1517
|
-
string argsname = tfunction->get_name() + "_args";
|
1518
|
-
string resultname = tfunction->get_name() + "_result";
|
1519
|
-
|
1520
|
-
f_service_ <<
|
1521
|
-
indent() << "args = " << argsname << "()" << endl <<
|
1522
|
-
indent() << "args.read(iprot)" << endl <<
|
1523
|
-
indent() << "iprot.readMessageEnd()" << endl;
|
1524
|
-
|
1525
|
-
t_struct* xs = tfunction->get_xceptions();
|
1526
|
-
const std::vector<t_field*>& xceptions = xs->get_members();
|
1527
|
-
vector<t_field*>::const_iterator x_iter;
|
1528
|
-
|
1529
|
-
// Declare result for non oneway function
|
1530
|
-
if (!tfunction->is_oneway()) {
|
1531
|
-
f_service_ <<
|
1532
|
-
indent() << "result = " << resultname << "()" << endl;
|
1533
|
-
}
|
1534
|
-
|
1535
|
-
if (gen_twisted_) {
|
1536
|
-
// Generate the function call
|
1537
|
-
t_struct* arg_struct = tfunction->get_arglist();
|
1538
|
-
const std::vector<t_field*>& fields = arg_struct->get_members();
|
1539
|
-
vector<t_field*>::const_iterator f_iter;
|
1540
|
-
|
1541
|
-
f_service_ <<
|
1542
|
-
indent() << "d = defer.maybeDeferred(self._handler." <<
|
1543
|
-
tfunction->get_name() << ", ";
|
1544
|
-
bool first = true;
|
1545
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
1546
|
-
if (first) {
|
1547
|
-
first = false;
|
1548
|
-
} else {
|
1549
|
-
f_service_ << ", ";
|
1550
|
-
}
|
1551
|
-
f_service_ << "args." << (*f_iter)->get_name();
|
1552
|
-
}
|
1553
|
-
f_service_ << ")" << endl;
|
1554
|
-
|
1555
|
-
// Shortcut out here for oneway functions
|
1556
|
-
if (tfunction->is_oneway()) {
|
1557
|
-
f_service_ <<
|
1558
|
-
indent() << "return d" << endl;
|
1559
|
-
indent_down();
|
1560
|
-
f_service_ << endl;
|
1561
|
-
return;
|
1562
|
-
}
|
1563
|
-
|
1564
|
-
f_service_ <<
|
1565
|
-
indent() <<
|
1566
|
-
"d.addCallback(self.write_results_success_" <<
|
1567
|
-
tfunction->get_name() << ", result, seqid, oprot)" << endl;
|
1568
|
-
|
1569
|
-
if (xceptions.size() > 0) {
|
1570
|
-
f_service_ <<
|
1571
|
-
indent() <<
|
1572
|
-
"d.addErrback(self.write_results_exception_" <<
|
1573
|
-
tfunction->get_name() << ", result, seqid, oprot)" << endl;
|
1574
|
-
}
|
1575
|
-
|
1576
|
-
f_service_ <<
|
1577
|
-
indent() << "return d" << endl;
|
1578
|
-
|
1579
|
-
indent_down();
|
1580
|
-
f_service_ << endl;
|
1581
|
-
|
1582
|
-
indent(f_service_) <<
|
1583
|
-
"def write_results_success_" << tfunction->get_name() <<
|
1584
|
-
"(self, success, result, seqid, oprot):" << endl;
|
1585
|
-
indent_up();
|
1586
|
-
f_service_ <<
|
1587
|
-
indent() << "result.success = success" << endl <<
|
1588
|
-
indent() << "oprot.writeMessageBegin(\"" << tfunction->get_name() <<
|
1589
|
-
"\", TMessageType.REPLY, seqid)" << endl <<
|
1590
|
-
indent() << "result.write(oprot)" << endl <<
|
1591
|
-
indent() << "oprot.writeMessageEnd()" << endl <<
|
1592
|
-
indent() << "oprot.trans.flush()" << endl;
|
1593
|
-
indent_down();
|
1594
|
-
f_service_ << endl;
|
1595
|
-
|
1596
|
-
// Try block for a function with exceptions
|
1597
|
-
if (!tfunction->is_oneway() && xceptions.size() > 0) {
|
1598
|
-
indent(f_service_) <<
|
1599
|
-
"def write_results_exception_" << tfunction->get_name() <<
|
1600
|
-
"(self, error, result, seqid, oprot):" << endl;
|
1601
|
-
indent_up();
|
1602
|
-
f_service_ <<
|
1603
|
-
indent() << "try:" << endl;
|
1604
|
-
|
1605
|
-
// Kinda absurd
|
1606
|
-
f_service_ <<
|
1607
|
-
indent() << " error.raiseException()" << endl;
|
1608
|
-
for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
|
1609
|
-
f_service_ <<
|
1610
|
-
indent() << "except " << type_name((*x_iter)->get_type()) << ", " << (*x_iter)->get_name() << ":" << endl;
|
1611
|
-
if (!tfunction->is_oneway()) {
|
1612
|
-
indent_up();
|
1613
|
-
f_service_ <<
|
1614
|
-
indent() << "result." << (*x_iter)->get_name() << " = " << (*x_iter)->get_name() << endl;
|
1615
|
-
indent_down();
|
1616
|
-
} else {
|
1617
|
-
f_service_ <<
|
1618
|
-
indent() << "pass" << endl;
|
1619
|
-
}
|
1620
|
-
}
|
1621
|
-
f_service_ <<
|
1622
|
-
indent() << "oprot.writeMessageBegin(\"" << tfunction->get_name() <<
|
1623
|
-
"\", TMessageType.REPLY, seqid)" << endl <<
|
1624
|
-
indent() << "result.write(oprot)" << endl <<
|
1625
|
-
indent() << "oprot.writeMessageEnd()" << endl <<
|
1626
|
-
indent() << "oprot.trans.flush()" << endl;
|
1627
|
-
indent_down();
|
1628
|
-
f_service_ << endl;
|
1629
|
-
}
|
1630
|
-
} else {
|
1631
|
-
|
1632
|
-
// Try block for a function with exceptions
|
1633
|
-
if (xceptions.size() > 0) {
|
1634
|
-
f_service_ <<
|
1635
|
-
indent() << "try:" << endl;
|
1636
|
-
indent_up();
|
1637
|
-
}
|
1638
|
-
|
1639
|
-
// Generate the function call
|
1640
|
-
t_struct* arg_struct = tfunction->get_arglist();
|
1641
|
-
const std::vector<t_field*>& fields = arg_struct->get_members();
|
1642
|
-
vector<t_field*>::const_iterator f_iter;
|
1643
|
-
|
1644
|
-
f_service_ << indent();
|
1645
|
-
if (!tfunction->is_oneway() && !tfunction->get_returntype()->is_void()) {
|
1646
|
-
f_service_ << "result.success = ";
|
1647
|
-
}
|
1648
|
-
f_service_ <<
|
1649
|
-
"self._handler." << tfunction->get_name() << "(";
|
1650
|
-
bool first = true;
|
1651
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
1652
|
-
if (first) {
|
1653
|
-
first = false;
|
1654
|
-
} else {
|
1655
|
-
f_service_ << ", ";
|
1656
|
-
}
|
1657
|
-
f_service_ << "args." << (*f_iter)->get_name();
|
1658
|
-
}
|
1659
|
-
f_service_ << ")" << endl;
|
1660
|
-
|
1661
|
-
if (!tfunction->is_oneway() && xceptions.size() > 0) {
|
1662
|
-
indent_down();
|
1663
|
-
for (x_iter = xceptions.begin(); x_iter != xceptions.end(); ++x_iter) {
|
1664
|
-
f_service_ <<
|
1665
|
-
indent() << "except " << type_name((*x_iter)->get_type()) << ", " << (*x_iter)->get_name() << ":" << endl;
|
1666
|
-
if (!tfunction->is_oneway()) {
|
1667
|
-
indent_up();
|
1668
|
-
f_service_ <<
|
1669
|
-
indent() << "result." << (*x_iter)->get_name() << " = " << (*x_iter)->get_name() << endl;
|
1670
|
-
indent_down();
|
1671
|
-
} else {
|
1672
|
-
f_service_ <<
|
1673
|
-
indent() << "pass" << endl;
|
1674
|
-
}
|
1675
|
-
}
|
1676
|
-
}
|
1677
|
-
|
1678
|
-
// Shortcut out here for oneway functions
|
1679
|
-
if (tfunction->is_oneway()) {
|
1680
|
-
f_service_ <<
|
1681
|
-
indent() << "return" << endl;
|
1682
|
-
indent_down();
|
1683
|
-
f_service_ << endl;
|
1684
|
-
return;
|
1685
|
-
}
|
1686
|
-
|
1687
|
-
f_service_ <<
|
1688
|
-
indent() << "oprot.writeMessageBegin(\"" << tfunction->get_name() << "\", TMessageType.REPLY, seqid)" << endl <<
|
1689
|
-
indent() << "result.write(oprot)" << endl <<
|
1690
|
-
indent() << "oprot.writeMessageEnd()" << endl <<
|
1691
|
-
indent() << "oprot.trans.flush()" << endl;
|
1692
|
-
|
1693
|
-
// Close function
|
1694
|
-
indent_down();
|
1695
|
-
f_service_ << endl;
|
1696
|
-
}
|
1697
|
-
}
|
1698
|
-
|
1699
|
-
/**
|
1700
|
-
* Deserializes a field of any type.
|
1701
|
-
*/
|
1702
|
-
void t_py_generator::generate_deserialize_field(ofstream &out,
|
1703
|
-
t_field* tfield,
|
1704
|
-
string prefix,
|
1705
|
-
bool inclass) {
|
1706
|
-
t_type* type = get_true_type(tfield->get_type());
|
1707
|
-
|
1708
|
-
if (type->is_void()) {
|
1709
|
-
throw "CANNOT GENERATE DESERIALIZE CODE FOR void TYPE: " +
|
1710
|
-
prefix + tfield->get_name();
|
1711
|
-
}
|
1712
|
-
|
1713
|
-
string name = prefix + tfield->get_name();
|
1714
|
-
|
1715
|
-
if (type->is_struct() || type->is_xception()) {
|
1716
|
-
generate_deserialize_struct(out,
|
1717
|
-
(t_struct*)type,
|
1718
|
-
name);
|
1719
|
-
} else if (type->is_container()) {
|
1720
|
-
generate_deserialize_container(out, type, name);
|
1721
|
-
} else if (type->is_base_type() || type->is_enum()) {
|
1722
|
-
indent(out) <<
|
1723
|
-
name << " = iprot.";
|
1724
|
-
|
1725
|
-
if (type->is_base_type()) {
|
1726
|
-
t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
|
1727
|
-
switch (tbase) {
|
1728
|
-
case t_base_type::TYPE_VOID:
|
1729
|
-
throw "compiler error: cannot serialize void field in a struct: " +
|
1730
|
-
name;
|
1731
|
-
break;
|
1732
|
-
case t_base_type::TYPE_STRING:
|
1733
|
-
out << "readString();";
|
1734
|
-
break;
|
1735
|
-
case t_base_type::TYPE_BOOL:
|
1736
|
-
out << "readBool();";
|
1737
|
-
break;
|
1738
|
-
case t_base_type::TYPE_BYTE:
|
1739
|
-
out << "readByte();";
|
1740
|
-
break;
|
1741
|
-
case t_base_type::TYPE_I16:
|
1742
|
-
out << "readI16();";
|
1743
|
-
break;
|
1744
|
-
case t_base_type::TYPE_I32:
|
1745
|
-
out << "readI32();";
|
1746
|
-
break;
|
1747
|
-
case t_base_type::TYPE_I64:
|
1748
|
-
out << "readI64();";
|
1749
|
-
break;
|
1750
|
-
case t_base_type::TYPE_DOUBLE:
|
1751
|
-
out << "readDouble();";
|
1752
|
-
break;
|
1753
|
-
default:
|
1754
|
-
throw "compiler error: no PHP name for base type " + t_base_type::t_base_name(tbase);
|
1755
|
-
}
|
1756
|
-
} else if (type->is_enum()) {
|
1757
|
-
out << "readI32();";
|
1758
|
-
}
|
1759
|
-
out << endl;
|
1760
|
-
|
1761
|
-
} else {
|
1762
|
-
printf("DO NOT KNOW HOW TO DESERIALIZE FIELD '%s' TYPE '%s'\n",
|
1763
|
-
tfield->get_name().c_str(), type->get_name().c_str());
|
1764
|
-
}
|
1765
|
-
}
|
1766
|
-
|
1767
|
-
/**
|
1768
|
-
* Generates an unserializer for a struct, calling read()
|
1769
|
-
*/
|
1770
|
-
void t_py_generator::generate_deserialize_struct(ofstream &out,
|
1771
|
-
t_struct* tstruct,
|
1772
|
-
string prefix) {
|
1773
|
-
out <<
|
1774
|
-
indent() << prefix << " = " << type_name(tstruct) << "()" << endl <<
|
1775
|
-
indent() << prefix << ".read(iprot)" << endl;
|
1776
|
-
}
|
1777
|
-
|
1778
|
-
/**
|
1779
|
-
* Serialize a container by writing out the header followed by
|
1780
|
-
* data and then a footer.
|
1781
|
-
*/
|
1782
|
-
void t_py_generator::generate_deserialize_container(ofstream &out,
|
1783
|
-
t_type* ttype,
|
1784
|
-
string prefix) {
|
1785
|
-
string size = tmp("_size");
|
1786
|
-
string ktype = tmp("_ktype");
|
1787
|
-
string vtype = tmp("_vtype");
|
1788
|
-
string etype = tmp("_etype");
|
1789
|
-
|
1790
|
-
t_field fsize(g_type_i32, size);
|
1791
|
-
t_field fktype(g_type_byte, ktype);
|
1792
|
-
t_field fvtype(g_type_byte, vtype);
|
1793
|
-
t_field fetype(g_type_byte, etype);
|
1794
|
-
|
1795
|
-
// Declare variables, read header
|
1796
|
-
if (ttype->is_map()) {
|
1797
|
-
out <<
|
1798
|
-
indent() << prefix << " = {}" << endl <<
|
1799
|
-
indent() << "(" << ktype << ", " << vtype << ", " << size << " ) = iprot.readMapBegin() " << endl;
|
1800
|
-
} else if (ttype->is_set()) {
|
1801
|
-
out <<
|
1802
|
-
indent() << prefix << " = set()" << endl <<
|
1803
|
-
indent() << "(" << etype << ", " << size << ") = iprot.readSetBegin()" << endl;
|
1804
|
-
} else if (ttype->is_list()) {
|
1805
|
-
out <<
|
1806
|
-
indent() << prefix << " = []" << endl <<
|
1807
|
-
indent() << "(" << etype << ", " << size << ") = iprot.readListBegin()" << endl;
|
1808
|
-
}
|
1809
|
-
|
1810
|
-
// For loop iterates over elements
|
1811
|
-
string i = tmp("_i");
|
1812
|
-
indent(out) <<
|
1813
|
-
"for " << i << " in xrange(" << size << "):" << endl;
|
1814
|
-
|
1815
|
-
indent_up();
|
1816
|
-
|
1817
|
-
if (ttype->is_map()) {
|
1818
|
-
generate_deserialize_map_element(out, (t_map*)ttype, prefix);
|
1819
|
-
} else if (ttype->is_set()) {
|
1820
|
-
generate_deserialize_set_element(out, (t_set*)ttype, prefix);
|
1821
|
-
} else if (ttype->is_list()) {
|
1822
|
-
generate_deserialize_list_element(out, (t_list*)ttype, prefix);
|
1823
|
-
}
|
1824
|
-
|
1825
|
-
indent_down();
|
1826
|
-
|
1827
|
-
// Read container end
|
1828
|
-
if (ttype->is_map()) {
|
1829
|
-
indent(out) << "iprot.readMapEnd()" << endl;
|
1830
|
-
} else if (ttype->is_set()) {
|
1831
|
-
indent(out) << "iprot.readSetEnd()" << endl;
|
1832
|
-
} else if (ttype->is_list()) {
|
1833
|
-
indent(out) << "iprot.readListEnd()" << endl;
|
1834
|
-
}
|
1835
|
-
}
|
1836
|
-
|
1837
|
-
|
1838
|
-
/**
|
1839
|
-
* Generates code to deserialize a map
|
1840
|
-
*/
|
1841
|
-
void t_py_generator::generate_deserialize_map_element(ofstream &out,
|
1842
|
-
t_map* tmap,
|
1843
|
-
string prefix) {
|
1844
|
-
string key = tmp("_key");
|
1845
|
-
string val = tmp("_val");
|
1846
|
-
t_field fkey(tmap->get_key_type(), key);
|
1847
|
-
t_field fval(tmap->get_val_type(), val);
|
1848
|
-
|
1849
|
-
generate_deserialize_field(out, &fkey);
|
1850
|
-
generate_deserialize_field(out, &fval);
|
1851
|
-
|
1852
|
-
indent(out) <<
|
1853
|
-
prefix << "[" << key << "] = " << val << endl;
|
1854
|
-
}
|
1855
|
-
|
1856
|
-
/**
|
1857
|
-
* Write a set element
|
1858
|
-
*/
|
1859
|
-
void t_py_generator::generate_deserialize_set_element(ofstream &out,
|
1860
|
-
t_set* tset,
|
1861
|
-
string prefix) {
|
1862
|
-
string elem = tmp("_elem");
|
1863
|
-
t_field felem(tset->get_elem_type(), elem);
|
1864
|
-
|
1865
|
-
generate_deserialize_field(out, &felem);
|
1866
|
-
|
1867
|
-
indent(out) <<
|
1868
|
-
prefix << ".add(" << elem << ")" << endl;
|
1869
|
-
}
|
1870
|
-
|
1871
|
-
/**
|
1872
|
-
* Write a list element
|
1873
|
-
*/
|
1874
|
-
void t_py_generator::generate_deserialize_list_element(ofstream &out,
|
1875
|
-
t_list* tlist,
|
1876
|
-
string prefix) {
|
1877
|
-
string elem = tmp("_elem");
|
1878
|
-
t_field felem(tlist->get_elem_type(), elem);
|
1879
|
-
|
1880
|
-
generate_deserialize_field(out, &felem);
|
1881
|
-
|
1882
|
-
indent(out) <<
|
1883
|
-
prefix << ".append(" << elem << ")" << endl;
|
1884
|
-
}
|
1885
|
-
|
1886
|
-
|
1887
|
-
/**
|
1888
|
-
* Serializes a field of any type.
|
1889
|
-
*
|
1890
|
-
* @param tfield The field to serialize
|
1891
|
-
* @param prefix Name to prepend to field name
|
1892
|
-
*/
|
1893
|
-
void t_py_generator::generate_serialize_field(ofstream &out,
|
1894
|
-
t_field* tfield,
|
1895
|
-
string prefix) {
|
1896
|
-
t_type* type = get_true_type(tfield->get_type());
|
1897
|
-
|
1898
|
-
// Do nothing for void types
|
1899
|
-
if (type->is_void()) {
|
1900
|
-
throw "CANNOT GENERATE SERIALIZE CODE FOR void TYPE: " +
|
1901
|
-
prefix + tfield->get_name();
|
1902
|
-
}
|
1903
|
-
|
1904
|
-
if (type->is_struct() || type->is_xception()) {
|
1905
|
-
generate_serialize_struct(out,
|
1906
|
-
(t_struct*)type,
|
1907
|
-
prefix + tfield->get_name());
|
1908
|
-
} else if (type->is_container()) {
|
1909
|
-
generate_serialize_container(out,
|
1910
|
-
type,
|
1911
|
-
prefix + tfield->get_name());
|
1912
|
-
} else if (type->is_base_type() || type->is_enum()) {
|
1913
|
-
|
1914
|
-
string name = prefix + tfield->get_name();
|
1915
|
-
|
1916
|
-
indent(out) <<
|
1917
|
-
"oprot.";
|
1918
|
-
|
1919
|
-
if (type->is_base_type()) {
|
1920
|
-
t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
|
1921
|
-
switch (tbase) {
|
1922
|
-
case t_base_type::TYPE_VOID:
|
1923
|
-
throw
|
1924
|
-
"compiler error: cannot serialize void field in a struct: " + name;
|
1925
|
-
break;
|
1926
|
-
case t_base_type::TYPE_STRING:
|
1927
|
-
out << "writeString(" << name << ")";
|
1928
|
-
break;
|
1929
|
-
case t_base_type::TYPE_BOOL:
|
1930
|
-
out << "writeBool(" << name << ")";
|
1931
|
-
break;
|
1932
|
-
case t_base_type::TYPE_BYTE:
|
1933
|
-
out << "writeByte(" << name << ")";
|
1934
|
-
break;
|
1935
|
-
case t_base_type::TYPE_I16:
|
1936
|
-
out << "writeI16(" << name << ")";
|
1937
|
-
break;
|
1938
|
-
case t_base_type::TYPE_I32:
|
1939
|
-
out << "writeI32(" << name << ")";
|
1940
|
-
break;
|
1941
|
-
case t_base_type::TYPE_I64:
|
1942
|
-
out << "writeI64(" << name << ")";
|
1943
|
-
break;
|
1944
|
-
case t_base_type::TYPE_DOUBLE:
|
1945
|
-
out << "writeDouble(" << name << ")";
|
1946
|
-
break;
|
1947
|
-
default:
|
1948
|
-
throw "compiler error: no PHP name for base type " + t_base_type::t_base_name(tbase);
|
1949
|
-
}
|
1950
|
-
} else if (type->is_enum()) {
|
1951
|
-
out << "writeI32(" << name << ")";
|
1952
|
-
}
|
1953
|
-
out << endl;
|
1954
|
-
} else {
|
1955
|
-
printf("DO NOT KNOW HOW TO SERIALIZE FIELD '%s%s' TYPE '%s'\n",
|
1956
|
-
prefix.c_str(),
|
1957
|
-
tfield->get_name().c_str(),
|
1958
|
-
type->get_name().c_str());
|
1959
|
-
}
|
1960
|
-
}
|
1961
|
-
|
1962
|
-
/**
|
1963
|
-
* Serializes all the members of a struct.
|
1964
|
-
*
|
1965
|
-
* @param tstruct The struct to serialize
|
1966
|
-
* @param prefix String prefix to attach to all fields
|
1967
|
-
*/
|
1968
|
-
void t_py_generator::generate_serialize_struct(ofstream &out,
|
1969
|
-
t_struct* tstruct,
|
1970
|
-
string prefix) {
|
1971
|
-
indent(out) <<
|
1972
|
-
prefix << ".write(oprot)" << endl;
|
1973
|
-
}
|
1974
|
-
|
1975
|
-
void t_py_generator::generate_serialize_container(ofstream &out,
|
1976
|
-
t_type* ttype,
|
1977
|
-
string prefix) {
|
1978
|
-
if (ttype->is_map()) {
|
1979
|
-
indent(out) <<
|
1980
|
-
"oprot.writeMapBegin(" <<
|
1981
|
-
type_to_enum(((t_map*)ttype)->get_key_type()) << ", " <<
|
1982
|
-
type_to_enum(((t_map*)ttype)->get_val_type()) << ", " <<
|
1983
|
-
"len(" << prefix << "))" << endl;
|
1984
|
-
} else if (ttype->is_set()) {
|
1985
|
-
indent(out) <<
|
1986
|
-
"oprot.writeSetBegin(" <<
|
1987
|
-
type_to_enum(((t_set*)ttype)->get_elem_type()) << ", " <<
|
1988
|
-
"len(" << prefix << "))" << endl;
|
1989
|
-
} else if (ttype->is_list()) {
|
1990
|
-
indent(out) <<
|
1991
|
-
"oprot.writeListBegin(" <<
|
1992
|
-
type_to_enum(((t_list*)ttype)->get_elem_type()) << ", " <<
|
1993
|
-
"len(" << prefix << "))" << endl;
|
1994
|
-
}
|
1995
|
-
|
1996
|
-
if (ttype->is_map()) {
|
1997
|
-
string kiter = tmp("kiter");
|
1998
|
-
string viter = tmp("viter");
|
1999
|
-
indent(out) <<
|
2000
|
-
"for " << kiter << "," << viter << " in " << prefix << ".items():" << endl;
|
2001
|
-
indent_up();
|
2002
|
-
generate_serialize_map_element(out, (t_map*)ttype, kiter, viter);
|
2003
|
-
indent_down();
|
2004
|
-
} else if (ttype->is_set()) {
|
2005
|
-
string iter = tmp("iter");
|
2006
|
-
indent(out) <<
|
2007
|
-
"for " << iter << " in " << prefix << ":" << endl;
|
2008
|
-
indent_up();
|
2009
|
-
generate_serialize_set_element(out, (t_set*)ttype, iter);
|
2010
|
-
indent_down();
|
2011
|
-
} else if (ttype->is_list()) {
|
2012
|
-
string iter = tmp("iter");
|
2013
|
-
indent(out) <<
|
2014
|
-
"for " << iter << " in " << prefix << ":" << endl;
|
2015
|
-
indent_up();
|
2016
|
-
generate_serialize_list_element(out, (t_list*)ttype, iter);
|
2017
|
-
indent_down();
|
2018
|
-
}
|
2019
|
-
|
2020
|
-
if (ttype->is_map()) {
|
2021
|
-
indent(out) <<
|
2022
|
-
"oprot.writeMapEnd()" << endl;
|
2023
|
-
} else if (ttype->is_set()) {
|
2024
|
-
indent(out) <<
|
2025
|
-
"oprot.writeSetEnd()" << endl;
|
2026
|
-
} else if (ttype->is_list()) {
|
2027
|
-
indent(out) <<
|
2028
|
-
"oprot.writeListEnd()" << endl;
|
2029
|
-
}
|
2030
|
-
}
|
2031
|
-
|
2032
|
-
/**
|
2033
|
-
* Serializes the members of a map.
|
2034
|
-
*
|
2035
|
-
*/
|
2036
|
-
void t_py_generator::generate_serialize_map_element(ofstream &out,
|
2037
|
-
t_map* tmap,
|
2038
|
-
string kiter,
|
2039
|
-
string viter) {
|
2040
|
-
t_field kfield(tmap->get_key_type(), kiter);
|
2041
|
-
generate_serialize_field(out, &kfield, "");
|
2042
|
-
|
2043
|
-
t_field vfield(tmap->get_val_type(), viter);
|
2044
|
-
generate_serialize_field(out, &vfield, "");
|
2045
|
-
}
|
2046
|
-
|
2047
|
-
/**
|
2048
|
-
* Serializes the members of a set.
|
2049
|
-
*/
|
2050
|
-
void t_py_generator::generate_serialize_set_element(ofstream &out,
|
2051
|
-
t_set* tset,
|
2052
|
-
string iter) {
|
2053
|
-
t_field efield(tset->get_elem_type(), iter);
|
2054
|
-
generate_serialize_field(out, &efield, "");
|
2055
|
-
}
|
2056
|
-
|
2057
|
-
/**
|
2058
|
-
* Serializes the members of a list.
|
2059
|
-
*/
|
2060
|
-
void t_py_generator::generate_serialize_list_element(ofstream &out,
|
2061
|
-
t_list* tlist,
|
2062
|
-
string iter) {
|
2063
|
-
t_field efield(tlist->get_elem_type(), iter);
|
2064
|
-
generate_serialize_field(out, &efield, "");
|
2065
|
-
}
|
2066
|
-
|
2067
|
-
/**
|
2068
|
-
* Generates the docstring for a given struct.
|
2069
|
-
*/
|
2070
|
-
void t_py_generator::generate_python_docstring(ofstream& out,
|
2071
|
-
t_struct* tstruct) {
|
2072
|
-
generate_python_docstring(out, tstruct, tstruct, "Attributes");
|
2073
|
-
}
|
2074
|
-
|
2075
|
-
/**
|
2076
|
-
* Generates the docstring for a given function.
|
2077
|
-
*/
|
2078
|
-
void t_py_generator::generate_python_docstring(ofstream& out,
|
2079
|
-
t_function* tfunction) {
|
2080
|
-
generate_python_docstring(out, tfunction, tfunction->get_arglist(), "Parameters");
|
2081
|
-
}
|
2082
|
-
|
2083
|
-
/**
|
2084
|
-
* Generates the docstring for a struct or function.
|
2085
|
-
*/
|
2086
|
-
void t_py_generator::generate_python_docstring(ofstream& out,
|
2087
|
-
t_doc* tdoc,
|
2088
|
-
t_struct* tstruct,
|
2089
|
-
const char* subheader) {
|
2090
|
-
bool has_doc = false;
|
2091
|
-
stringstream ss;
|
2092
|
-
if (tdoc->has_doc()) {
|
2093
|
-
has_doc = true;
|
2094
|
-
ss << tdoc->get_doc();
|
2095
|
-
}
|
2096
|
-
|
2097
|
-
const vector<t_field*>& fields = tstruct->get_members();
|
2098
|
-
if (fields.size() > 0) {
|
2099
|
-
if (has_doc) {
|
2100
|
-
ss << endl;
|
2101
|
-
}
|
2102
|
-
has_doc = true;
|
2103
|
-
ss << subheader << ":\n";
|
2104
|
-
vector<t_field*>::const_iterator p_iter;
|
2105
|
-
for (p_iter = fields.begin(); p_iter != fields.end(); ++p_iter) {
|
2106
|
-
t_field* p = *p_iter;
|
2107
|
-
ss << " - " << p->get_name();
|
2108
|
-
if (p->has_doc()) {
|
2109
|
-
ss << ": " << p->get_doc();
|
2110
|
-
} else {
|
2111
|
-
ss << endl;
|
2112
|
-
}
|
2113
|
-
}
|
2114
|
-
}
|
2115
|
-
|
2116
|
-
if (has_doc) {
|
2117
|
-
generate_docstring_comment(out,
|
2118
|
-
"\"\"\"\n",
|
2119
|
-
"", ss.str(),
|
2120
|
-
"\"\"\"\n");
|
2121
|
-
}
|
2122
|
-
}
|
2123
|
-
|
2124
|
-
/**
|
2125
|
-
* Generates the docstring for a generic object.
|
2126
|
-
*/
|
2127
|
-
void t_py_generator::generate_python_docstring(ofstream& out,
|
2128
|
-
t_doc* tdoc) {
|
2129
|
-
if (tdoc->has_doc()) {
|
2130
|
-
generate_docstring_comment(out,
|
2131
|
-
"\"\"\"\n",
|
2132
|
-
"", tdoc->get_doc(),
|
2133
|
-
"\"\"\"\n");
|
2134
|
-
}
|
2135
|
-
}
|
2136
|
-
|
2137
|
-
/**
|
2138
|
-
* Declares an argument, which may include initialization as necessary.
|
2139
|
-
*
|
2140
|
-
* @param tfield The field
|
2141
|
-
*/
|
2142
|
-
string t_py_generator::declare_argument(t_field* tfield) {
|
2143
|
-
std::ostringstream result;
|
2144
|
-
result << tfield->get_name() << "=";
|
2145
|
-
if (tfield->get_value() != NULL) {
|
2146
|
-
result << "thrift_spec[" <<
|
2147
|
-
tfield->get_key() << "][4]";
|
2148
|
-
} else {
|
2149
|
-
result << "None";
|
2150
|
-
}
|
2151
|
-
return result.str();
|
2152
|
-
}
|
2153
|
-
|
2154
|
-
/**
|
2155
|
-
* Renders a field default value, returns None otherwise.
|
2156
|
-
*
|
2157
|
-
* @param tfield The field
|
2158
|
-
*/
|
2159
|
-
string t_py_generator::render_field_default_value(t_field* tfield) {
|
2160
|
-
t_type* type = get_true_type(tfield->get_type());
|
2161
|
-
if (tfield->get_value() != NULL) {
|
2162
|
-
return render_const_value(type, tfield->get_value());
|
2163
|
-
} else {
|
2164
|
-
return "None";
|
2165
|
-
}
|
2166
|
-
}
|
2167
|
-
|
2168
|
-
/**
|
2169
|
-
* Renders a function signature of the form 'type name(args)'
|
2170
|
-
*
|
2171
|
-
* @param tfunction Function definition
|
2172
|
-
* @return String of rendered function definition
|
2173
|
-
*/
|
2174
|
-
string t_py_generator::function_signature(t_function* tfunction,
|
2175
|
-
string prefix) {
|
2176
|
-
// TODO(mcslee): Nitpicky, no ',' if argument_list is empty
|
2177
|
-
return
|
2178
|
-
prefix + tfunction->get_name() +
|
2179
|
-
"(self, " + argument_list(tfunction->get_arglist()) + ")";
|
2180
|
-
}
|
2181
|
-
|
2182
|
-
/**
|
2183
|
-
* Renders an interface function signature of the form 'type name(args)'
|
2184
|
-
*
|
2185
|
-
* @param tfunction Function definition
|
2186
|
-
* @return String of rendered function definition
|
2187
|
-
*/
|
2188
|
-
string t_py_generator::function_signature_if(t_function* tfunction,
|
2189
|
-
string prefix) {
|
2190
|
-
// TODO(mcslee): Nitpicky, no ',' if argument_list is empty
|
2191
|
-
string signature = prefix + tfunction->get_name() + "(";
|
2192
|
-
if (!gen_twisted_) {
|
2193
|
-
signature += "self, ";
|
2194
|
-
}
|
2195
|
-
signature += argument_list(tfunction->get_arglist()) + ")";
|
2196
|
-
return signature;
|
2197
|
-
}
|
2198
|
-
|
2199
|
-
|
2200
|
-
/**
|
2201
|
-
* Renders a field list
|
2202
|
-
*/
|
2203
|
-
string t_py_generator::argument_list(t_struct* tstruct) {
|
2204
|
-
string result = "";
|
2205
|
-
|
2206
|
-
const vector<t_field*>& fields = tstruct->get_members();
|
2207
|
-
vector<t_field*>::const_iterator f_iter;
|
2208
|
-
bool first = true;
|
2209
|
-
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
|
2210
|
-
if (first) {
|
2211
|
-
first = false;
|
2212
|
-
} else {
|
2213
|
-
result += ", ";
|
2214
|
-
}
|
2215
|
-
result += (*f_iter)->get_name();
|
2216
|
-
}
|
2217
|
-
return result;
|
2218
|
-
}
|
2219
|
-
|
2220
|
-
string t_py_generator::type_name(t_type* ttype) {
|
2221
|
-
t_program* program = ttype->get_program();
|
2222
|
-
if (ttype->is_service()) {
|
2223
|
-
return get_real_py_module(program) + "." + ttype->get_name();
|
2224
|
-
}
|
2225
|
-
if (program != NULL && program != program_) {
|
2226
|
-
return get_real_py_module(program) + ".ttypes." + ttype->get_name();
|
2227
|
-
}
|
2228
|
-
return ttype->get_name();
|
2229
|
-
}
|
2230
|
-
|
2231
|
-
/**
|
2232
|
-
* Converts the parse type to a Python tyoe
|
2233
|
-
*/
|
2234
|
-
string t_py_generator::type_to_enum(t_type* type) {
|
2235
|
-
type = get_true_type(type);
|
2236
|
-
|
2237
|
-
if (type->is_base_type()) {
|
2238
|
-
t_base_type::t_base tbase = ((t_base_type*)type)->get_base();
|
2239
|
-
switch (tbase) {
|
2240
|
-
case t_base_type::TYPE_VOID:
|
2241
|
-
throw "NO T_VOID CONSTRUCT";
|
2242
|
-
case t_base_type::TYPE_STRING:
|
2243
|
-
return "TType.STRING";
|
2244
|
-
case t_base_type::TYPE_BOOL:
|
2245
|
-
return "TType.BOOL";
|
2246
|
-
case t_base_type::TYPE_BYTE:
|
2247
|
-
return "TType.BYTE";
|
2248
|
-
case t_base_type::TYPE_I16:
|
2249
|
-
return "TType.I16";
|
2250
|
-
case t_base_type::TYPE_I32:
|
2251
|
-
return "TType.I32";
|
2252
|
-
case t_base_type::TYPE_I64:
|
2253
|
-
return "TType.I64";
|
2254
|
-
case t_base_type::TYPE_DOUBLE:
|
2255
|
-
return "TType.DOUBLE";
|
2256
|
-
}
|
2257
|
-
} else if (type->is_enum()) {
|
2258
|
-
return "TType.I32";
|
2259
|
-
} else if (type->is_struct() || type->is_xception()) {
|
2260
|
-
return "TType.STRUCT";
|
2261
|
-
} else if (type->is_map()) {
|
2262
|
-
return "TType.MAP";
|
2263
|
-
} else if (type->is_set()) {
|
2264
|
-
return "TType.SET";
|
2265
|
-
} else if (type->is_list()) {
|
2266
|
-
return "TType.LIST";
|
2267
|
-
}
|
2268
|
-
|
2269
|
-
throw "INVALID TYPE IN type_to_enum: " + type->get_name();
|
2270
|
-
}
|
2271
|
-
|
2272
|
-
/** See the comment inside generate_py_struct_definition for what this is. */
|
2273
|
-
string t_py_generator::type_to_spec_args(t_type* ttype) {
|
2274
|
-
while (ttype->is_typedef()) {
|
2275
|
-
ttype = ((t_typedef*)ttype)->get_type();
|
2276
|
-
}
|
2277
|
-
|
2278
|
-
if (ttype->is_base_type() || ttype->is_enum()) {
|
2279
|
-
return "None";
|
2280
|
-
} else if (ttype->is_struct() || ttype->is_xception()) {
|
2281
|
-
return "(" + type_name(ttype) + ", " + type_name(ttype) + ".thrift_spec)";
|
2282
|
-
} else if (ttype->is_map()) {
|
2283
|
-
return "(" +
|
2284
|
-
type_to_enum(((t_map*)ttype)->get_key_type()) + "," +
|
2285
|
-
type_to_spec_args(((t_map*)ttype)->get_key_type()) + "," +
|
2286
|
-
type_to_enum(((t_map*)ttype)->get_val_type()) + "," +
|
2287
|
-
type_to_spec_args(((t_map*)ttype)->get_val_type()) +
|
2288
|
-
")";
|
2289
|
-
|
2290
|
-
} else if (ttype->is_set()) {
|
2291
|
-
return "(" +
|
2292
|
-
type_to_enum(((t_set*)ttype)->get_elem_type()) + "," +
|
2293
|
-
type_to_spec_args(((t_set*)ttype)->get_elem_type()) +
|
2294
|
-
")";
|
2295
|
-
|
2296
|
-
} else if (ttype->is_list()) {
|
2297
|
-
return "(" +
|
2298
|
-
type_to_enum(((t_list*)ttype)->get_elem_type()) + "," +
|
2299
|
-
type_to_spec_args(((t_list*)ttype)->get_elem_type()) +
|
2300
|
-
")";
|
2301
|
-
}
|
2302
|
-
|
2303
|
-
throw "INVALID TYPE IN type_to_spec_args: " + ttype->get_name();
|
2304
|
-
}
|
2305
|
-
|
2306
|
-
|
2307
|
-
THRIFT_REGISTER_GENERATOR(py, "Python",
|
2308
|
-
" new_style: Generate new-style classes.\n" \
|
2309
|
-
" twisted: Generate Twisted-friendly RPC services.\n"
|
2310
|
-
);
|