descriptor.cc 71 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2008 Google Inc. All rights reserved.
  3. // https://developers.google.com/protocol-buffers/
  4. //
  5. // Redistribution and use in source and binary forms, with or without
  6. // modification, are permitted provided that the following conditions are
  7. // met:
  8. //
  9. // * Redistributions of source code must retain the above copyright
  10. // notice, this list of conditions and the following disclaimer.
  11. // * Redistributions in binary form must reproduce the above
  12. // copyright notice, this list of conditions and the following disclaimer
  13. // in the documentation and/or other materials provided with the
  14. // distribution.
  15. // * Neither the name of Google Inc. nor the names of its
  16. // contributors may be used to endorse or promote products derived from
  17. // this software without specific prior written permission.
  18. //
  19. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30. // Author: petar@google.com (Petar Petrov)
  31. #include <Python.h>
  32. #include <frameobject.h>
  33. #include <string>
  34. #include <unordered_map>
  35. #include <google/protobuf/io/coded_stream.h>
  36. #include <google/protobuf/descriptor.pb.h>
  37. #include <google/protobuf/dynamic_message.h>
  38. #include <google/protobuf/pyext/descriptor.h>
  39. #include <google/protobuf/pyext/descriptor_containers.h>
  40. #include <google/protobuf/pyext/descriptor_pool.h>
  41. #include <google/protobuf/pyext/message.h>
  42. #include <google/protobuf/pyext/message_factory.h>
  43. #include <google/protobuf/pyext/scoped_pyobject_ptr.h>
  44. #include <google/protobuf/stubs/hash.h>
  45. #if PY_MAJOR_VERSION >= 3
  46. #define PyString_FromStringAndSize PyUnicode_FromStringAndSize
  47. #define PyString_Check PyUnicode_Check
  48. #define PyString_InternFromString PyUnicode_InternFromString
  49. #define PyInt_FromLong PyLong_FromLong
  50. #define PyInt_FromSize_t PyLong_FromSize_t
  51. #if PY_VERSION_HEX < 0x03030000
  52. #error "Python 3.0 - 3.2 are not supported."
  53. #endif
  54. #define PyString_AsStringAndSize(ob, charpp, sizep) \
  55. (PyUnicode_Check(ob) ? ((*(charpp) = const_cast<char*>( \
  56. PyUnicode_AsUTF8AndSize(ob, (sizep)))) == NULL \
  57. ? -1 \
  58. : 0) \
  59. : PyBytes_AsStringAndSize(ob, (charpp), (sizep)))
  60. #endif
  61. namespace google {
  62. namespace protobuf {
  63. namespace python {
  64. // Store interned descriptors, so that the same C++ descriptor yields the same
  65. // Python object. Objects are not immortal: this map does not own the
  66. // references, and items are deleted when the last reference to the object is
  67. // released.
  68. // This is enough to support the "is" operator on live objects.
  69. // All descriptors are stored here.
  70. std::unordered_map<const void*, PyObject*>* interned_descriptors;
  71. PyObject* PyString_FromCppString(const std::string& str) {
  72. return PyString_FromStringAndSize(str.c_str(), str.size());
  73. }
  74. // Check that the calling Python code is the global scope of a _pb2.py module.
  75. // This function is used to support the current code generated by the proto
  76. // compiler, which creates descriptors, then update some properties.
  77. // For example:
  78. // message_descriptor = Descriptor(
  79. // name='Message',
  80. // fields = [FieldDescriptor(name='field')]
  81. // message_descriptor.fields[0].containing_type = message_descriptor
  82. //
  83. // This code is still executed, but the descriptors now have no other storage
  84. // than the (const) C++ pointer, and are immutable.
  85. // So we let this code pass, by simply ignoring the new value.
  86. //
  87. // From user code, descriptors still look immutable.
  88. //
  89. // TODO(amauryfa): Change the proto2 compiler to remove the assignments, and
  90. // remove this hack.
  91. bool _CalledFromGeneratedFile(int stacklevel) {
  92. #ifndef PYPY_VERSION
  93. // This check is not critical and is somewhat difficult to implement correctly
  94. // in PyPy.
  95. PyFrameObject* frame = PyEval_GetFrame();
  96. if (frame == NULL) {
  97. return false;
  98. }
  99. while (stacklevel-- > 0) {
  100. frame = frame->f_back;
  101. if (frame == NULL) {
  102. return false;
  103. }
  104. }
  105. if (frame->f_code->co_filename == NULL) {
  106. return false;
  107. }
  108. char* filename;
  109. Py_ssize_t filename_size;
  110. if (PyString_AsStringAndSize(frame->f_code->co_filename,
  111. &filename, &filename_size) < 0) {
  112. // filename is not a string.
  113. PyErr_Clear();
  114. return false;
  115. }
  116. if ((filename_size < 3) ||
  117. (strcmp(&filename[filename_size - 3], ".py") != 0)) {
  118. // Cython's stack does not have .py file name and is not at global module
  119. // scope.
  120. return true;
  121. }
  122. if (filename_size < 7) {
  123. // filename is too short.
  124. return false;
  125. }
  126. if (strcmp(&filename[filename_size - 7], "_pb2.py") != 0) {
  127. // Filename is not ending with _pb2.
  128. return false;
  129. }
  130. if (frame->f_globals != frame->f_locals) {
  131. // Not at global module scope
  132. return false;
  133. }
  134. #endif
  135. return true;
  136. }
  137. // If the calling code is not a _pb2.py file, raise AttributeError.
  138. // To be used in attribute setters.
  139. static int CheckCalledFromGeneratedFile(const char* attr_name) {
  140. if (_CalledFromGeneratedFile(0)) {
  141. return 0;
  142. }
  143. PyErr_Format(PyExc_AttributeError,
  144. "attribute is not writable: %s", attr_name);
  145. return -1;
  146. }
  147. #ifndef PyVarObject_HEAD_INIT
  148. #define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
  149. #endif
  150. #ifndef Py_TYPE
  151. #define Py_TYPE(ob) (((PyObject*)(ob))->ob_type)
  152. #endif
  153. // Helper functions for descriptor objects.
  154. // A set of templates to retrieve the C++ FileDescriptor of any descriptor.
  155. template<class DescriptorClass>
  156. const FileDescriptor* GetFileDescriptor(const DescriptorClass* descriptor) {
  157. return descriptor->file();
  158. }
  159. template<>
  160. const FileDescriptor* GetFileDescriptor(const FileDescriptor* descriptor) {
  161. return descriptor;
  162. }
  163. template<>
  164. const FileDescriptor* GetFileDescriptor(const EnumValueDescriptor* descriptor) {
  165. return descriptor->type()->file();
  166. }
  167. template<>
  168. const FileDescriptor* GetFileDescriptor(const OneofDescriptor* descriptor) {
  169. return descriptor->containing_type()->file();
  170. }
  171. template<>
  172. const FileDescriptor* GetFileDescriptor(const MethodDescriptor* descriptor) {
  173. return descriptor->service()->file();
  174. }
  175. bool Reparse(
  176. PyMessageFactory* message_factory, const Message& from, Message* to) {
  177. // Reparse message.
  178. std::string serialized;
  179. from.SerializeToString(&serialized);
  180. io::CodedInputStream input(
  181. reinterpret_cast<const uint8*>(serialized.c_str()), serialized.size());
  182. input.SetExtensionRegistry(message_factory->pool->pool,
  183. message_factory->message_factory);
  184. bool success = to->ParseFromCodedStream(&input);
  185. if (!success) {
  186. return false;
  187. }
  188. return true;
  189. }
  190. // Converts options into a Python protobuf, and cache the result.
  191. //
  192. // This is a bit tricky because options can contain extension fields defined in
  193. // the same proto file. In this case the options parsed from the serialized_pb
  194. // have unknown fields, and we need to parse them again.
  195. //
  196. // Always returns a new reference.
  197. template<class DescriptorClass>
  198. static PyObject* GetOrBuildOptions(const DescriptorClass *descriptor) {
  199. // Options are cached in the pool that owns the descriptor.
  200. // First search in the cache.
  201. PyDescriptorPool* caching_pool = GetDescriptorPool_FromPool(
  202. GetFileDescriptor(descriptor)->pool());
  203. std::unordered_map<const void*, PyObject*>* descriptor_options =
  204. caching_pool->descriptor_options;
  205. if (descriptor_options->find(descriptor) != descriptor_options->end()) {
  206. PyObject *value = (*descriptor_options)[descriptor];
  207. Py_INCREF(value);
  208. return value;
  209. }
  210. // Similar to the C++ implementation, we return an Options object from the
  211. // default (generated) factory, so that client code know that they can use
  212. // extensions from generated files:
  213. // d.GetOptions().Extensions[some_pb2.extension]
  214. //
  215. // The consequence is that extensions not defined in the default pool won't
  216. // be available. If needed, we could add an optional 'message_factory'
  217. // parameter to the GetOptions() function.
  218. PyMessageFactory* message_factory =
  219. GetDefaultDescriptorPool()->py_message_factory;
  220. // Build the Options object: get its Python class, and make a copy of the C++
  221. // read-only instance.
  222. const Message& options(descriptor->options());
  223. const Descriptor *message_type = options.GetDescriptor();
  224. CMessageClass* message_class = message_factory::GetOrCreateMessageClass(
  225. message_factory, message_type);
  226. if (message_class == NULL) {
  227. PyErr_Format(PyExc_TypeError, "Could not retrieve class for Options: %s",
  228. message_type->full_name().c_str());
  229. return NULL;
  230. }
  231. ScopedPyObjectPtr value(
  232. PyEval_CallObject(message_class->AsPyObject(), NULL));
  233. Py_DECREF(message_class);
  234. if (value == NULL) {
  235. return NULL;
  236. }
  237. if (!PyObject_TypeCheck(value.get(), CMessage_Type)) {
  238. PyErr_Format(PyExc_TypeError, "Invalid class for %s: %s",
  239. message_type->full_name().c_str(),
  240. Py_TYPE(value.get())->tp_name);
  241. return NULL;
  242. }
  243. CMessage* cmsg = reinterpret_cast<CMessage*>(value.get());
  244. const Reflection* reflection = options.GetReflection();
  245. const UnknownFieldSet& unknown_fields(reflection->GetUnknownFields(options));
  246. if (unknown_fields.empty()) {
  247. cmsg->message->CopyFrom(options);
  248. } else {
  249. // Reparse options string! XXX call cmessage::MergeFromString
  250. if (!Reparse(message_factory, options, cmsg->message)) {
  251. PyErr_Format(PyExc_ValueError, "Error reparsing Options message");
  252. return NULL;
  253. }
  254. }
  255. // Cache the result.
  256. Py_INCREF(value.get());
  257. (*descriptor_options)[descriptor] = value.get();
  258. return value.release();
  259. }
  260. // Copy the C++ descriptor to a Python message.
  261. // The Python message is an instance of descriptor_pb2.DescriptorProto
  262. // or similar.
  263. template<class DescriptorProtoClass, class DescriptorClass>
  264. static PyObject* CopyToPythonProto(const DescriptorClass *descriptor,
  265. PyObject *target) {
  266. const Descriptor* self_descriptor =
  267. DescriptorProtoClass::default_instance().GetDescriptor();
  268. CMessage* message = reinterpret_cast<CMessage*>(target);
  269. if (!PyObject_TypeCheck(target, CMessage_Type) ||
  270. message->message->GetDescriptor() != self_descriptor) {
  271. PyErr_Format(PyExc_TypeError, "Not a %s message",
  272. self_descriptor->full_name().c_str());
  273. return NULL;
  274. }
  275. cmessage::AssureWritable(message);
  276. DescriptorProtoClass* descriptor_message =
  277. static_cast<DescriptorProtoClass*>(message->message);
  278. descriptor->CopyTo(descriptor_message);
  279. // Custom options might in unknown extensions. Reparse
  280. // the descriptor_message. Can't skip reparse when options unknown
  281. // fields is empty, because they might in sub descriptors' options.
  282. PyMessageFactory* message_factory =
  283. GetDefaultDescriptorPool()->py_message_factory;
  284. if (!Reparse(message_factory, *descriptor_message, descriptor_message)) {
  285. PyErr_Format(PyExc_ValueError, "Error reparsing descriptor message");
  286. return nullptr;
  287. }
  288. Py_RETURN_NONE;
  289. }
  290. // All Descriptors classes share the same memory layout.
  291. typedef struct PyBaseDescriptor {
  292. PyObject_HEAD
  293. // Pointer to the C++ proto2 descriptor.
  294. // Like all descriptors, it is owned by the global DescriptorPool.
  295. const void* descriptor;
  296. // Owned reference to the DescriptorPool, to ensure it is kept alive.
  297. PyDescriptorPool* pool;
  298. } PyBaseDescriptor;
  299. // FileDescriptor structure "inherits" from the base descriptor.
  300. typedef struct PyFileDescriptor {
  301. PyBaseDescriptor base;
  302. // The cached version of serialized pb. Either NULL, or a Bytes string.
  303. // We own the reference.
  304. PyObject *serialized_pb;
  305. } PyFileDescriptor;
  306. namespace descriptor {
  307. // Creates or retrieve a Python descriptor of the specified type.
  308. // Objects are interned: the same descriptor will return the same object if it
  309. // was kept alive.
  310. // 'was_created' is an optional pointer to a bool, and is set to true if a new
  311. // object was allocated.
  312. // Always return a new reference.
  313. template<class DescriptorClass>
  314. PyObject* NewInternedDescriptor(PyTypeObject* type,
  315. const DescriptorClass* descriptor,
  316. bool* was_created) {
  317. if (was_created) {
  318. *was_created = false;
  319. }
  320. if (descriptor == NULL) {
  321. PyErr_BadInternalCall();
  322. return NULL;
  323. }
  324. // See if the object is in the map of interned descriptors
  325. std::unordered_map<const void*, PyObject*>::iterator it =
  326. interned_descriptors->find(descriptor);
  327. if (it != interned_descriptors->end()) {
  328. GOOGLE_DCHECK(Py_TYPE(it->second) == type);
  329. Py_INCREF(it->second);
  330. return it->second;
  331. }
  332. // Create a new descriptor object
  333. PyBaseDescriptor* py_descriptor = PyObject_GC_New(
  334. PyBaseDescriptor, type);
  335. if (py_descriptor == NULL) {
  336. return NULL;
  337. }
  338. py_descriptor->descriptor = descriptor;
  339. // and cache it.
  340. interned_descriptors->insert(
  341. std::make_pair(descriptor, reinterpret_cast<PyObject*>(py_descriptor)));
  342. // Ensures that the DescriptorPool stays alive.
  343. PyDescriptorPool* pool = GetDescriptorPool_FromPool(
  344. GetFileDescriptor(descriptor)->pool());
  345. if (pool == NULL) {
  346. // Don't DECREF, the object is not fully initialized.
  347. PyObject_Del(py_descriptor);
  348. return NULL;
  349. }
  350. Py_INCREF(pool);
  351. py_descriptor->pool = pool;
  352. PyObject_GC_Track(py_descriptor);
  353. if (was_created) {
  354. *was_created = true;
  355. }
  356. return reinterpret_cast<PyObject*>(py_descriptor);
  357. }
  358. static void Dealloc(PyObject* pself) {
  359. PyBaseDescriptor* self = reinterpret_cast<PyBaseDescriptor*>(pself);
  360. // Remove from interned dictionary
  361. interned_descriptors->erase(self->descriptor);
  362. Py_CLEAR(self->pool);
  363. Py_TYPE(self)->tp_free(pself);
  364. }
  365. static int GcTraverse(PyObject* pself, visitproc visit, void* arg) {
  366. PyBaseDescriptor* self = reinterpret_cast<PyBaseDescriptor*>(pself);
  367. Py_VISIT(self->pool);
  368. return 0;
  369. }
  370. static int GcClear(PyObject* pself) {
  371. PyBaseDescriptor* self = reinterpret_cast<PyBaseDescriptor*>(pself);
  372. Py_CLEAR(self->pool);
  373. return 0;
  374. }
  375. static PyGetSetDef Getters[] = {
  376. {NULL}
  377. };
  378. PyTypeObject PyBaseDescriptor_Type = {
  379. PyVarObject_HEAD_INIT(&PyType_Type, 0) FULL_MODULE_NAME
  380. ".DescriptorBase", // tp_name
  381. sizeof(PyBaseDescriptor), // tp_basicsize
  382. 0, // tp_itemsize
  383. (destructor)Dealloc, // tp_dealloc
  384. 0, // tp_print
  385. 0, // tp_getattr
  386. 0, // tp_setattr
  387. 0, // tp_compare
  388. 0, // tp_repr
  389. 0, // tp_as_number
  390. 0, // tp_as_sequence
  391. 0, // tp_as_mapping
  392. 0, // tp_hash
  393. 0, // tp_call
  394. 0, // tp_str
  395. 0, // tp_getattro
  396. 0, // tp_setattro
  397. 0, // tp_as_buffer
  398. Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, // tp_flags
  399. "Descriptors base class", // tp_doc
  400. GcTraverse, // tp_traverse
  401. GcClear, // tp_clear
  402. 0, // tp_richcompare
  403. 0, // tp_weaklistoffset
  404. 0, // tp_iter
  405. 0, // tp_iternext
  406. 0, // tp_methods
  407. 0, // tp_members
  408. Getters, // tp_getset
  409. };
  410. } // namespace descriptor
  411. const void* PyDescriptor_AsVoidPtr(PyObject* obj) {
  412. if (!PyObject_TypeCheck(obj, &descriptor::PyBaseDescriptor_Type)) {
  413. PyErr_SetString(PyExc_TypeError, "Not a BaseDescriptor");
  414. return NULL;
  415. }
  416. return reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor;
  417. }
  418. namespace message_descriptor {
  419. // Unchecked accessor to the C++ pointer.
  420. static const Descriptor* _GetDescriptor(PyBaseDescriptor* self) {
  421. return reinterpret_cast<const Descriptor*>(self->descriptor);
  422. }
  423. static PyObject* GetName(PyBaseDescriptor* self, void *closure) {
  424. return PyString_FromCppString(_GetDescriptor(self)->name());
  425. }
  426. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  427. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  428. }
  429. static PyObject* GetFile(PyBaseDescriptor *self, void *closure) {
  430. return PyFileDescriptor_FromDescriptor(_GetDescriptor(self)->file());
  431. }
  432. static PyObject* GetConcreteClass(PyBaseDescriptor* self, void *closure) {
  433. // Returns the canonical class for the given descriptor.
  434. // This is the class that was registered with the primary descriptor pool
  435. // which contains this descriptor.
  436. // This might not be the one you expect! For example the returned object does
  437. // not know about extensions defined in a custom pool.
  438. CMessageClass* concrete_class(message_factory::GetMessageClass(
  439. GetDescriptorPool_FromPool(
  440. _GetDescriptor(self)->file()->pool())->py_message_factory,
  441. _GetDescriptor(self)));
  442. Py_XINCREF(concrete_class);
  443. return concrete_class->AsPyObject();
  444. }
  445. static PyObject* GetFieldsByName(PyBaseDescriptor* self, void *closure) {
  446. return NewMessageFieldsByName(_GetDescriptor(self));
  447. }
  448. static PyObject* GetFieldsByCamelcaseName(PyBaseDescriptor* self,
  449. void *closure) {
  450. return NewMessageFieldsByCamelcaseName(_GetDescriptor(self));
  451. }
  452. static PyObject* GetFieldsByNumber(PyBaseDescriptor* self, void *closure) {
  453. return NewMessageFieldsByNumber(_GetDescriptor(self));
  454. }
  455. static PyObject* GetFieldsSeq(PyBaseDescriptor* self, void *closure) {
  456. return NewMessageFieldsSeq(_GetDescriptor(self));
  457. }
  458. static PyObject* GetNestedTypesByName(PyBaseDescriptor* self, void *closure) {
  459. return NewMessageNestedTypesByName(_GetDescriptor(self));
  460. }
  461. static PyObject* GetNestedTypesSeq(PyBaseDescriptor* self, void *closure) {
  462. return NewMessageNestedTypesSeq(_GetDescriptor(self));
  463. }
  464. static PyObject* GetExtensionsByName(PyBaseDescriptor* self, void *closure) {
  465. return NewMessageExtensionsByName(_GetDescriptor(self));
  466. }
  467. static PyObject* GetExtensions(PyBaseDescriptor* self, void *closure) {
  468. return NewMessageExtensionsSeq(_GetDescriptor(self));
  469. }
  470. static PyObject* GetEnumsSeq(PyBaseDescriptor* self, void *closure) {
  471. return NewMessageEnumsSeq(_GetDescriptor(self));
  472. }
  473. static PyObject* GetEnumTypesByName(PyBaseDescriptor* self, void *closure) {
  474. return NewMessageEnumsByName(_GetDescriptor(self));
  475. }
  476. static PyObject* GetEnumValuesByName(PyBaseDescriptor* self, void *closure) {
  477. return NewMessageEnumValuesByName(_GetDescriptor(self));
  478. }
  479. static PyObject* GetOneofsByName(PyBaseDescriptor* self, void *closure) {
  480. return NewMessageOneofsByName(_GetDescriptor(self));
  481. }
  482. static PyObject* GetOneofsSeq(PyBaseDescriptor* self, void *closure) {
  483. return NewMessageOneofsSeq(_GetDescriptor(self));
  484. }
  485. static PyObject* IsExtendable(PyBaseDescriptor *self, void *closure) {
  486. if (_GetDescriptor(self)->extension_range_count() > 0) {
  487. Py_RETURN_TRUE;
  488. } else {
  489. Py_RETURN_FALSE;
  490. }
  491. }
  492. static PyObject* GetExtensionRanges(PyBaseDescriptor *self, void *closure) {
  493. const Descriptor* descriptor = _GetDescriptor(self);
  494. PyObject* range_list = PyList_New(descriptor->extension_range_count());
  495. for (int i = 0; i < descriptor->extension_range_count(); i++) {
  496. const Descriptor::ExtensionRange* range = descriptor->extension_range(i);
  497. PyObject* start = PyInt_FromLong(range->start);
  498. PyObject* end = PyInt_FromLong(range->end);
  499. PyList_SetItem(range_list, i, PyTuple_Pack(2, start, end));
  500. }
  501. return range_list;
  502. }
  503. static PyObject* GetContainingType(PyBaseDescriptor *self, void *closure) {
  504. const Descriptor* containing_type =
  505. _GetDescriptor(self)->containing_type();
  506. if (containing_type) {
  507. return PyMessageDescriptor_FromDescriptor(containing_type);
  508. } else {
  509. Py_RETURN_NONE;
  510. }
  511. }
  512. static int SetContainingType(PyBaseDescriptor *self, PyObject *value,
  513. void *closure) {
  514. return CheckCalledFromGeneratedFile("containing_type");
  515. }
  516. static PyObject* GetHasOptions(PyBaseDescriptor *self, void *closure) {
  517. const MessageOptions& options(_GetDescriptor(self)->options());
  518. if (&options != &MessageOptions::default_instance()) {
  519. Py_RETURN_TRUE;
  520. } else {
  521. Py_RETURN_FALSE;
  522. }
  523. }
  524. static int SetHasOptions(PyBaseDescriptor *self, PyObject *value,
  525. void *closure) {
  526. return CheckCalledFromGeneratedFile("has_options");
  527. }
  528. static PyObject* GetOptions(PyBaseDescriptor *self) {
  529. return GetOrBuildOptions(_GetDescriptor(self));
  530. }
  531. static int SetOptions(PyBaseDescriptor *self, PyObject *value,
  532. void *closure) {
  533. return CheckCalledFromGeneratedFile("_options");
  534. }
  535. static int SetSerializedOptions(PyBaseDescriptor *self, PyObject *value,
  536. void *closure) {
  537. return CheckCalledFromGeneratedFile("_serialized_options");
  538. }
  539. static PyObject* CopyToProto(PyBaseDescriptor *self, PyObject *target) {
  540. return CopyToPythonProto<DescriptorProto>(_GetDescriptor(self), target);
  541. }
  542. static PyObject* EnumValueName(PyBaseDescriptor *self, PyObject *args) {
  543. const char *enum_name;
  544. int number;
  545. if (!PyArg_ParseTuple(args, "si", &enum_name, &number))
  546. return NULL;
  547. const EnumDescriptor *enum_type =
  548. _GetDescriptor(self)->FindEnumTypeByName(enum_name);
  549. if (enum_type == NULL) {
  550. PyErr_SetString(PyExc_KeyError, enum_name);
  551. return NULL;
  552. }
  553. const EnumValueDescriptor *enum_value =
  554. enum_type->FindValueByNumber(number);
  555. if (enum_value == NULL) {
  556. PyErr_Format(PyExc_KeyError, "%d", number);
  557. return NULL;
  558. }
  559. return PyString_FromCppString(enum_value->name());
  560. }
  561. static PyObject* GetSyntax(PyBaseDescriptor *self, void *closure) {
  562. return PyString_InternFromString(
  563. FileDescriptor::SyntaxName(_GetDescriptor(self)->file()->syntax()));
  564. }
  565. static PyGetSetDef Getters[] = {
  566. { "name", (getter)GetName, NULL, "Last name"},
  567. { "full_name", (getter)GetFullName, NULL, "Full name"},
  568. { "_concrete_class", (getter)GetConcreteClass, NULL, "concrete class"},
  569. { "file", (getter)GetFile, NULL, "File descriptor"},
  570. { "fields", (getter)GetFieldsSeq, NULL, "Fields sequence"},
  571. { "fields_by_name", (getter)GetFieldsByName, NULL, "Fields by name"},
  572. { "fields_by_camelcase_name", (getter)GetFieldsByCamelcaseName, NULL,
  573. "Fields by camelCase name"},
  574. { "fields_by_number", (getter)GetFieldsByNumber, NULL, "Fields by number"},
  575. { "nested_types", (getter)GetNestedTypesSeq, NULL, "Nested types sequence"},
  576. { "nested_types_by_name", (getter)GetNestedTypesByName, NULL,
  577. "Nested types by name"},
  578. { "extensions", (getter)GetExtensions, NULL, "Extensions Sequence"},
  579. { "extensions_by_name", (getter)GetExtensionsByName, NULL,
  580. "Extensions by name"},
  581. { "extension_ranges", (getter)GetExtensionRanges, NULL, "Extension ranges"},
  582. { "enum_types", (getter)GetEnumsSeq, NULL, "Enum sequence"},
  583. { "enum_types_by_name", (getter)GetEnumTypesByName, NULL,
  584. "Enum types by name"},
  585. { "enum_values_by_name", (getter)GetEnumValuesByName, NULL,
  586. "Enum values by name"},
  587. { "oneofs_by_name", (getter)GetOneofsByName, NULL, "Oneofs by name"},
  588. { "oneofs", (getter)GetOneofsSeq, NULL, "Oneofs by name"},
  589. { "containing_type", (getter)GetContainingType, (setter)SetContainingType,
  590. "Containing type"},
  591. { "is_extendable", (getter)IsExtendable, (setter)NULL},
  592. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  593. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  594. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  595. "Serialized Options"},
  596. { "syntax", (getter)GetSyntax, (setter)NULL, "Syntax"},
  597. {NULL}
  598. };
  599. static PyMethodDef Methods[] = {
  600. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  601. { "CopyToProto", (PyCFunction)CopyToProto, METH_O, },
  602. { "EnumValueName", (PyCFunction)EnumValueName, METH_VARARGS, },
  603. {NULL}
  604. };
  605. } // namespace message_descriptor
  606. PyTypeObject PyMessageDescriptor_Type = {
  607. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  608. FULL_MODULE_NAME ".MessageDescriptor", // tp_name
  609. sizeof(PyBaseDescriptor), // tp_basicsize
  610. 0, // tp_itemsize
  611. 0, // tp_dealloc
  612. 0, // tp_print
  613. 0, // tp_getattr
  614. 0, // tp_setattr
  615. 0, // tp_compare
  616. 0, // tp_repr
  617. 0, // tp_as_number
  618. 0, // tp_as_sequence
  619. 0, // tp_as_mapping
  620. 0, // tp_hash
  621. 0, // tp_call
  622. 0, // tp_str
  623. 0, // tp_getattro
  624. 0, // tp_setattro
  625. 0, // tp_as_buffer
  626. Py_TPFLAGS_DEFAULT, // tp_flags
  627. "A Message Descriptor", // tp_doc
  628. 0, // tp_traverse
  629. 0, // tp_clear
  630. 0, // tp_richcompare
  631. 0, // tp_weaklistoffset
  632. 0, // tp_iter
  633. 0, // tp_iternext
  634. message_descriptor::Methods, // tp_methods
  635. 0, // tp_members
  636. message_descriptor::Getters, // tp_getset
  637. &descriptor::PyBaseDescriptor_Type, // tp_base
  638. };
  639. PyObject* PyMessageDescriptor_FromDescriptor(
  640. const Descriptor* message_descriptor) {
  641. return descriptor::NewInternedDescriptor(
  642. &PyMessageDescriptor_Type, message_descriptor, NULL);
  643. }
  644. const Descriptor* PyMessageDescriptor_AsDescriptor(PyObject* obj) {
  645. if (!PyObject_TypeCheck(obj, &PyMessageDescriptor_Type)) {
  646. PyErr_SetString(PyExc_TypeError, "Not a MessageDescriptor");
  647. return NULL;
  648. }
  649. return reinterpret_cast<const Descriptor*>(
  650. reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor);
  651. }
  652. namespace field_descriptor {
  653. // Unchecked accessor to the C++ pointer.
  654. static const FieldDescriptor* _GetDescriptor(
  655. PyBaseDescriptor *self) {
  656. return reinterpret_cast<const FieldDescriptor*>(self->descriptor);
  657. }
  658. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  659. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  660. }
  661. static PyObject* GetName(PyBaseDescriptor *self, void *closure) {
  662. return PyString_FromCppString(_GetDescriptor(self)->name());
  663. }
  664. static PyObject* GetCamelcaseName(PyBaseDescriptor* self, void *closure) {
  665. return PyString_FromCppString(_GetDescriptor(self)->camelcase_name());
  666. }
  667. static PyObject* GetJsonName(PyBaseDescriptor* self, void *closure) {
  668. return PyString_FromCppString(_GetDescriptor(self)->json_name());
  669. }
  670. static PyObject* GetFile(PyBaseDescriptor *self, void *closure) {
  671. return PyFileDescriptor_FromDescriptor(_GetDescriptor(self)->file());
  672. }
  673. static PyObject* GetType(PyBaseDescriptor *self, void *closure) {
  674. return PyInt_FromLong(_GetDescriptor(self)->type());
  675. }
  676. static PyObject* GetCppType(PyBaseDescriptor *self, void *closure) {
  677. return PyInt_FromLong(_GetDescriptor(self)->cpp_type());
  678. }
  679. static PyObject* GetLabel(PyBaseDescriptor *self, void *closure) {
  680. return PyInt_FromLong(_GetDescriptor(self)->label());
  681. }
  682. static PyObject* GetNumber(PyBaseDescriptor *self, void *closure) {
  683. return PyInt_FromLong(_GetDescriptor(self)->number());
  684. }
  685. static PyObject* GetIndex(PyBaseDescriptor *self, void *closure) {
  686. return PyInt_FromLong(_GetDescriptor(self)->index());
  687. }
  688. static PyObject* GetID(PyBaseDescriptor *self, void *closure) {
  689. return PyLong_FromVoidPtr(self);
  690. }
  691. static PyObject* IsExtension(PyBaseDescriptor *self, void *closure) {
  692. return PyBool_FromLong(_GetDescriptor(self)->is_extension());
  693. }
  694. static PyObject* HasDefaultValue(PyBaseDescriptor *self, void *closure) {
  695. return PyBool_FromLong(_GetDescriptor(self)->has_default_value());
  696. }
  697. static PyObject* GetDefaultValue(PyBaseDescriptor *self, void *closure) {
  698. PyObject *result;
  699. if (_GetDescriptor(self)->is_repeated()) {
  700. return PyList_New(0);
  701. }
  702. switch (_GetDescriptor(self)->cpp_type()) {
  703. case FieldDescriptor::CPPTYPE_INT32: {
  704. int32 value = _GetDescriptor(self)->default_value_int32();
  705. result = PyInt_FromLong(value);
  706. break;
  707. }
  708. case FieldDescriptor::CPPTYPE_INT64: {
  709. int64 value = _GetDescriptor(self)->default_value_int64();
  710. result = PyLong_FromLongLong(value);
  711. break;
  712. }
  713. case FieldDescriptor::CPPTYPE_UINT32: {
  714. uint32 value = _GetDescriptor(self)->default_value_uint32();
  715. result = PyInt_FromSize_t(value);
  716. break;
  717. }
  718. case FieldDescriptor::CPPTYPE_UINT64: {
  719. uint64 value = _GetDescriptor(self)->default_value_uint64();
  720. result = PyLong_FromUnsignedLongLong(value);
  721. break;
  722. }
  723. case FieldDescriptor::CPPTYPE_FLOAT: {
  724. float value = _GetDescriptor(self)->default_value_float();
  725. result = PyFloat_FromDouble(value);
  726. break;
  727. }
  728. case FieldDescriptor::CPPTYPE_DOUBLE: {
  729. double value = _GetDescriptor(self)->default_value_double();
  730. result = PyFloat_FromDouble(value);
  731. break;
  732. }
  733. case FieldDescriptor::CPPTYPE_BOOL: {
  734. bool value = _GetDescriptor(self)->default_value_bool();
  735. result = PyBool_FromLong(value);
  736. break;
  737. }
  738. case FieldDescriptor::CPPTYPE_STRING: {
  739. const std::string& value = _GetDescriptor(self)->default_value_string();
  740. result = ToStringObject(_GetDescriptor(self), value);
  741. break;
  742. }
  743. case FieldDescriptor::CPPTYPE_ENUM: {
  744. const EnumValueDescriptor* value =
  745. _GetDescriptor(self)->default_value_enum();
  746. result = PyInt_FromLong(value->number());
  747. break;
  748. }
  749. case FieldDescriptor::CPPTYPE_MESSAGE: {
  750. Py_RETURN_NONE;
  751. break;
  752. }
  753. default:
  754. PyErr_Format(PyExc_NotImplementedError, "default value for %s",
  755. _GetDescriptor(self)->full_name().c_str());
  756. return NULL;
  757. }
  758. return result;
  759. }
  760. static PyObject* GetCDescriptor(PyObject *self, void *closure) {
  761. Py_INCREF(self);
  762. return self;
  763. }
  764. static PyObject *GetEnumType(PyBaseDescriptor *self, void *closure) {
  765. const EnumDescriptor* enum_type = _GetDescriptor(self)->enum_type();
  766. if (enum_type) {
  767. return PyEnumDescriptor_FromDescriptor(enum_type);
  768. } else {
  769. Py_RETURN_NONE;
  770. }
  771. }
  772. static int SetEnumType(PyBaseDescriptor *self, PyObject *value, void *closure) {
  773. return CheckCalledFromGeneratedFile("enum_type");
  774. }
  775. static PyObject *GetMessageType(PyBaseDescriptor *self, void *closure) {
  776. const Descriptor* message_type = _GetDescriptor(self)->message_type();
  777. if (message_type) {
  778. return PyMessageDescriptor_FromDescriptor(message_type);
  779. } else {
  780. Py_RETURN_NONE;
  781. }
  782. }
  783. static int SetMessageType(PyBaseDescriptor *self, PyObject *value,
  784. void *closure) {
  785. return CheckCalledFromGeneratedFile("message_type");
  786. }
  787. static PyObject* GetContainingType(PyBaseDescriptor *self, void *closure) {
  788. const Descriptor* containing_type =
  789. _GetDescriptor(self)->containing_type();
  790. if (containing_type) {
  791. return PyMessageDescriptor_FromDescriptor(containing_type);
  792. } else {
  793. Py_RETURN_NONE;
  794. }
  795. }
  796. static int SetContainingType(PyBaseDescriptor *self, PyObject *value,
  797. void *closure) {
  798. return CheckCalledFromGeneratedFile("containing_type");
  799. }
  800. static PyObject* GetExtensionScope(PyBaseDescriptor *self, void *closure) {
  801. const Descriptor* extension_scope =
  802. _GetDescriptor(self)->extension_scope();
  803. if (extension_scope) {
  804. return PyMessageDescriptor_FromDescriptor(extension_scope);
  805. } else {
  806. Py_RETURN_NONE;
  807. }
  808. }
  809. static PyObject* GetContainingOneof(PyBaseDescriptor *self, void *closure) {
  810. const OneofDescriptor* containing_oneof =
  811. _GetDescriptor(self)->containing_oneof();
  812. if (containing_oneof) {
  813. return PyOneofDescriptor_FromDescriptor(containing_oneof);
  814. } else {
  815. Py_RETURN_NONE;
  816. }
  817. }
  818. static int SetContainingOneof(PyBaseDescriptor *self, PyObject *value,
  819. void *closure) {
  820. return CheckCalledFromGeneratedFile("containing_oneof");
  821. }
  822. static PyObject* GetHasOptions(PyBaseDescriptor *self, void *closure) {
  823. const FieldOptions& options(_GetDescriptor(self)->options());
  824. if (&options != &FieldOptions::default_instance()) {
  825. Py_RETURN_TRUE;
  826. } else {
  827. Py_RETURN_FALSE;
  828. }
  829. }
  830. static int SetHasOptions(PyBaseDescriptor *self, PyObject *value,
  831. void *closure) {
  832. return CheckCalledFromGeneratedFile("has_options");
  833. }
  834. static PyObject* GetOptions(PyBaseDescriptor *self) {
  835. return GetOrBuildOptions(_GetDescriptor(self));
  836. }
  837. static int SetOptions(PyBaseDescriptor *self, PyObject *value,
  838. void *closure) {
  839. return CheckCalledFromGeneratedFile("_options");
  840. }
  841. static int SetSerializedOptions(PyBaseDescriptor *self, PyObject *value,
  842. void *closure) {
  843. return CheckCalledFromGeneratedFile("_serialized_options");
  844. }
  845. static PyGetSetDef Getters[] = {
  846. { "full_name", (getter)GetFullName, NULL, "Full name"},
  847. { "name", (getter)GetName, NULL, "Unqualified name"},
  848. { "camelcase_name", (getter)GetCamelcaseName, NULL, "Camelcase name"},
  849. { "json_name", (getter)GetJsonName, NULL, "Json name"},
  850. { "file", (getter)GetFile, NULL, "File Descriptor"},
  851. { "type", (getter)GetType, NULL, "C++ Type"},
  852. { "cpp_type", (getter)GetCppType, NULL, "C++ Type"},
  853. { "label", (getter)GetLabel, NULL, "Label"},
  854. { "number", (getter)GetNumber, NULL, "Number"},
  855. { "index", (getter)GetIndex, NULL, "Index"},
  856. { "default_value", (getter)GetDefaultValue, NULL, "Default Value"},
  857. { "has_default_value", (getter)HasDefaultValue},
  858. { "is_extension", (getter)IsExtension, NULL, "ID"},
  859. { "id", (getter)GetID, NULL, "ID"},
  860. { "_cdescriptor", (getter)GetCDescriptor, NULL, "HAACK REMOVE ME"},
  861. { "message_type", (getter)GetMessageType, (setter)SetMessageType,
  862. "Message type"},
  863. { "enum_type", (getter)GetEnumType, (setter)SetEnumType, "Enum type"},
  864. { "containing_type", (getter)GetContainingType, (setter)SetContainingType,
  865. "Containing type"},
  866. { "extension_scope", (getter)GetExtensionScope, (setter)NULL,
  867. "Extension scope"},
  868. { "containing_oneof", (getter)GetContainingOneof, (setter)SetContainingOneof,
  869. "Containing oneof"},
  870. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  871. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  872. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  873. "Serialized Options"},
  874. {NULL}
  875. };
  876. static PyMethodDef Methods[] = {
  877. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  878. {NULL}
  879. };
  880. } // namespace field_descriptor
  881. PyTypeObject PyFieldDescriptor_Type = {
  882. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  883. FULL_MODULE_NAME ".FieldDescriptor", // tp_name
  884. sizeof(PyBaseDescriptor), // tp_basicsize
  885. 0, // tp_itemsize
  886. 0, // tp_dealloc
  887. 0, // tp_print
  888. 0, // tp_getattr
  889. 0, // tp_setattr
  890. 0, // tp_compare
  891. 0, // tp_repr
  892. 0, // tp_as_number
  893. 0, // tp_as_sequence
  894. 0, // tp_as_mapping
  895. 0, // tp_hash
  896. 0, // tp_call
  897. 0, // tp_str
  898. 0, // tp_getattro
  899. 0, // tp_setattro
  900. 0, // tp_as_buffer
  901. Py_TPFLAGS_DEFAULT, // tp_flags
  902. "A Field Descriptor", // tp_doc
  903. 0, // tp_traverse
  904. 0, // tp_clear
  905. 0, // tp_richcompare
  906. 0, // tp_weaklistoffset
  907. 0, // tp_iter
  908. 0, // tp_iternext
  909. field_descriptor::Methods, // tp_methods
  910. 0, // tp_members
  911. field_descriptor::Getters, // tp_getset
  912. &descriptor::PyBaseDescriptor_Type, // tp_base
  913. };
  914. PyObject* PyFieldDescriptor_FromDescriptor(
  915. const FieldDescriptor* field_descriptor) {
  916. return descriptor::NewInternedDescriptor(
  917. &PyFieldDescriptor_Type, field_descriptor, NULL);
  918. }
  919. const FieldDescriptor* PyFieldDescriptor_AsDescriptor(PyObject* obj) {
  920. if (!PyObject_TypeCheck(obj, &PyFieldDescriptor_Type)) {
  921. PyErr_SetString(PyExc_TypeError, "Not a FieldDescriptor");
  922. return NULL;
  923. }
  924. return reinterpret_cast<const FieldDescriptor*>(
  925. reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor);
  926. }
  927. namespace enum_descriptor {
  928. // Unchecked accessor to the C++ pointer.
  929. static const EnumDescriptor* _GetDescriptor(
  930. PyBaseDescriptor *self) {
  931. return reinterpret_cast<const EnumDescriptor*>(self->descriptor);
  932. }
  933. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  934. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  935. }
  936. static PyObject* GetName(PyBaseDescriptor *self, void *closure) {
  937. return PyString_FromCppString(_GetDescriptor(self)->name());
  938. }
  939. static PyObject* GetFile(PyBaseDescriptor *self, void *closure) {
  940. return PyFileDescriptor_FromDescriptor(_GetDescriptor(self)->file());
  941. }
  942. static PyObject* GetEnumvaluesByName(PyBaseDescriptor* self, void *closure) {
  943. return NewEnumValuesByName(_GetDescriptor(self));
  944. }
  945. static PyObject* GetEnumvaluesByNumber(PyBaseDescriptor* self, void *closure) {
  946. return NewEnumValuesByNumber(_GetDescriptor(self));
  947. }
  948. static PyObject* GetEnumvaluesSeq(PyBaseDescriptor* self, void *closure) {
  949. return NewEnumValuesSeq(_GetDescriptor(self));
  950. }
  951. static PyObject* GetContainingType(PyBaseDescriptor *self, void *closure) {
  952. const Descriptor* containing_type =
  953. _GetDescriptor(self)->containing_type();
  954. if (containing_type) {
  955. return PyMessageDescriptor_FromDescriptor(containing_type);
  956. } else {
  957. Py_RETURN_NONE;
  958. }
  959. }
  960. static int SetContainingType(PyBaseDescriptor *self, PyObject *value,
  961. void *closure) {
  962. return CheckCalledFromGeneratedFile("containing_type");
  963. }
  964. static PyObject* GetHasOptions(PyBaseDescriptor *self, void *closure) {
  965. const EnumOptions& options(_GetDescriptor(self)->options());
  966. if (&options != &EnumOptions::default_instance()) {
  967. Py_RETURN_TRUE;
  968. } else {
  969. Py_RETURN_FALSE;
  970. }
  971. }
  972. static int SetHasOptions(PyBaseDescriptor *self, PyObject *value,
  973. void *closure) {
  974. return CheckCalledFromGeneratedFile("has_options");
  975. }
  976. static PyObject* GetOptions(PyBaseDescriptor *self) {
  977. return GetOrBuildOptions(_GetDescriptor(self));
  978. }
  979. static int SetOptions(PyBaseDescriptor *self, PyObject *value,
  980. void *closure) {
  981. return CheckCalledFromGeneratedFile("_options");
  982. }
  983. static int SetSerializedOptions(PyBaseDescriptor *self, PyObject *value,
  984. void *closure) {
  985. return CheckCalledFromGeneratedFile("_serialized_options");
  986. }
  987. static PyObject* CopyToProto(PyBaseDescriptor *self, PyObject *target) {
  988. return CopyToPythonProto<EnumDescriptorProto>(_GetDescriptor(self), target);
  989. }
  990. static PyMethodDef Methods[] = {
  991. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  992. { "CopyToProto", (PyCFunction)CopyToProto, METH_O, },
  993. {NULL}
  994. };
  995. static PyGetSetDef Getters[] = {
  996. { "full_name", (getter)GetFullName, NULL, "Full name"},
  997. { "name", (getter)GetName, NULL, "last name"},
  998. { "file", (getter)GetFile, NULL, "File descriptor"},
  999. { "values", (getter)GetEnumvaluesSeq, NULL, "values"},
  1000. { "values_by_name", (getter)GetEnumvaluesByName, NULL,
  1001. "Enum values by name"},
  1002. { "values_by_number", (getter)GetEnumvaluesByNumber, NULL,
  1003. "Enum values by number"},
  1004. { "containing_type", (getter)GetContainingType, (setter)SetContainingType,
  1005. "Containing type"},
  1006. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  1007. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  1008. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  1009. "Serialized Options"},
  1010. {NULL}
  1011. };
  1012. } // namespace enum_descriptor
  1013. PyTypeObject PyEnumDescriptor_Type = {
  1014. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  1015. FULL_MODULE_NAME ".EnumDescriptor", // tp_name
  1016. sizeof(PyBaseDescriptor), // tp_basicsize
  1017. 0, // tp_itemsize
  1018. 0, // tp_dealloc
  1019. 0, // tp_print
  1020. 0, // tp_getattr
  1021. 0, // tp_setattr
  1022. 0, // tp_compare
  1023. 0, // tp_repr
  1024. 0, // tp_as_number
  1025. 0, // tp_as_sequence
  1026. 0, // tp_as_mapping
  1027. 0, // tp_hash
  1028. 0, // tp_call
  1029. 0, // tp_str
  1030. 0, // tp_getattro
  1031. 0, // tp_setattro
  1032. 0, // tp_as_buffer
  1033. Py_TPFLAGS_DEFAULT, // tp_flags
  1034. "A Enum Descriptor", // tp_doc
  1035. 0, // tp_traverse
  1036. 0, // tp_clear
  1037. 0, // tp_richcompare
  1038. 0, // tp_weaklistoffset
  1039. 0, // tp_iter
  1040. 0, // tp_iternext
  1041. enum_descriptor::Methods, // tp_methods
  1042. 0, // tp_members
  1043. enum_descriptor::Getters, // tp_getset
  1044. &descriptor::PyBaseDescriptor_Type, // tp_base
  1045. };
  1046. PyObject* PyEnumDescriptor_FromDescriptor(
  1047. const EnumDescriptor* enum_descriptor) {
  1048. return descriptor::NewInternedDescriptor(
  1049. &PyEnumDescriptor_Type, enum_descriptor, NULL);
  1050. }
  1051. const EnumDescriptor* PyEnumDescriptor_AsDescriptor(PyObject* obj) {
  1052. if (!PyObject_TypeCheck(obj, &PyEnumDescriptor_Type)) {
  1053. PyErr_SetString(PyExc_TypeError, "Not an EnumDescriptor");
  1054. return NULL;
  1055. }
  1056. return reinterpret_cast<const EnumDescriptor*>(
  1057. reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor);
  1058. }
  1059. namespace enumvalue_descriptor {
  1060. // Unchecked accessor to the C++ pointer.
  1061. static const EnumValueDescriptor* _GetDescriptor(
  1062. PyBaseDescriptor *self) {
  1063. return reinterpret_cast<const EnumValueDescriptor*>(self->descriptor);
  1064. }
  1065. static PyObject* GetName(PyBaseDescriptor *self, void *closure) {
  1066. return PyString_FromCppString(_GetDescriptor(self)->name());
  1067. }
  1068. static PyObject* GetNumber(PyBaseDescriptor *self, void *closure) {
  1069. return PyInt_FromLong(_GetDescriptor(self)->number());
  1070. }
  1071. static PyObject* GetIndex(PyBaseDescriptor *self, void *closure) {
  1072. return PyInt_FromLong(_GetDescriptor(self)->index());
  1073. }
  1074. static PyObject* GetType(PyBaseDescriptor *self, void *closure) {
  1075. return PyEnumDescriptor_FromDescriptor(_GetDescriptor(self)->type());
  1076. }
  1077. static PyObject* GetHasOptions(PyBaseDescriptor *self, void *closure) {
  1078. const EnumValueOptions& options(_GetDescriptor(self)->options());
  1079. if (&options != &EnumValueOptions::default_instance()) {
  1080. Py_RETURN_TRUE;
  1081. } else {
  1082. Py_RETURN_FALSE;
  1083. }
  1084. }
  1085. static int SetHasOptions(PyBaseDescriptor *self, PyObject *value,
  1086. void *closure) {
  1087. return CheckCalledFromGeneratedFile("has_options");
  1088. }
  1089. static PyObject* GetOptions(PyBaseDescriptor *self) {
  1090. return GetOrBuildOptions(_GetDescriptor(self));
  1091. }
  1092. static int SetOptions(PyBaseDescriptor *self, PyObject *value,
  1093. void *closure) {
  1094. return CheckCalledFromGeneratedFile("_options");
  1095. }
  1096. static int SetSerializedOptions(PyBaseDescriptor *self, PyObject *value,
  1097. void *closure) {
  1098. return CheckCalledFromGeneratedFile("_serialized_options");
  1099. }
  1100. static PyGetSetDef Getters[] = {
  1101. { "name", (getter)GetName, NULL, "name"},
  1102. { "number", (getter)GetNumber, NULL, "number"},
  1103. { "index", (getter)GetIndex, NULL, "index"},
  1104. { "type", (getter)GetType, NULL, "index"},
  1105. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  1106. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  1107. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  1108. "Serialized Options"},
  1109. {NULL}
  1110. };
  1111. static PyMethodDef Methods[] = {
  1112. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  1113. {NULL}
  1114. };
  1115. } // namespace enumvalue_descriptor
  1116. PyTypeObject PyEnumValueDescriptor_Type = {
  1117. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  1118. FULL_MODULE_NAME ".EnumValueDescriptor", // tp_name
  1119. sizeof(PyBaseDescriptor), // tp_basicsize
  1120. 0, // tp_itemsize
  1121. 0, // tp_dealloc
  1122. 0, // tp_print
  1123. 0, // tp_getattr
  1124. 0, // tp_setattr
  1125. 0, // tp_compare
  1126. 0, // tp_repr
  1127. 0, // tp_as_number
  1128. 0, // tp_as_sequence
  1129. 0, // tp_as_mapping
  1130. 0, // tp_hash
  1131. 0, // tp_call
  1132. 0, // tp_str
  1133. 0, // tp_getattro
  1134. 0, // tp_setattro
  1135. 0, // tp_as_buffer
  1136. Py_TPFLAGS_DEFAULT, // tp_flags
  1137. "A EnumValue Descriptor", // tp_doc
  1138. 0, // tp_traverse
  1139. 0, // tp_clear
  1140. 0, // tp_richcompare
  1141. 0, // tp_weaklistoffset
  1142. 0, // tp_iter
  1143. 0, // tp_iternext
  1144. enumvalue_descriptor::Methods, // tp_methods
  1145. 0, // tp_members
  1146. enumvalue_descriptor::Getters, // tp_getset
  1147. &descriptor::PyBaseDescriptor_Type, // tp_base
  1148. };
  1149. PyObject* PyEnumValueDescriptor_FromDescriptor(
  1150. const EnumValueDescriptor* enumvalue_descriptor) {
  1151. return descriptor::NewInternedDescriptor(
  1152. &PyEnumValueDescriptor_Type, enumvalue_descriptor, NULL);
  1153. }
  1154. namespace file_descriptor {
  1155. // Unchecked accessor to the C++ pointer.
  1156. static const FileDescriptor* _GetDescriptor(PyFileDescriptor *self) {
  1157. return reinterpret_cast<const FileDescriptor*>(self->base.descriptor);
  1158. }
  1159. static void Dealloc(PyFileDescriptor* self) {
  1160. Py_XDECREF(self->serialized_pb);
  1161. descriptor::Dealloc(reinterpret_cast<PyObject*>(self));
  1162. }
  1163. static PyObject* GetPool(PyFileDescriptor *self, void *closure) {
  1164. PyObject* pool = reinterpret_cast<PyObject*>(
  1165. GetDescriptorPool_FromPool(_GetDescriptor(self)->pool()));
  1166. Py_XINCREF(pool);
  1167. return pool;
  1168. }
  1169. static PyObject* GetName(PyFileDescriptor *self, void *closure) {
  1170. return PyString_FromCppString(_GetDescriptor(self)->name());
  1171. }
  1172. static PyObject* GetPackage(PyFileDescriptor *self, void *closure) {
  1173. return PyString_FromCppString(_GetDescriptor(self)->package());
  1174. }
  1175. static PyObject* GetSerializedPb(PyFileDescriptor *self, void *closure) {
  1176. PyObject *serialized_pb = self->serialized_pb;
  1177. if (serialized_pb != NULL) {
  1178. Py_INCREF(serialized_pb);
  1179. return serialized_pb;
  1180. }
  1181. FileDescriptorProto file_proto;
  1182. _GetDescriptor(self)->CopyTo(&file_proto);
  1183. std::string contents;
  1184. file_proto.SerializePartialToString(&contents);
  1185. self->serialized_pb = PyBytes_FromStringAndSize(
  1186. contents.c_str(), contents.size());
  1187. if (self->serialized_pb == NULL) {
  1188. return NULL;
  1189. }
  1190. Py_INCREF(self->serialized_pb);
  1191. return self->serialized_pb;
  1192. }
  1193. static PyObject* GetMessageTypesByName(PyFileDescriptor* self, void *closure) {
  1194. return NewFileMessageTypesByName(_GetDescriptor(self));
  1195. }
  1196. static PyObject* GetEnumTypesByName(PyFileDescriptor* self, void *closure) {
  1197. return NewFileEnumTypesByName(_GetDescriptor(self));
  1198. }
  1199. static PyObject* GetExtensionsByName(PyFileDescriptor* self, void *closure) {
  1200. return NewFileExtensionsByName(_GetDescriptor(self));
  1201. }
  1202. static PyObject* GetServicesByName(PyFileDescriptor* self, void *closure) {
  1203. return NewFileServicesByName(_GetDescriptor(self));
  1204. }
  1205. static PyObject* GetDependencies(PyFileDescriptor* self, void *closure) {
  1206. return NewFileDependencies(_GetDescriptor(self));
  1207. }
  1208. static PyObject* GetPublicDependencies(PyFileDescriptor* self, void *closure) {
  1209. return NewFilePublicDependencies(_GetDescriptor(self));
  1210. }
  1211. static PyObject* GetHasOptions(PyFileDescriptor *self, void *closure) {
  1212. const FileOptions& options(_GetDescriptor(self)->options());
  1213. if (&options != &FileOptions::default_instance()) {
  1214. Py_RETURN_TRUE;
  1215. } else {
  1216. Py_RETURN_FALSE;
  1217. }
  1218. }
  1219. static int SetHasOptions(PyFileDescriptor *self, PyObject *value,
  1220. void *closure) {
  1221. return CheckCalledFromGeneratedFile("has_options");
  1222. }
  1223. static PyObject* GetOptions(PyFileDescriptor *self) {
  1224. return GetOrBuildOptions(_GetDescriptor(self));
  1225. }
  1226. static int SetOptions(PyFileDescriptor *self, PyObject *value,
  1227. void *closure) {
  1228. return CheckCalledFromGeneratedFile("_options");
  1229. }
  1230. static int SetSerializedOptions(PyFileDescriptor *self, PyObject *value,
  1231. void *closure) {
  1232. return CheckCalledFromGeneratedFile("_serialized_options");
  1233. }
  1234. static PyObject* GetSyntax(PyFileDescriptor *self, void *closure) {
  1235. return PyString_InternFromString(
  1236. FileDescriptor::SyntaxName(_GetDescriptor(self)->syntax()));
  1237. }
  1238. static PyObject* CopyToProto(PyFileDescriptor *self, PyObject *target) {
  1239. return CopyToPythonProto<FileDescriptorProto>(_GetDescriptor(self), target);
  1240. }
  1241. static PyGetSetDef Getters[] = {
  1242. { "pool", (getter)GetPool, NULL, "pool"},
  1243. { "name", (getter)GetName, NULL, "name"},
  1244. { "package", (getter)GetPackage, NULL, "package"},
  1245. { "serialized_pb", (getter)GetSerializedPb},
  1246. { "message_types_by_name", (getter)GetMessageTypesByName, NULL,
  1247. "Messages by name"},
  1248. { "enum_types_by_name", (getter)GetEnumTypesByName, NULL, "Enums by name"},
  1249. { "extensions_by_name", (getter)GetExtensionsByName, NULL,
  1250. "Extensions by name"},
  1251. { "services_by_name", (getter)GetServicesByName, NULL, "Services by name"},
  1252. { "dependencies", (getter)GetDependencies, NULL, "Dependencies"},
  1253. { "public_dependencies", (getter)GetPublicDependencies, NULL, "Dependencies"},
  1254. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  1255. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  1256. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  1257. "Serialized Options"},
  1258. { "syntax", (getter)GetSyntax, (setter)NULL, "Syntax"},
  1259. {NULL}
  1260. };
  1261. static PyMethodDef Methods[] = {
  1262. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  1263. { "CopyToProto", (PyCFunction)CopyToProto, METH_O, },
  1264. {NULL}
  1265. };
  1266. } // namespace file_descriptor
  1267. PyTypeObject PyFileDescriptor_Type = {
  1268. PyVarObject_HEAD_INIT(&PyType_Type, 0) FULL_MODULE_NAME
  1269. ".FileDescriptor", // tp_name
  1270. sizeof(PyFileDescriptor), // tp_basicsize
  1271. 0, // tp_itemsize
  1272. (destructor)file_descriptor::Dealloc, // tp_dealloc
  1273. 0, // tp_print
  1274. 0, // tp_getattr
  1275. 0, // tp_setattr
  1276. 0, // tp_compare
  1277. 0, // tp_repr
  1278. 0, // tp_as_number
  1279. 0, // tp_as_sequence
  1280. 0, // tp_as_mapping
  1281. 0, // tp_hash
  1282. 0, // tp_call
  1283. 0, // tp_str
  1284. 0, // tp_getattro
  1285. 0, // tp_setattro
  1286. 0, // tp_as_buffer
  1287. Py_TPFLAGS_DEFAULT, // tp_flags
  1288. "A File Descriptor", // tp_doc
  1289. 0, // tp_traverse
  1290. 0, // tp_clear
  1291. 0, // tp_richcompare
  1292. 0, // tp_weaklistoffset
  1293. 0, // tp_iter
  1294. 0, // tp_iternext
  1295. file_descriptor::Methods, // tp_methods
  1296. 0, // tp_members
  1297. file_descriptor::Getters, // tp_getset
  1298. &descriptor::PyBaseDescriptor_Type, // tp_base
  1299. 0, // tp_dict
  1300. 0, // tp_descr_get
  1301. 0, // tp_descr_set
  1302. 0, // tp_dictoffset
  1303. 0, // tp_init
  1304. 0, // tp_alloc
  1305. 0, // tp_new
  1306. PyObject_GC_Del, // tp_free
  1307. };
  1308. PyObject* PyFileDescriptor_FromDescriptor(
  1309. const FileDescriptor* file_descriptor) {
  1310. return PyFileDescriptor_FromDescriptorWithSerializedPb(file_descriptor,
  1311. NULL);
  1312. }
  1313. PyObject* PyFileDescriptor_FromDescriptorWithSerializedPb(
  1314. const FileDescriptor* file_descriptor, PyObject *serialized_pb) {
  1315. bool was_created;
  1316. PyObject* py_descriptor = descriptor::NewInternedDescriptor(
  1317. &PyFileDescriptor_Type, file_descriptor, &was_created);
  1318. if (py_descriptor == NULL) {
  1319. return NULL;
  1320. }
  1321. if (was_created) {
  1322. PyFileDescriptor* cfile_descriptor =
  1323. reinterpret_cast<PyFileDescriptor*>(py_descriptor);
  1324. Py_XINCREF(serialized_pb);
  1325. cfile_descriptor->serialized_pb = serialized_pb;
  1326. }
  1327. // TODO(amauryfa): In the case of a cached object, check that serialized_pb
  1328. // is the same as before.
  1329. return py_descriptor;
  1330. }
  1331. const FileDescriptor* PyFileDescriptor_AsDescriptor(PyObject* obj) {
  1332. if (!PyObject_TypeCheck(obj, &PyFileDescriptor_Type)) {
  1333. PyErr_SetString(PyExc_TypeError, "Not a FileDescriptor");
  1334. return NULL;
  1335. }
  1336. return reinterpret_cast<const FileDescriptor*>(
  1337. reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor);
  1338. }
  1339. namespace oneof_descriptor {
  1340. // Unchecked accessor to the C++ pointer.
  1341. static const OneofDescriptor* _GetDescriptor(
  1342. PyBaseDescriptor *self) {
  1343. return reinterpret_cast<const OneofDescriptor*>(self->descriptor);
  1344. }
  1345. static PyObject* GetName(PyBaseDescriptor* self, void *closure) {
  1346. return PyString_FromCppString(_GetDescriptor(self)->name());
  1347. }
  1348. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  1349. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  1350. }
  1351. static PyObject* GetIndex(PyBaseDescriptor *self, void *closure) {
  1352. return PyInt_FromLong(_GetDescriptor(self)->index());
  1353. }
  1354. static PyObject* GetFields(PyBaseDescriptor* self, void *closure) {
  1355. return NewOneofFieldsSeq(_GetDescriptor(self));
  1356. }
  1357. static PyObject* GetContainingType(PyBaseDescriptor *self, void *closure) {
  1358. const Descriptor* containing_type =
  1359. _GetDescriptor(self)->containing_type();
  1360. if (containing_type) {
  1361. return PyMessageDescriptor_FromDescriptor(containing_type);
  1362. } else {
  1363. Py_RETURN_NONE;
  1364. }
  1365. }
  1366. static PyObject* GetHasOptions(PyBaseDescriptor *self, void *closure) {
  1367. const OneofOptions& options(_GetDescriptor(self)->options());
  1368. if (&options != &OneofOptions::default_instance()) {
  1369. Py_RETURN_TRUE;
  1370. } else {
  1371. Py_RETURN_FALSE;
  1372. }
  1373. }
  1374. static int SetHasOptions(PyBaseDescriptor *self, PyObject *value,
  1375. void *closure) {
  1376. return CheckCalledFromGeneratedFile("has_options");
  1377. }
  1378. static PyObject* GetOptions(PyBaseDescriptor *self) {
  1379. return GetOrBuildOptions(_GetDescriptor(self));
  1380. }
  1381. static int SetOptions(PyBaseDescriptor *self, PyObject *value,
  1382. void *closure) {
  1383. return CheckCalledFromGeneratedFile("_options");
  1384. }
  1385. static int SetSerializedOptions(PyBaseDescriptor *self, PyObject *value,
  1386. void *closure) {
  1387. return CheckCalledFromGeneratedFile("_serialized_options");
  1388. }
  1389. static PyGetSetDef Getters[] = {
  1390. { "name", (getter)GetName, NULL, "Name"},
  1391. { "full_name", (getter)GetFullName, NULL, "Full name"},
  1392. { "index", (getter)GetIndex, NULL, "Index"},
  1393. { "containing_type", (getter)GetContainingType, NULL, "Containing type"},
  1394. { "has_options", (getter)GetHasOptions, (setter)SetHasOptions, "Has Options"},
  1395. { "_options", (getter)NULL, (setter)SetOptions, "Options"},
  1396. { "_serialized_options", (getter)NULL, (setter)SetSerializedOptions,
  1397. "Serialized Options"},
  1398. { "fields", (getter)GetFields, NULL, "Fields"},
  1399. {NULL}
  1400. };
  1401. static PyMethodDef Methods[] = {
  1402. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS },
  1403. {NULL}
  1404. };
  1405. } // namespace oneof_descriptor
  1406. PyTypeObject PyOneofDescriptor_Type = {
  1407. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  1408. FULL_MODULE_NAME ".OneofDescriptor", // tp_name
  1409. sizeof(PyBaseDescriptor), // tp_basicsize
  1410. 0, // tp_itemsize
  1411. 0, // tp_dealloc
  1412. 0, // tp_print
  1413. 0, // tp_getattr
  1414. 0, // tp_setattr
  1415. 0, // tp_compare
  1416. 0, // tp_repr
  1417. 0, // tp_as_number
  1418. 0, // tp_as_sequence
  1419. 0, // tp_as_mapping
  1420. 0, // tp_hash
  1421. 0, // tp_call
  1422. 0, // tp_str
  1423. 0, // tp_getattro
  1424. 0, // tp_setattro
  1425. 0, // tp_as_buffer
  1426. Py_TPFLAGS_DEFAULT, // tp_flags
  1427. "A Oneof Descriptor", // tp_doc
  1428. 0, // tp_traverse
  1429. 0, // tp_clear
  1430. 0, // tp_richcompare
  1431. 0, // tp_weaklistoffset
  1432. 0, // tp_iter
  1433. 0, // tp_iternext
  1434. oneof_descriptor::Methods, // tp_methods
  1435. 0, // tp_members
  1436. oneof_descriptor::Getters, // tp_getset
  1437. &descriptor::PyBaseDescriptor_Type, // tp_base
  1438. };
  1439. PyObject* PyOneofDescriptor_FromDescriptor(
  1440. const OneofDescriptor* oneof_descriptor) {
  1441. return descriptor::NewInternedDescriptor(
  1442. &PyOneofDescriptor_Type, oneof_descriptor, NULL);
  1443. }
  1444. namespace service_descriptor {
  1445. // Unchecked accessor to the C++ pointer.
  1446. static const ServiceDescriptor* _GetDescriptor(
  1447. PyBaseDescriptor *self) {
  1448. return reinterpret_cast<const ServiceDescriptor*>(self->descriptor);
  1449. }
  1450. static PyObject* GetName(PyBaseDescriptor* self, void *closure) {
  1451. return PyString_FromCppString(_GetDescriptor(self)->name());
  1452. }
  1453. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  1454. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  1455. }
  1456. static PyObject* GetFile(PyBaseDescriptor *self, void *closure) {
  1457. return PyFileDescriptor_FromDescriptor(_GetDescriptor(self)->file());
  1458. }
  1459. static PyObject* GetIndex(PyBaseDescriptor *self, void *closure) {
  1460. return PyInt_FromLong(_GetDescriptor(self)->index());
  1461. }
  1462. static PyObject* GetMethods(PyBaseDescriptor* self, void *closure) {
  1463. return NewServiceMethodsSeq(_GetDescriptor(self));
  1464. }
  1465. static PyObject* GetMethodsByName(PyBaseDescriptor* self, void *closure) {
  1466. return NewServiceMethodsByName(_GetDescriptor(self));
  1467. }
  1468. static PyObject* FindMethodByName(PyBaseDescriptor *self, PyObject* arg) {
  1469. Py_ssize_t name_size;
  1470. char* name;
  1471. if (PyString_AsStringAndSize(arg, &name, &name_size) < 0) {
  1472. return NULL;
  1473. }
  1474. const MethodDescriptor* method_descriptor =
  1475. _GetDescriptor(self)->FindMethodByName(StringParam(name, name_size));
  1476. if (method_descriptor == NULL) {
  1477. PyErr_Format(PyExc_KeyError, "Couldn't find method %.200s", name);
  1478. return NULL;
  1479. }
  1480. return PyMethodDescriptor_FromDescriptor(method_descriptor);
  1481. }
  1482. static PyObject* GetOptions(PyBaseDescriptor *self) {
  1483. return GetOrBuildOptions(_GetDescriptor(self));
  1484. }
  1485. static PyObject* CopyToProto(PyBaseDescriptor *self, PyObject *target) {
  1486. return CopyToPythonProto<ServiceDescriptorProto>(_GetDescriptor(self),
  1487. target);
  1488. }
  1489. static PyGetSetDef Getters[] = {
  1490. { "name", (getter)GetName, NULL, "Name", NULL},
  1491. { "full_name", (getter)GetFullName, NULL, "Full name", NULL},
  1492. { "file", (getter)GetFile, NULL, "File descriptor"},
  1493. { "index", (getter)GetIndex, NULL, "Index", NULL},
  1494. { "methods", (getter)GetMethods, NULL, "Methods", NULL},
  1495. { "methods_by_name", (getter)GetMethodsByName, NULL, "Methods by name", NULL},
  1496. {NULL}
  1497. };
  1498. static PyMethodDef Methods[] = {
  1499. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS },
  1500. { "CopyToProto", (PyCFunction)CopyToProto, METH_O, },
  1501. { "FindMethodByName", (PyCFunction)FindMethodByName, METH_O },
  1502. {NULL}
  1503. };
  1504. } // namespace service_descriptor
  1505. PyTypeObject PyServiceDescriptor_Type = {
  1506. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  1507. FULL_MODULE_NAME ".ServiceDescriptor", // tp_name
  1508. sizeof(PyBaseDescriptor), // tp_basicsize
  1509. 0, // tp_itemsize
  1510. 0, // tp_dealloc
  1511. 0, // tp_print
  1512. 0, // tp_getattr
  1513. 0, // tp_setattr
  1514. 0, // tp_compare
  1515. 0, // tp_repr
  1516. 0, // tp_as_number
  1517. 0, // tp_as_sequence
  1518. 0, // tp_as_mapping
  1519. 0, // tp_hash
  1520. 0, // tp_call
  1521. 0, // tp_str
  1522. 0, // tp_getattro
  1523. 0, // tp_setattro
  1524. 0, // tp_as_buffer
  1525. Py_TPFLAGS_DEFAULT, // tp_flags
  1526. "A Service Descriptor", // tp_doc
  1527. 0, // tp_traverse
  1528. 0, // tp_clear
  1529. 0, // tp_richcompare
  1530. 0, // tp_weaklistoffset
  1531. 0, // tp_iter
  1532. 0, // tp_iternext
  1533. service_descriptor::Methods, // tp_methods
  1534. 0, // tp_members
  1535. service_descriptor::Getters, // tp_getset
  1536. &descriptor::PyBaseDescriptor_Type, // tp_base
  1537. };
  1538. PyObject* PyServiceDescriptor_FromDescriptor(
  1539. const ServiceDescriptor* service_descriptor) {
  1540. return descriptor::NewInternedDescriptor(
  1541. &PyServiceDescriptor_Type, service_descriptor, NULL);
  1542. }
  1543. const ServiceDescriptor* PyServiceDescriptor_AsDescriptor(PyObject* obj) {
  1544. if (!PyObject_TypeCheck(obj, &PyServiceDescriptor_Type)) {
  1545. PyErr_SetString(PyExc_TypeError, "Not a ServiceDescriptor");
  1546. return NULL;
  1547. }
  1548. return reinterpret_cast<const ServiceDescriptor*>(
  1549. reinterpret_cast<PyBaseDescriptor*>(obj)->descriptor);
  1550. }
  1551. namespace method_descriptor {
  1552. // Unchecked accessor to the C++ pointer.
  1553. static const MethodDescriptor* _GetDescriptor(
  1554. PyBaseDescriptor *self) {
  1555. return reinterpret_cast<const MethodDescriptor*>(self->descriptor);
  1556. }
  1557. static PyObject* GetName(PyBaseDescriptor* self, void *closure) {
  1558. return PyString_FromCppString(_GetDescriptor(self)->name());
  1559. }
  1560. static PyObject* GetFullName(PyBaseDescriptor* self, void *closure) {
  1561. return PyString_FromCppString(_GetDescriptor(self)->full_name());
  1562. }
  1563. static PyObject* GetIndex(PyBaseDescriptor *self, void *closure) {
  1564. return PyInt_FromLong(_GetDescriptor(self)->index());
  1565. }
  1566. static PyObject* GetContainingService(PyBaseDescriptor *self, void *closure) {
  1567. const ServiceDescriptor* containing_service =
  1568. _GetDescriptor(self)->service();
  1569. return PyServiceDescriptor_FromDescriptor(containing_service);
  1570. }
  1571. static PyObject* GetInputType(PyBaseDescriptor *self, void *closure) {
  1572. const Descriptor* input_type = _GetDescriptor(self)->input_type();
  1573. return PyMessageDescriptor_FromDescriptor(input_type);
  1574. }
  1575. static PyObject* GetOutputType(PyBaseDescriptor *self, void *closure) {
  1576. const Descriptor* output_type = _GetDescriptor(self)->output_type();
  1577. return PyMessageDescriptor_FromDescriptor(output_type);
  1578. }
  1579. static PyObject* GetOptions(PyBaseDescriptor *self) {
  1580. return GetOrBuildOptions(_GetDescriptor(self));
  1581. }
  1582. static PyObject* CopyToProto(PyBaseDescriptor *self, PyObject *target) {
  1583. return CopyToPythonProto<MethodDescriptorProto>(_GetDescriptor(self), target);
  1584. }
  1585. static PyGetSetDef Getters[] = {
  1586. { "name", (getter)GetName, NULL, "Name", NULL},
  1587. { "full_name", (getter)GetFullName, NULL, "Full name", NULL},
  1588. { "index", (getter)GetIndex, NULL, "Index", NULL},
  1589. { "containing_service", (getter)GetContainingService, NULL,
  1590. "Containing service", NULL},
  1591. { "input_type", (getter)GetInputType, NULL, "Input type", NULL},
  1592. { "output_type", (getter)GetOutputType, NULL, "Output type", NULL},
  1593. {NULL}
  1594. };
  1595. static PyMethodDef Methods[] = {
  1596. { "GetOptions", (PyCFunction)GetOptions, METH_NOARGS, },
  1597. { "CopyToProto", (PyCFunction)CopyToProto, METH_O, },
  1598. {NULL}
  1599. };
  1600. } // namespace method_descriptor
  1601. PyTypeObject PyMethodDescriptor_Type = {
  1602. PyVarObject_HEAD_INIT(&PyType_Type, 0)
  1603. FULL_MODULE_NAME ".MethodDescriptor", // tp_name
  1604. sizeof(PyBaseDescriptor), // tp_basicsize
  1605. 0, // tp_itemsize
  1606. 0, // tp_dealloc
  1607. 0, // tp_print
  1608. 0, // tp_getattr
  1609. 0, // tp_setattr
  1610. 0, // tp_compare
  1611. 0, // tp_repr
  1612. 0, // tp_as_number
  1613. 0, // tp_as_sequence
  1614. 0, // tp_as_mapping
  1615. 0, // tp_hash
  1616. 0, // tp_call
  1617. 0, // tp_str
  1618. 0, // tp_getattro
  1619. 0, // tp_setattro
  1620. 0, // tp_as_buffer
  1621. Py_TPFLAGS_DEFAULT, // tp_flags
  1622. "A Method Descriptor", // tp_doc
  1623. 0, // tp_traverse
  1624. 0, // tp_clear
  1625. 0, // tp_richcompare
  1626. 0, // tp_weaklistoffset
  1627. 0, // tp_iter
  1628. 0, // tp_iternext
  1629. method_descriptor::Methods, // tp_methods
  1630. 0, // tp_members
  1631. method_descriptor::Getters, // tp_getset
  1632. &descriptor::PyBaseDescriptor_Type, // tp_base
  1633. };
  1634. PyObject* PyMethodDescriptor_FromDescriptor(
  1635. const MethodDescriptor* method_descriptor) {
  1636. return descriptor::NewInternedDescriptor(
  1637. &PyMethodDescriptor_Type, method_descriptor, NULL);
  1638. }
  1639. // Add a enum values to a type dictionary.
  1640. static bool AddEnumValues(PyTypeObject *type,
  1641. const EnumDescriptor* enum_descriptor) {
  1642. for (int i = 0; i < enum_descriptor->value_count(); ++i) {
  1643. const EnumValueDescriptor* value = enum_descriptor->value(i);
  1644. ScopedPyObjectPtr obj(PyInt_FromLong(value->number()));
  1645. if (obj == NULL) {
  1646. return false;
  1647. }
  1648. if (PyDict_SetItemString(type->tp_dict, value->name().c_str(), obj.get()) <
  1649. 0) {
  1650. return false;
  1651. }
  1652. }
  1653. return true;
  1654. }
  1655. static bool AddIntConstant(PyTypeObject *type, const char* name, int value) {
  1656. ScopedPyObjectPtr obj(PyInt_FromLong(value));
  1657. if (PyDict_SetItemString(type->tp_dict, name, obj.get()) < 0) {
  1658. return false;
  1659. }
  1660. return true;
  1661. }
  1662. bool InitDescriptor() {
  1663. if (PyType_Ready(&PyMessageDescriptor_Type) < 0)
  1664. return false;
  1665. if (PyType_Ready(&PyFieldDescriptor_Type) < 0)
  1666. return false;
  1667. if (!AddEnumValues(&PyFieldDescriptor_Type,
  1668. FieldDescriptorProto::Label_descriptor())) {
  1669. return false;
  1670. }
  1671. if (!AddEnumValues(&PyFieldDescriptor_Type,
  1672. FieldDescriptorProto::Type_descriptor())) {
  1673. return false;
  1674. }
  1675. #define ADD_FIELDDESC_CONSTANT(NAME) AddIntConstant( \
  1676. &PyFieldDescriptor_Type, #NAME, FieldDescriptor::NAME)
  1677. if (!ADD_FIELDDESC_CONSTANT(CPPTYPE_INT32) ||
  1678. !ADD_FIELDDESC_CONSTANT(CPPTYPE_INT64) ||
  1679. !ADD_FIELDDESC_CONSTANT(CPPTYPE_UINT32) ||
  1680. !ADD_FIELDDESC_CONSTANT(CPPTYPE_UINT64) ||
  1681. !ADD_FIELDDESC_CONSTANT(CPPTYPE_DOUBLE) ||
  1682. !ADD_FIELDDESC_CONSTANT(CPPTYPE_FLOAT) ||
  1683. !ADD_FIELDDESC_CONSTANT(CPPTYPE_BOOL) ||
  1684. !ADD_FIELDDESC_CONSTANT(CPPTYPE_ENUM) ||
  1685. !ADD_FIELDDESC_CONSTANT(CPPTYPE_STRING) ||
  1686. !ADD_FIELDDESC_CONSTANT(CPPTYPE_MESSAGE)) {
  1687. return false;
  1688. }
  1689. #undef ADD_FIELDDESC_CONSTANT
  1690. if (PyType_Ready(&PyEnumDescriptor_Type) < 0)
  1691. return false;
  1692. if (PyType_Ready(&PyEnumValueDescriptor_Type) < 0)
  1693. return false;
  1694. if (PyType_Ready(&PyFileDescriptor_Type) < 0)
  1695. return false;
  1696. if (PyType_Ready(&PyOneofDescriptor_Type) < 0)
  1697. return false;
  1698. if (PyType_Ready(&PyServiceDescriptor_Type) < 0)
  1699. return false;
  1700. if (PyType_Ready(&PyMethodDescriptor_Type) < 0)
  1701. return false;
  1702. if (!InitDescriptorMappingTypes())
  1703. return false;
  1704. // Initialize globals defined in this file.
  1705. interned_descriptors = new std::unordered_map<const void*, PyObject*>;
  1706. return true;
  1707. }
  1708. } // namespace python
  1709. } // namespace protobuf
  1710. } // namespace google