g "bddlZddlZddlmZddlZddlmZddlmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZddlmZmZmZddlmZmZdd lmZd Zd Zd e d e!fdZ"de#d e#fdZ$d e#fdZ%deded efdZ&ded efdZ'dede d e efdZ(ded efdZ)deded efdZ*de e#efded e e#effdZ+y)N)md5) ascii_letters)AnyIteratorDictListcast) INT_MIN_VALUE INT_MAX_VALUELONG_MIN_VALUELONG_MAX_VALUE DAYS_SHIFT MLS_PER_HOUR MCS_PER_HOUR)extract_record_typeextract_logical_type parse_schema)Schema NamedSchemas) PRIMITIVESllnumreturncRtj|dzj|dS)Nlittle)random getrandbitsto_bytes)rs C/var/www/openai/venv/lib/python3.12/site-packages/fastavro/utils.py _randbytesr!s$   cAg & / /X >>stringcPt|jjS)N)rencode hexdigest)r#s r _md5r'!s v}}  ) ) ++r"cVdjtjtdS)N )k)joinrchoicesrr"r _gen_utf8r/%s 776>>-26 77r"schema named_schemasct|}t|}|dk(ry|dk(r-|dk(rtjjSt S|dk(r|dk(rKt jt dztjjjtz S|dk(r t jdtd zdz St jttS|d k(r|d k(r t jdt d zdz S|d k(s|d k(rt jdt"S|dk(s|dk(rt jdt$St jt&t(S|dk(rt j S|dk(rt j S|dk(rt+t jddS|dk(r t-dS|dk(r+t/t0t2t4f|}t-|dS|dk(rJt/t0t2t4f|}t jdt7|ddz }|d|S|dk(rCt/t0t2t8f|}t;dDcgc]}t=|d|c}S|dk(rLt/t0t2t8f|} t;dDcic]}t t=| d|c}S|dk(s|dk(rHt/t>t8|} t jdt7| dz }t=| ||S|dk(s|d k(rAt/t0t2t4f|} | d!D cic]} | d"t=| d#|c} St/t2|} t=|| |Scc}wcc}wcc} w)$Nnullr#z string-uuidintzint-dater zint-time-millisrlongzlong-time-microszlong-timestamp-milliszlong-local-timestamp-milliszlong-timestamp-microszlong-local-timestamp-microsfloatdoublebooleanbytesr*fixedsizeenumsymbolsarrayitemsmapvaluesunion error_unionrecorderrorfieldsnametype) rruuiduuid4hexr/rrandintrdatetimedatemax toordinalrr r rMAX_TIMESTAMP_MILLISMAX_TIMESTAMP_MICROSr rboolr!r rstrrlenrrangegen_datar)r0r1 record_type logical_type fixed_schema enum_schema real_index array_schema_ map_schema union_schema record_schemafield named_schemas r rXrX)s%f-K'/Lf  = (::<## #{   : %>> a!2!2!!K  , ,>>!\B%6%:; ;~~m];;   - ->>!\B%6%:; ; 3 3<<>>!%9: : 3 3<<>>!%9: :~~nn==  }}  }}  !FNN1a())  "~  DcNF3 ,v.//  4S>62 ^^As;y+A'BQ'FG 9%j11  Df-v6 HMb R 1g. > RR  $sF{+V4 2Y  K*X"6 F F    ;-#?DL&1 ^^As<'81'<=  Z0-@@  K7$:T#s(^V4 'x0 0 &M8E&M=A A0  C(  l3]CC'S  s N1N65N;c,tt|dS)a Returns a single instance of arbitrary data that conforms to the schema. Parameters ---------- schema Schema that data should conform to Example:: from fastavro import schemaless_writer from fastavro.utils import generate_one schema = { 'doc': 'A weather reading.', 'name': 'Weather', 'namespace': 'test', 'type': 'record', 'fields': [ {'name': 'station', 'type': 'string'}, {'name': 'time', 'type': 'long'}, {'name': 'temp', 'type': 'int'}, ], } with open('weather.avro', 'wb') as out: schemaless_writer(out, schema, generate_one(schema)) r )next generate_many)r0s r generate_onerhrs<  fa( ))r"countc#dKi}t||}t|D]}t||yw)a[ A generator that yields arbitrary data that conforms to the schema. It will yield a number of data structures equal to what is given in the count Parameters ---------- schema Schema that data should conform to count Number of objects to generate Example:: from fastavro import writer from fastavro.utils import generate_many schema = { 'doc': 'A weather reading.', 'name': 'Weather', 'namespace': 'test', 'type': 'record', 'fields': [ {'name': 'station', 'type': 'string'}, {'name': 'time', 'type': 'long'}, {'name': 'temp', 'type': 'int'}, ], } with open('weather.avro', 'wb') as out: writer(out, schema, generate_many(schema, 5)) N)rrWrX)r0rir1 parsed_schemar_s r rgrgs5B#%M 7M 5\}m44s.0c6i}t||}t||S)zReturns an anonymized schema Parameters ---------- schema Schema to anonymize Example:: from fastavro.utils import anonymize_schema anonymized_schema = anonymize_schema(original_schema) )r_anonymize_schema)r0r1rks r anonymize_schemarns"#%M 7M ]M ::r"ct|tr|Dcgc]}t||c}St|ts|tvr|Sdt |S|d}i}||d<d|vrt |d|d<|dk(rt|d||d<|S|dk(rt|d||d<|S|dk(r8dt |d |d <|d Dcgc]}dt |c}|d <|S|d k(rdt |d |d <|d |d <|S|d k(s|dk(r6dt |d |d <|dDcgc]}t ||c}|d<|S|tvr||d<|Scc}wcc}wcc}w)NA_rIdocr?r@rArBr=rHr>r;r<rErFrG) isinstancelistrmdictrr'anonymize_field)r0r1s schema_typerksymbolrcs r rmrms&$=CDV!!]3VDD % Z MV ~& &Vn  + f F?#'u #6M% ' !%6vg %VM' "2/E !&7x8H-&XM( #,)F "&(fVn)=(>$?M& !282C(2C"T&\N#2C(M) $$G #&(fVn)=(>$?M& !$*6NM& !H $ w(>&(fVn)=(>$?M& !CI(CS'CS%}5CS'M( #J &$/M& ![E6('sEE+Ercci}d|vrt|d|d<d|vr|dDcgc] }t|c}|d<d|vr|d|d<t|d|d<t|d||d<|Scc}w)NrqaliasesdefaultrHrI)r'rm)rcr1 parsed_fieldaliass r rurus$&L ~"5<0 UErs)  22LK'&?C?E? ,,,838FDVFDLFDSFDR**C*B$5&$5$5#$5N;V;;(0f0\0f0f S>*6 #s(^r"