P1iddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddl mZdd lmZeZGd d ej&ZGd d ej&ZGddej,ZGddej,ZGddej,ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZGd d!ej&Z Gd"d#ej,Z!Gd$d%ej,Z"Gd&d'ej,Z#Gd(d)ej,Z$Gd*d+ej,Z%Gd,d-ej,Z&Gd.d/ej,Z'Gd0d1ej,Z(Gd2d3ej,Z)Gd4d5ej,Z*Gd6d7ej,Z+Gd8d9ej,Z,Gd:d;ej,Z-Gd<d=ej&Z.Gd>d?ej,Z/Gd@dAej,Z0GdBdCej,Z1GdDdEej,Z2GdFdGej&Z3GdHdIej&Z4GdJdKej&Z5ddLl6m7Z7m8Z8m9Z9GdMdNej&Z:GdOdPej&Z;GdQdRej&Z<GdSdTej,Z=GdUdVej,Z>GdWdXej,Z?yY)Z) serializers)get_user_model)validate_passwordValidationError UserProfile)liuyao)datetime)RegexValidator)PersonceZdZejddZej ddZejddZ ejddZ ejddZ ejddd Z Gd d Zy ) !UserProfileRegistrationSerializerF*Birth date (optional) - format: YYYY-MM-DDrequired help_textz(Birth time (optional) - format: HH:MM:SSzPTwin type (optional) - choices: 0 (Not a twin), 1 (Elder twin), 2 (Younger twin)4Father date of birth (optional) - format: YYYY-MM-DD4Mother date of birth (optional) - format: YYYY-MM-DDzBAI regeneration permission (optional) - boolean, defaults to False)rdefaultrcBeZdZeZgdZddiddiddiddiddiddidZy)&UserProfileRegistrationSerializer.Meta birth_date birth_time twin_type father_dob mother_dobcan_regenerate_airFN__name__ __module__ __qualname__r modelfields extra_kwargs1/home/cursorai/projects/iching/api/serializers.pyMetar(sCk%u-%u-$e,%u-%u-",e!4  r(r*N)r!r"r#r DateFieldr TimeFieldr IntegerFieldrrr BooleanFieldrr*r'r(r)rr s&&&>J'&&<J) ((dI'&&HJ'&&HJ1 00V   r(rceZdZejddddiZejddiddZedZejdZ Gdd Z d Z d Z y ) UserRegistrationSerializerT input_typepassword) write_onlyrstyleF)r4r3rrc6eZdZeZgdZddiddiddiddidZy)UserRegistrationSerializer.Metaphoneemailr2 password2 first_name last_namegenderprofiler3TrF)r2r<r=r>Nr!r"r#Userr$r%r&r'r(r)r*r7:s3l%t,%u-$e,!5)  r(r*cd|vr"|d|dk7rtjddi t|d|j di}|j d}|dk(r+|j dstjdddgii|d k(r+|j d stjdd d gii|S#t$r/}tjdt|jid}~wwxYw) Nr;r2Password fields didn't match.r?rr$双大需要填写父亲出生日期r$双小需要填写母亲出生日期rrrlistmessagesgetselfdatae profile_datars r)validatez#UserRegistrationSerializer.validateDs $ 4 #3tK7H#H--{<[.\] ] N d:. / xx 2.  $$[1 >,"2"2<"@--yr*rQr`r'r(r)r0r04sk${$$tLZdKefH% %%\:,FSWbghI/?G "[ " "E 2F  &r(r0cZeZdZejdZejddZy)LoginSerializerTr5rr3N)r!r"r#rrar9r2r'r(r)rcrcms( !K ! !4 0E${$$dtDHr(rcc2eZdZejdZy)PasswordResetRequestSerializerTr5N)r!r"r#rrar:r'r(r)rfrfqs !K ! !4 0Er(rfceZdZejdZejdZejddZejddZdZ y)PasswordResetConfirmSerializerTr5)r3rcJ|d|dk7rtjddi|S)Nr2r;rCrrrMattrss r)rQz'PasswordResetConfirmSerializer.validatezs0  k 2 2--z;Z.[\ \ r(N) r!r"r#rrauidb64tokenr2r;rQr'r(r)rhrhtsS "[ " "D 1F !K ! !4 0E${$$tDH% %%EIr(rhc eZdZGddZy)UserProfileSerializerc>eZdZeZdZddiddiddiddiddiddidZy)UserProfileSerializer.MetarrF read_onlyTNr r'r(r)r*rrsCk%u-%u-$e,%u-%u-"-t!4  r(r*N)r!r"r#r*r'r(r)rprps    r(rpc8eZdZedZGddZdZy)UserSerializerFr5c8eZdZeZdZddiddiddiddiddidZy)UserSerializer.Meta)idr9r:r<r=r>r?is_temporary_userrFr9r:r<r=r>Nr@r'r(r)r*rws;n %( %(%u-$e,!5)  r(r*cL|jdd}|jD]\}}t||||j|Zddlm}|j j|\}}|jD]\}}t||||j|S)Nr?rrrS)rUrYrZr[accounts.modelsr rVrX) rMinstancer\rPattrr_r r?r]s r)updatezUserSerializer.updates%)))T: *//1KD% HdE *2   # 3*22@@h@O GW+113 eu- 4 LLNr(N)r!r"r#rpr?r*rr'r(r)rurus#U3G   r(rucFeZdZejdZGddZy)LiuyaoSerializerzContains the complete calculation result with the same structure as returned by the /api/liuyao/calc endpoint. See the /api/liuyao/calc endpoint documentation for the detailed structure.rceZdZeZdZdZy)LiuyaoSerializer.Meta__all__)rTuuid created_atlast_modified_atNr!r"r#r r$r%read_only_fieldsr'r(r)r*rsMr(r*N)r!r"r#r JSONFieldrNr*r'r(r)rrs& ; _ D NNr(rcLeZdZdZej ddZGddZy)LiuyaoRequestSerializerz@Serializer for request documentation - excludes read-only fieldsFz3JSON data containing calculation results (optional)rceZdZeZgdZy)LiuyaoRequestSerializer.Meta) qdatequestiony1y2y3y4y5y6readingfeedbackrNNr!r"r#r r$r%r'r(r)r*rs ir(r*N)r!r"r#__doc__rrrNr*r'r(r)rrs)J ; G D jjr(rceZdZejddZej dddZej dddZej dddZ ejd de d d g Z GddZ dZy)LiuyaoCalculatorSerializerF)rri') max_value min_valuer rDz ^\d{2}:\d{2}$!Invalid time format. Use 'HH:MM'.)regexmessage) max_lengthr validatorsceZdZeZgdZy)LiuyaoCalculatorSerializer.Meta) rrrrrrrusecuryearmonthdaytimeNrr'r(r)r*rs kr(r*c Z|jdd}|jd}|jd}|jd}|jdd}|s||| t||| t t |jd \}}d|cxkrdkrntd|cxkr dks tt |S#t$r>tjd|d |d d |d d d|d |d d |d d d|d |d d |d d d wxYw#t$rtjddiwxYw)NrFrrrrz00:00zInvalid date (-02dz).)rrr:r;r)rKr ValueErrorrrmapintsplit) rMrlrrrrrhoursminutess r)rQz#LiuyaoCalculatorSerializer.validatesw8U+yy  '"iiyy)4+0Aco uc*  djjo6NE7$"$  *+g););  *<  *< # !11,TF!E#;aCyK-dV1U3KqS L+D65+Qs3irJ3  --;/  s! B?/A D ?AD !D*N)r!r"r#rr.rr-rrrrar rr*rQr'r(r)rrs %[ % %ue DF #; # #ddU SD $K $ $rQ OE "+ " "R1u MC ;  &;   Dllr(rc$eZdZdZGddZy)BaziSerializerz1Serializer for Bazi Person model with full fieldsceZdZeZdZdZy)BaziSerializer.Meta)relation_good_countrelation_bad_count) created_byr updated_at bazi_resultN)r!r"r# BaziPersonr$excluderr'r(r)r*rs?Tr(r*Nr!r"r#rr*r'r(r)rrs;UUr(rc$eZdZdZGddZy)BaziRequestSerializerzESerializer for Bazi request documentation - excludes read-only fieldsceZdZeZgdZy)BaziRequestSerializer.Meta)namer>rrrrrnotesN)r!r"r#rr$r%r'r(r)r*rs qr(r*Nrr'r(r)rrsOrrr(rceZdZdZej ddZej ddZej ddZy)TongshuCalendarQuerySerializerz%Serializer for querying calendar dataFz/Year (defaults to current year if not provided)rz7Month (1-12, defaults to current month if not provided)zNumber of extra days to include before/after the month (default varies based on month layout, maximum 100, negative values treated as 0)N) r!r"r#rrr-rr extra_daysr'r(r)rrsK/ #; # #U>o pD $K $ $e?x yE)))5EOPJr(rceZdZdZej dZej dZej dZejddZ y ) LunarDateSerializerzSerializer for lunar date dataz Lunar yearrz Lunar monthz Lunar dayFzWhether this is a leap monthrN) r!r"r#rrr-ymdr.leapr'r(r)rr sP(   <8A   =9A   ;7A #; # #U>\ ]Dr(rc\eZdZdZej dZej dZy)BaziPillarForTongshuSerializerz)Serializer for Bazi pillar in Tongshu APIu!Heavenly stem (天干) in Chineseru"Earthly branch (地支) in ChineseN)r!r"r#rrragodearthr'r(r)rrs+3 +  *M NC !K ! !,P QEr(rcXeZdZdZedZedZedZedZy)BaziForTongshuSerializerz'Serializer for Bazi data in Tongshu APIz Year pillarrz Month pillarz Day pillarz Hour pillarN) r!r"r#rrrrrhourr'r(r)rrs-1 )M BD *^ DE (< @C )M BDr(rc"eZdZdZej dZejdZejdZ ejdZ e dZ edZej d Zej d Zej d Zej d Zej d ZejdZejdZej dZej dZy)DayDataSerializerz Serializer for detailed day datazDate in YYYY-MM-DD formatrz Day of monthz*Whether this day belongs to adjacent monthzWhether this day is todayzFour pillars informationzRaw lunar date informationz*Formatted lunar date in Chinese characterszLunar month or solar term namezLunar day or solar term timez$Leap month indicator (if applicable)u'Jianchu (建除) cycle value in Chinesez!Whether this day has a solar termIndex of the solar term (0-23)z!Month's earthly branch in ChinesezMonth's five element in EnglishN)r!r"r#rrradater-rr.is_adjacent_monthis_todayrbazirlunarlunar_formatted lunar_title lunar_desc lunar_leapjianchuhas_solar_termsolar_term_index month_earth month_elementr'r(r)rrs* ; +F GD "+ " "^ ( ; +N OD ; +Z [D $K $ $/O PEr(rceZdZdZej dZej dZejeje ddZ eje dZ e d Ze d Zy ) CalendarMonthSerializerz'Serializer for full month calendar dataGregorian yearrzGregorian monthzArray of days in a weekchildrz2Two-dimensional array of days (organized by weeks)zSolar terms in this monthzPrevious month referencezNext month referenceN)r!r"r#rrr-rr ListFieldrdaysr solar_termsr prev_month next_monthr'r(r)rr7s1 #; # #.> ?D $K $ $/@ AE ; #k###%/ G  D(+''!#-K0:TUJ/:PQJr(rc8eZdZdZej ddZy)Calendar10kQuerySerializerz(Serializer for querying calendar10k dataTz5Year to retrieve calendar data for (range: 1900-2100)rN)r!r"r#rrr-rr'r(r)rrIs2 #; # #T=t uDr(rceZdZdZej dZej dZej dZy)YearDataSerializerz'Serializer for year data in Calendar10kz7Year's element in English (wood/fire/earth/metal/water)rzChinese zodiac animalz Flying starN) r!r"r#rrraelementanimalstarr'r(r)rrMs=1#k##.ghG "[ " "-D EF ; = 9Dr(rceZdZdZej dZej dZejdZ ejdZ y)MonthCycleSerializerz3Serializer for a monthly cycle with two solar termsz"First solar term data of the monthrz#Second solar term data of the monthzCount of days in this monthzFlying star for this monthN) r!r"r#rr DictFieldfirstsecondr-countrarr'r(r)rrSsQ= !K ! !,P QE "[ " "-R SF $K $ $/L ME ; +G HDr(rcdeZdZdZej dddZej dddZy)PaginationSerializerz%Serializer for pagination informationFTz-Previous year (or false if at lower boundary)r allow_nullrz)Next year (or false if at upper boundary)N)r!r"r#rrr-prevnextr'r(r)r r Zs3/ #; # #UtO~ D #; # #UtOz {Dr(r ceZdZdZej dZedZeje dZ e dZ y) Calendar10kSerializerz)Serializer for full year calendar10k datarrzInformation about the yearzMonthly cycles with solar termsrzPagination informationN)r!r"r#rrr-rr year_datarrcyclesr paginater'r(r)rr_sG3 #; # #.> ?D"-IJI "[ " ")=)?Kl mF#.FGHr(rc^eZdZdZej dZej ddZy)TempLoginSerializerzm Serializer for temporary user login endpoint. Migrates temp user data to existing user account. Tr5rdN)r!r"r#rrrar9r2r'r(r)rrfs/ "K ! !4 0E${$$dtDHr(rceZdZdZej ddddidZej ddiddd Zedd Z Gd d Z dZ dZ dZ y)TempRegisterSerializerz Serializer for temporary user registration endpoint. Converts temporary user to permanent user (only phone & email required). TFr1r2zIPassword (optional) - if provided, password2 is required for confirmation)r3rr4rzXPassword confirmation (optional) - required if password is provided, must match password)r4r3rrz\Profile data (optional) - contains birth_date, birth_time, twin_type, father_dob, mother_dobrcBeZdZeZgdZddiddidddddddd dd Zy ) TempRegisterSerializer.Metar8rz+Phone number (required) - max 16 charactersz,"2"2<"@--y$((<"8--O/ !^DHH\$:--Q/  r(c<|r|dvrtjd|S)N)MFNz@Gender must be 'M' (male), 'F' (female), or 'N' (not specified).rjrMr_s r)validate_genderz(TempUserCreateSerializer.validate_gender6s) U/1--R  r(N)r!r"r#rrrarr>r+rr,rr-rrr birth_citytimezonerQr@r'r(r)r6r6s  !; [ D #[ " "UF '&&>J'&&9J) ((X I '&&HJ'&&HJ'&&)J %{$$=H 2r(r6c~eZdZdZej ejdZejddddZ d Z y ) ReportSubmissionSerializerz-Serializer for submitting AI analysis reportsz3Report category - reason for reporting the analysis)choicesrrFTz9Optional message with additional details about the report)rrr3rctjDcgc]}|d }}||vrtjd||Scc}w)z:Validate the report category is one of the allowed choicesrz"Invalid category. Must be one of: )rREPORT_CATEGORY_CHOICESrr)rMr_choicevalid_categoriess r)validate_categoryz,ReportSubmissionSerializer.validate_categoryNsR4>4V4VW4V&F1I4VW ( (--0RScRd.ef f Xs ?N) r!r"r#rr ChoiceFieldrrGcategoryrarrJr'r(r)rDrD@sH7&{&&22GH$k##M Gr(rDceZdZdZej ddZej ddZej ddZGddZ y ) BaziReportSerializerz3Serializer for viewing BaZi analysis report details get_bazi_report_category_displayTsourcersget_bazi_report_status_displayz%bazi_report_resolved_by.get_full_namec eZdZeZgdZgdZy)BaziReportSerializer.Meta) rxrbazi_analysis_reportedbazi_report_categorycategory_displaybazi_report_messagebazi_report_timestampbazi_report_statusstatus_displaybazi_report_admin_notesbazi_admin_actionresolved_by_namebazi_report_resolved_at)rxrrYrWr[r^r_Nr!r"r#rr$r%rr'r(r)r*rT]   r(r*N r!r"r#rrrarWr[r^r*r'r(r)rNrNVsS=,{,,4Vbfg*[**2R^bcN,{,,4[gkl   r(rNceZdZdZej ddZej ddZej ddZGddZ y ) NumberReportSerializerz5Serializer for viewing Number analysis report details"get_number_report_category_displayTrP get_number_report_status_displayz'number_report_resolved_by.get_full_namec eZdZeZgdZgdZy)NumberReportSerializer.Meta) rxrnumber_analysis_reportednumber_report_categoryrWnumber_report_messagenumber_report_timestampnumber_report_statusr[number_report_admin_notesnumber_admin_actionr^number_report_resolved_at)rxrrlrWr[r^rpNr`r'r(r)r*rhqrar(r*Nrbr'r(r)rdrdjsS?,{,,4Xdhi*[**2T`deN,{,,4]imn   r(rdceZdZdZej ddZej ddZej ddZGddZ y ) LiuYaoReportSerializerz5Serializer for viewing LiuYao analysis report detailsget_report_category_displayTrPget_report_status_displayz report_resolved_by.get_full_namec eZdZeZgdZgdZy)LiuYaoReportSerializer.Meta) rxranalysis_reportedreport_categoryrWreport_messagereport_timestamp report_statusr[report_admin_notes admin_actionr^report_resolved_at)rxrrzrWr[r^r~Nrr'r(r)r*rvs   r(r*Nrbr'r(r)rrrr~sS?,{,,4Q]ab*[**2MY]^N,{,,4Vbfg   r(rr) ConversationMessageConversationConfigc$eZdZdZGddZy)MessageSerializerzSerializer for Message modelc eZdZeZgdZddgZy)MessageSerializer.Meta) rxrolecontentrproviderr$metastatus error_messagerxrN)r!r"r#rr$r%rr'r(r)r*rsp ,/r(r*Nrr'r(r)rrs&00r(rceZdZdZej Zej ZejddZ ej Z ejddZ e ddZGddZd Zd Zd Zy ) ConversationSerializerz/Serializer for Conversation model with messagesget_message_countTrPr)manyrsc eZdZeZgdZgdZy)ConversationSerializer.Meta) rx person_id person_nametitle last_messagelast_message_at message_countrrrJrxrrNr!r"r#rr$r%rr'r(r)r*r >r(r*c|jr/|jjdk(r|jjS|jr|jjSyz1Get person ID from subject or legacy person fieldrNsubject content_type object_idpersonrxrMobjs r) get_person_idz$ConversationSerializer.get_person_idB ;;3;;33v=;;(( ( ZZ::== r(c|jrN|jjdk(r5|jj}|rt|dr |jSy|j r|j jSyz3Get person name from subject or legacy person fieldrrNrr get_objecthasattrrrrMr subject_objs r)get_person_namez&ConversationSerializer.get_person_namec ;;3;;33v=++002Kw{F;"'''ZZ::?? "r(c|j}|r*t|jdkDr|jdddzS|r |jSdSzGet the last message contentdNz...get_last_messagelenrrMrlast_msgs r)rz'ConversationSerializer.get_last_message_'')19c(BRBR>SVY>Yx%- Ft|`h`p`p FCE Fr(N)r!r"r#rrSerializerMethodFieldrrr-rr DateTimeFieldrrrJr*rrrr'r(r)rrs91 113I3+335K,K,,4GSWXM4;446L/k//|tTO dd;H>>Fr(rceZdZdZej Zej ZejddZ ej Z ejddZ GddZ dZd Zd Zy ) ConversationListSerializerz>Fr(rceZdZdZy)CreateConversationSerializerz*Serializer for creating a new conversationN)r!r"r#rr'r(r)rrs4r(rceZdZdZej dddZejddgdd d Zejd d gdddZ ej dddZ dZ y)SendMessageSerializerz2Serializer for sending a message in a conversationTz'User message text (max 2000 characters)r8)zh-hanszChinese (Simplified))enEnglishFrzLanguage for response)rErrr)groqGroq)openaiOpenAIz;AI provider override (requires can_regenerate_ai privilege))rErr rz8AI model override (requires can_regenerate_ai privilege)r ct|jdk(rtjdt|dkDrtjd|jS)zValidate message lengthrzMessage cannot be emptyrz%Message cannot exceed 2000 characters)rstriprrr?s r)validate_messagez&SendMessageSerializer.validate_messagesQ u{{}  "--.GH H u: --.UV V{{}r(N) r!r"r#rrrarrKlanguagerr$rr'r(r)rrs<#k##;G '{&&46GH) H '{&&!#78O H "K ! !L E r(rc8eZdZdZej ddZy)ConversationConfigSerializerz)Serializer for conversation configurationTz\Maximum number of messages allowed per conversation (only messages with status="sent" count))rsrN)r!r"r#rrr- max_messagesr'r(r)rrs3+;++pLr(rN)@rest_frameworkrdjango.contrib.authr'django.contrib.auth.password_validationrr%rr|r liuyao.modelsr r r$r bazi.modelsr rrAModelSerializerrr0 Serializerrcrfrhrprurrrrrrrrrrrrrrrr rrrr(r2r6rDrNrdrr ai.modelsrrrrrrrrrr'r(r)rs#&.E2' 1,% (C(C% N7!