valitydev/thrift
14
0
Fork 0
mirror of https://github.com/valitydev/thrift.git synced 2024-11-07 18:58:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

THRIFT-257. py: Support validation of required fields

Browse Source 
Patch: Esteve Fernandez

git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@991987 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Bryan Duxbury 2010-09-02 15:37:19 +00:00
parent a9ca25a81e
commit c377c32168
1 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
compiler/cpp/src/generate/t_py_generator.cc
View File

@ -90,6 +90,7 @@ class t_py_generator : public t_generator {
void generate_py_struct_definition(std::ofstream& out, t_struct* tstruct, bool is_xception=false, bool is_result=false);
void generate_py_struct_reader(std::ofstream& out, t_struct* tstruct);
void generate_py_struct_writer(std::ofstream& out, t_struct* tstruct);
void generate_py_struct_required_validator(std::ofstream& out, t_struct* tstruct);
void generate_py_function_helpers(t_function* tfunction);
/**
@ -323,7 +324,7 @@ string t_py_generator::render_includes() {
string t_py_generator::render_fastbinary_includes() {
return
"from thrift.transport import TTransport\n"
"from thrift.protocol import TBinaryProtocol\n"
"from thrift.protocol import TBinaryProtocol, TProtocol\n"
"try:\n"
" from thrift.protocol import fastbinary\n"
"except:\n"
@ -853,11 +854,37 @@ void t_py_generator::generate_py_struct_writer(ofstream& out,
indent() << "oprot.writeFieldStop()" << endl <<
indent() << "oprot.writeStructEnd()" << endl;
generate_py_struct_required_validator(out, tstruct);
indent_down();
out <<
endl;
}
void t_py_generator::generate_py_struct_required_validator(ofstream& out,
t_struct* tstruct) {
indent(out) << "def validate(self):" << endl;
indent_up();
const vector<t_field*>& fields = tstruct->get_members();
if (fields.size() > 0) {
vector<t_field*>::const_iterator f_iter;
for (f_iter = fields.begin(); f_iter != fields.end(); ++f_iter) {
t_field* field = (*f_iter);
if (field->get_req() == t_field::T_REQUIRED) {
indent(out) << "if self." << field->get_name() << " is None:" << endl;
indent(out) << " raise TProtocol.TProtocolException(message='Required field " <<
field->get_name() << " is unset!')" << endl;
}
}
}
indent(out) << "return" << endl << endl;
indent_down();
}
/**
* Generates a thrift service.
*