U dRf@sdZddlmZddlZddlZddlZddlmZmZm Z m Z ddl m Z ddl mZdd d Zd d ZddZddZddZddZdS)zG thriftpy2.parser ~~~~~~~~~~~~~~~ Thrift parser using ply )absolute_importN)parseparse_fp threadlocal_cast)ThriftParserError) TPayloadMetautf-8c Cst|}t|||||d}tjr*t|||r~|tj|<|jddd}|r~|}|tjkrJ|tj|j <| |jdqJ|S)a2Load thrift file as a module. The module loaded and objects inside may only be pickled if module_name was provided. Note: `include_dir` will be depreacated in the future, use `include_dirs` instead. If `include_dir` was provided (not None), it will be appended to `include_dirs`. ) include_dirs include_direncodingZincludesN) boolrrincomplete_typefill_incomplete_ttypesysmodulesZ__thrift_meta__pop__name__extend) path module_namer r rZ real_modulethriftZ sub_modulesmoduler=/tmp/pip-unpacked-wheel-jqs7l_7o/thriftpy2/parser/__init__.pyloads     rc st|tr|ddkrDttj|dd|d}t||dS|dtjkrr|dtftj|dfS|dtjkrtftj|d}|d|d|d|dfSt|dtr|dt|dfSnt|tjr"|j D](\}}| dr qt ||t||qnt|t rp|j D]2\}}|dtjkrttftj|d}t|tr|d|d|d|df|j|<n6ttftj|dft|dd|j|<n|dtjkr"|d|dttftj|d|df|j|<nHt|dtr8fdd |d|dt|d|df|j|<q8nXt|d r|j D]@\}}t|d sq|j D]\}}t||j|<qq|S) z>Second pass of parser to handle out-of-order definitions. rZ UNKNOWN_CONSTrr __NcsBt|tr tfdd|DS|tjkr>tftj|S|S)Nc3s|]}|VqdS)Nr).0x)walkrr sz6fill_incomplete_ttype..walk..) isinstancetuplerrget_definition)parttmoduler"rrr"s   z#fill_incomplete_ttype..walkZthrift_services thrift_spec)r$r%r&rrrrtypes ModuleType__dict__items startswithsetattrr r*hasattr)r)Z definitionttypeZ real_typenameattrindexvaluerr(rr5s               rcCs|}|dD]p}t||d}|dkr6td||ft|trZ|dkrZtdtj|t|drvt|d|fS|SqdS)z?Get definition from thrift module and incomplete type map. .NzNo type found: %r, at line %drZ_ttype)splitgetattrrr$intrrr1)rr3linenoZref_typenrrrr&s  r&cCst||}|tj|<|S)z.Load thrift file like object as a module. )rrr)sourcerrrrrload_fps  r>cCs:d|kr.|dd\}}tt|dd|g|St|SdS)Nr7r)rsplitr9 __import__) import_namerobjrrr_import_modulesrCcCs|dstd|tjkr&tj|Sd|krl|dd\}}t|}tjtj |j }tj ||}n|}d |dd}t ||d}|tj|<tj|S) zLoad thrift_file by fullname, fullname should have '_thrift' as suffix. The loader will replace the '_thrift' with '.thrift' and use it as filename to locate the real thrift file. Z_thriftz4thriftpy2 can only load module with '_thrift' suffixr7rz {}.thriftNi)r)endswith ImportErrorrrr?rCosrdirnameabspath__file__joinformatr)fullnamerZthrift_module_namerZ path_prefixrZ thrift_filerrr load_modules      rM)NNNr )__doc__ __future__rrFrr+parserrrrrexcrrr rrr&r>rCrMrrrrs"    a