o PDi}0@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZddlmZddlmZedZgd Zd Zed gZd gZd gZGdddejZddZGdddZGdddZdS)N)UUID)RequestDataTooBig) Resolver404resolve)models) DataCollector) SilkyConfigzsilk.model_factory)zapplication/jsonzapplication/x-javascriptztext/javascriptztext/x-javascriptz text/x-jsonzmultipart/form-dataz!application/x-www-form-urlencodedz text/htmlztext/cssc@seZdZddZdS)DefaultEncodercCst|tr t|SdSN) isinstancerstr)selforK/var/www/Datamplify/venv/lib/python3.10/site-packages/silk/model_factory.pydefaults zDefaultEncoder.defaultN)__name__ __module__ __qualname__rrrrrr s r c Cs~d}|r;|d}|d}z|d}|d\}}|dkr(d}W||fSW||fSttfy:Y||fSw||fS)zRbest efforts on pulling out the content type and encoding from content-type headerN;r=charset)stripsplit IndexError ValueError) content_typechar_setsplt raw_char_setkeyrrr_parse_content_type$s"  r"csdeZdZdZdZfddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ZS)RequestModelFactoryz2Produce Request models from Django request objectsz********************cst||_dSr )super__init__request)r r& __class__rrr%9s  zRequestModelFactory.__init__cCs|jjdd}t|S)N content-type)r&headersgetr")r rrrrr=sz RequestModelFactory.content_typecCs|tttjtj}|dtjr|di}|jj D]\}}|}||vr/t j }|||<q t j|ttjdS)zu From Django docs (https://docs.djangoproject.com/en/2.0/ref/request-response/#httprequest-objects): authorizationcookie)cls ensure_ascii)setmapr lowerrSILKY_SENSITIVE_KEYSaddSILKY_HIDE_COOKIESr&r+itemsr#CLEANSED_SUBSTITUTEjsondumpsr SILKY_JSON_ENSURE_ASCII)r sensitive_headersr+kvrrrencoded_headersAs   z#RequestModelFactory.encoded_headersc stj}d|fddzt|}WnQtyh}zErOtddtjtj B}zt |dt j d|}WntyNt t|YnwWYd}~|SWYd}~|SWYd}~|Sd}~wwtj|tjd }|S) zU Mask credentials of potentially sensitive info before saving to db. |csttj}t|tr*|D]}r||rtj||<q||||<q|St|t r@t |D] \}}|||<q3|SrL|t |rLtjS|Sr ) recompileIr dictkeyssearchr#r8list enumerater )objpatternr!indexitem key_stringreplace_pattern_valuesrrrOZs    zERequestModelFactory._mask_credentials..replace_pattern_valuesz((z)[^=]*)=(.*?)(&|$)z\1=z\4Nr0)rr4joinr9loads ExceptionrArBMrCsubr#r8Loggerdebugr r:r;)r bodysensitive_keys json_bodyerJrrMr_mask_credentialsSs.     z%RequestModelFactory._mask_credentialscCszd}|tvr|jj}tjt|ddtjd}|S|tvr;ztjt |ddtjd}W|St y:|}Y|Sw|S)z Encode body as JSON if possible so can be used as a dictionary in generation of curl/django test client code r*T sort_keysindentr0) content_type_formr&POSTr9r:rDrr;content_types_jsonrRrS)r raw_bodyrrXrrr_bodyxs$ zRequestModelFactory._bodyc Cs|\}}|tkrd}d}||fSz|jj}Wn&ty>d}|jj}|jjD] \}}| ||q-||fYSw|rz| |}Wnet yQYn]t ytz| d}Wnt ygYn t yqd}YnwYn:ty}ztd||ftWYd}~n d}~wwz| d}Wnt yYn t yd}Ynwtj}d}|r|dkrtdtj|dd} |jj} | std | n-| |krtd | | |f|||}ntd | | |fd}n td |||}||}||}||fS) NsQRaw body not available for multipart_form data, Silk is not showing file uploads.r*sORaw body exceeds DATA_UPLOAD_MAX_MEMORY_SIZE, Silk is not showing file uploads.zUTF-8zZUnable to decode request body using char_set %s due to error: %s. Will ignore. Stacktrace:z*A max request size is set so checking size)rzBNo way in which to get size of request body for %s, will ignore itzJRequest %s has body of size %d which is less than %d so will save the bodyzKRequest %s has body of size %d which is greater than %d, therefore ignoringz0No maximum request body size is set, continuing.)rmultipart_formr&rXrrbcopyFILESr7 appendlistdecodeAttributeError LookupErrorUnicodeDecodeErrorrSrVerror traceback print_excrSILKY_MAX_REQUEST_BODY_SIZErWsys getsizeofpathrer\) r rrrdrXr=r>r[max_sizesizerequest_identifierrrrrXs                zRequestModelFactory.bodycCs<|jj}d}|rtt||}tj|tj d}|S)Nr*rP) r&GETrDziprEvaluesr9r:rr;)r query_paramsencoded_query_paramsquery_params_dictrrrr|s z RequestModelFactory.query_paramscCs*z t|jj}W|jStyYdSwr )rr& path_infor view_name)r resolvedrrrrs  zRequestModelFactory.view_namecCs|\}}|}|jj}|}tjjj|| |jj |||d}z||_ Wnt y6t dYnwt d|j|S)N)rur?methodr|rrXzNYI: Binary request bodiesz$Created new request model with pk %s)rXr|r&rurrRequestobjectscreater?rrdrnrVrWpk)r rXrdr|rur request_modelrrrconstruct_request_models&   z+RequestModelFactory.construct_request_model)rrr__doc__r8r%rr?r\rerXr|rr __classcell__rrr'rr#4s %Gr#cs0eZdZdZfddZddZddZZS)ResponseModelFactoryz6given a response object, craft the silk response modelcst||_tj|_dSr )r$r%responserr&)r rr'rrr%s zResponseModelFactory.__init__c Csd}t|jdd\}}t|jdd}|rtj}|dkrRtdt |d}|s3t dd}n||krFd}td||j j |fn td||j j |f|r|t vrztjt|d d tjd }W||fSttfytd |j j|fY||fSw||fS) Nr*r)contentrfz-Max size of response body defined so checkingz-Could not get size of response body. Ignoringz=Size of %d for %s is bigger than %d so ignoring response bodyz9Size of %d for %s is less than %d so saving response bodyTr]r^zMResponse to request with pk %s has content type %s but was unable to parse it)r"rr,getattrrSILKY_MAX_RESPONSE_BODY_SIZErVrWrsrtror&rurcr9r:rRr; TypeErrorrwarningr)r rXrrr max_body_sizerwrrrrXsL       zResponseModelFactory.bodyc Cs|jsJdtd|jj|\}}i}|jjD]$\}}zz|\}}Wnty7||}}YnwW|||<q|||<wt j |jj |jj t j|tjd|d}zt|} Wntyot|d} Ynw| d|_||S)Nzheadervalsilky_responserdrrrconstruct_response_model$s<     z-ResponseModelFactory.construct_response_model)rrrrr%rXrrrrr'rrs  $r)rr9loggingrArsrpuuidrdjango.core.exceptionsr django.urlsrrsilkrsilk.collectorr silk.configr getLoggerrVrcrgracontent_type_htmlcontent_type_css JSONEncoderr r"r#rrrrrs2      E