svn commit: r987566 - /incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc

[br...@apache.org]

Author: bryanduxbury
Date: Fri Aug 20 16:45:13 2010
New Revision: 987566

URL: http://svn.apache.org/viewvc?rev=987566&view=rev
Log:
THRIFT-859. py: Allow py:twisted to be generated in different namespace than py

This patch adds support for the 'twisted' sub-namespace to the python code generator.

Patch: Bruce Lowekamp

Modified:
    incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc

Modified: incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc
URL: http://svn.apache.org/viewvc/incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc?rev=987566&r1=987565&r2=987566&view=diff
==============================================================================
--- incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc (original)
+++ incubator/thrift/trunk/compiler/cpp/src/generate/t_py_generator.cc Fri Aug 20 16:45:13 2010
@@ -188,7 +188,18 @@ class t_py_generator : public t_generato
   std::string type_to_enum(t_type* ttype);
   std::string type_to_spec_args(t_type* ttype);
 
-  static std::string get_real_py_module(const t_program* program) {
+  static bool is_valid_namespace(const std::string& sub_namespace) {
+    return sub_namespace == "twisted";
+  }
+
+  static std::string get_real_py_module(const t_program* program, bool gen_twisted) {
+    if(gen_twisted) {
+      std::string twisted_module = program->get_namespace("py.twisted");
+      if(!twisted_module.empty()){
+        return twisted_module;
+      }
+    }
+
     std::string real_module = program->get_namespace("py");
     if (real_module.empty()) {
       return program->get_name();
@@ -234,7 +245,7 @@ class t_py_generator : public t_generato
  */
 void t_py_generator::init_generator() {
   // Make output directory
-  string module = get_real_py_module(program_);
+  string module = get_real_py_module(program_, gen_twisted_);
   package_dir_ = get_out_dir();
   while (true) {
     // TODO: Do better error checking here.
@@ -298,7 +309,7 @@ string t_py_generator::render_includes()
   const vector<t_program*>& includes = program_->get_includes();
   string result = "";
   for (size_t i = 0; i < includes.size(); ++i) {
-    result += "import " + get_real_py_module(includes[i]) + ".ttypes\n";
+    result += "import " + get_real_py_module(includes[i], gen_twisted_) + ".ttypes\n";
   }
   if (includes.size() > 0) {
     result += "\n";
@@ -864,7 +875,7 @@ void t_py_generator::generate_service(t_
 
   if (tservice->get_extends() != NULL) {
     f_service_ <<
-      "import " << get_real_py_module(tservice->get_extends()->get_program()) <<
+      "import " << get_real_py_module(tservice->get_extends()->get_program(), gen_twisted_) <<
       "." << tservice->get_extends()->get_name() << endl;
   }
 
@@ -2258,10 +2269,10 @@ string t_py_generator::argument_list(t_s
 string t_py_generator::type_name(t_type* ttype) {
   t_program* program = ttype->get_program();
   if (ttype->is_service()) {
-    return get_real_py_module(program) + "." + ttype->get_name();
+    return get_real_py_module(program, gen_twisted_) + "." + ttype->get_name();
   }
   if (program != NULL && program != program_) {
-    return get_real_py_module(program) + ".ttypes." + ttype->get_name();
+    return get_real_py_module(program, gen_twisted_) + ".ttypes." + ttype->get_name();
   }
   return ttype->get_name();
 }