}Dje8dZddlmZmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZddlmZmZmZmZmZmZmZmZmZmZddlmZdd lm Z dd l!m"Z"m#Z#dd l$m%Z%dd l&m'Z'dd l(m)Z)m*Z*m+Z+m,Z,m-Z-GddZ.e/dk(rddl0m1Z1e1jddZ3e.jie3Z5e6de5jne6de8e5jre6de8e5jte6de5jvjxde6de5jvjzde6de5jvj|de6de5jvj~de6de6e5jdyy) a Chart Data Factory Module This module provides factory classes for creating comprehensive chart data models that include all the pure data from astrological charts, including subjects, aspects, house comparisons, and other analytical data without the visual rendering components. This is designed to be the "pure data" counterpart to ChartDrawer, providing structured access to all chart information for API consumption, data analysis, or other programmatic uses. Key Features: - Comprehensive chart data including subjects and aspects - House comparison analysis for dual charts - Element and quality distributions - Relationship scoring for synastry charts - Flexible point and aspect filtering - Support for all chart types (Natal, Transit, Synastry, Composite, Return) Classes: ElementDistributionModel: Model for element distribution analysis QualityDistributionModel: Model for quality distribution analysis SingleChartDataModel: Model for single-subject chart data DualChartDataModel: Model for dual-subject chart data ChartDataFactory: Factory for creating chart data models Author: Giacomo Battaglia Copyright: (C) 2025 Kerykeion Project License: AGPL-3.0 )MappingUnionOptionalListLiteralcast)AspectsFactory)HouseComparisonFactory)RelationshipScoreFactory)KerykeionException ChartType ActiveAspect) AstrologicalSubjectModelCompositeSubjectModelPlanetReturnModelSingleChartAspectsModelDualChartAspectsModelElementDistributionModelQualityDistributionModelSingleChartDataModelDualChartDataModelChartDataModel)$KerykeionSettingsCelestialPointModel)AstrologicalPoint)find_common_active_pointsdistribute_percentages_to_100)DEFAULT_ACTIVE_ASPECTS)!DEFAULT_CELESTIAL_POINTS_SETTINGS) ElementQualityDistributionMethodcalculate_element_pointscalculate_quality_points!calculate_synastry_element_points!calculate_synastry_quality_pointsceZdZdZeddeddfdddddedeee e fd e eee fd e e e d e ed ed ede edede eeefdefdZedefddddeee e fd e e e d e edede eeefdef dZededdfdddded ed e e e d e ed ed edede eeefdefdZededfddddeded e e e d e ed edede eeefdefdZedefdddde d e e e d e edede eeefdef dZededfddddede d e e e d e ed edede eeefdefdZedefdddde d e e e d e edede eeefdef dZy)ChartDataFactorya Factory class for creating comprehensive chart data models. This factory creates ChartDataModel instances containing all the pure data from astrological charts, including subjects, aspects, house comparisons, and analytical metrics. It provides the structured data equivalent of ChartDrawer's visual output. The factory handles all chart types and automatically includes relevant analyses based on chart type (e.g., house comparison for dual charts, relationship scoring for synastry charts). Example: >>> # Create natal chart data >>> john = AstrologicalSubjectFactory.from_birth_data("John", 1990, 1, 1, 12, 0, "London", "GB") >>> natal_data = ChartDataFactory.create_chart_data("Natal", john) >>> print(f"Elements: Fire {natal_data.element_distribution.fire_percentage}%") >>> >>> # Create synastry chart data >>> jane = AstrologicalSubjectFactory.from_birth_data("Jane", 1992, 6, 15, 14, 30, "Paris", "FR") >>> synastry_data = ChartDataFactory.create_chart_data("Synastry", john, jane) >>> print(f"Relationship score: {synastry_data.relationship_score.score_value}") NTFweighted)axis_orb_limitdistribution_methodcustom_distribution_weights chart_type first_subjectsecond_subject active_pointsactive_aspectsinclude_house_comparisoninclude_relationship_scorer'r(r)returnc |dvr|std|d|dk(rt|ts td|dk(rt|ts td|dk(rt|ts td |s |j} nt ||j} |rt | |j} |d vrt j|| || } nN|td|dd } d } |dk(rd} d} n|dk(rd} d } n |dk(rd} d } t j||| ||| | } d }|rJ|rH|dvrDt|tr4t|ttfrt||| }|j}d }|dk(rB|r@|r>t|tr.t|trt|||}|j}g}tD]+}|d| vs t|}d|d<|j!|-|Dcgc] }t#d&i|}}|Dcgc]}|j$j'}}|dk(rf|rdt|tr3t|tr#t)|||||| }t+|||||| }nAt-||||| }t/||||| }n t-||||| }t/||||| }|d|dz|dz|dz}|dkDr t1|nddddd}t3|d|d|d|d|d|d|d|d}|d|d z|d!z}|dkDr t1|ndddd"}t5|d|d |d!|d|d |d!#}|d vr;t7t9t:d ||t9t<| j>||| |$S|td|dtAt9t:d|||t9tB| j>||||| |% Scc}wcc}w)'a< Create comprehensive chart data for the specified chart type. Args: chart_type: Type of chart to create data for first_subject: Primary astrological subject second_subject: Secondary subject (required for dual charts) active_points: Points to include in calculations (defaults to first_subject.active_points) active_aspects: Aspect types and orbs to use include_house_comparison: Whether to include house comparison for dual charts include_relationship_score: Whether to include relationship scoring for synastry axis_orb_limit: Optional orb threshold for chart axes (applies only to single chart aspects) distribution_method: Strategy for element/modality weighting ("pure_count" or "weighted") custom_distribution_weights: Optional overrides for the distribution weights Returns: ChartDataModel: Comprehensive chart data model Raises: KerykeionException: If chart type requirements are not met )TransitSynastryDualReturnChartzSecond subject is required for z charts. CompositezCFirst subject must be a CompositeSubjectModel for Composite charts.Returnz=Second subject must be a PlanetReturnModel for Return charts.SingleReturnChartzGFirst subject must be a PlanetReturnModel for SingleReturnChart charts.)Natalr6r8)r-r.r'NFr4Tr3r5)r-r.r'first_subject_is_fixedsecond_subject_is_fixed)r-)r'name is_active)methodcustom_weightsfirewaterearthairr)r@rBrCrA)r@rBrCrAfire_percentageearth_percentageair_percentagewater_percentagecardinalfixedmutable)rHrIrJ)rHrIrJcardinal_percentagefixed_percentagemutable_percentage)r*subjectaspectselement_distributionquality_distributionr-r.) r*r+r,rOhouse_comparisonrelationship_scorerPrQr-r.)"r isinstancerrr-rr single_chart_aspectsdual_chart_aspectsrr get_house_comparisonr get_relationship_scorerdictappendrr<lowerr"r#r r!rrrrrrrrOrr)r*r+r,r-r.r/r0r'r(r)effective_active_points aspects_modelr:r;rRhouse_comparison_factoryrSrelationship_score_factoryavailable_planets_setting_dictsbody body_dictavailable_planets_settingcelestial_points_nameselement_totalsquality_totalstotal_elementselement_percentagesrPtotal_qualitiesquality_percentagesrQs h/var/www/html/Astrologybackend/backend/venv/lib/python3.12/site-packages/kerykeion/chart_data_factory.pycreate_chart_dataz"ChartDataFactory.create_chart_data[skJ C CN$'FzlRZ%[\ \  $Z G\-]$%jk k  !*^EV*W$%de e , ,Z O`5a$%no o&3&A&A #&? }OjOj&k # &?@WYgYuYu&v # D D*??5-- M%(+J:,V^)_``&+ "&+ #Z')-&*.'y()-&*/'00)-&*/'*==5--'=(?M  6:Is;s-)ABz!9;L MH,B!>AX,($<#P#P#R "  #(B~-)ABz 8H.F!"#1.* &@%V%V%X"DF'5 BDF|66 J )- +&/66yA  B8Q  1 84 8Q !Q AZ!Z$))//"3!Z!Z  #-)ABz 8H"C-*!".#> ""C-*!".#> "":-*!.#> "":-*!.#> "6)&*: N6)&*: N 6 "^G%< <~g?V VYghmYn n  ! *. 9aA>  8' )u% )/709.u509 )4~g7NNQ_`iQjj" *. 9!:  8#J/ )"9- 3J ?0929=   D D'(Q RT^_%4mDLL%9%95- %(+J:,V^)_``%(P QS]^+-2MBJJ!1#5%9%95-  cQ "[s 0O'!O,)r(r)rNc8tj|d||||S)a Convenience method for creating natal chart data. Args: subject: Astrological subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Natal chart data r9r+r*r-r.r(r)r%rm)rNr-r.r(r)s rlcreate_natal_chart_dataz(ChartDataFactory.create_natal_chart_data\s., 11!') 3(C 2  c >tj|d||||||| S)a Convenience method for creating synastry chart data. Args: first_subject: First astrological subject second_subject: Second astrological subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use include_house_comparison: Whether to include house comparison include_relationship_score: Whether to include relationship scoring distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Synastry chart data r4) r+r*r,r-r.r/r0r(r)rp)r+r,r-r.r/r0r(r)s rlcreate_synastry_chart_dataz+ChartDataFactory.create_synastry_chart_data{s78 11'!)')%='A 3(C2  rr natal_subjecttransit_subjectc <tj|d||||||S)ac Convenience method for creating transit chart data. Args: natal_subject: Natal astrological subject transit_subject: Transit astrological subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use include_house_comparison: Whether to include house comparison distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Transit chart data r3r+r*r,r-r.r/r(r)rp)rurvr-r.r/r(r)s rlcreate_transit_chart_dataz*ChartDataFactory.create_transit_chart_datas44 11' *')%= 3(C2  rrcomposite_subjectc8tj|d||||S)a Convenience method for creating composite chart data. Args: composite_subject: Composite astrological subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Composite chart data r6rorp)rzr-r.r(r)s rlcreate_composite_chart_dataz,ChartDataFactory.create_composite_chart_datas., 11+"') 3(C 2  rrreturn_subjectc <tj|d||||||S)af Convenience method for creating planetary return chart data. Args: natal_subject: Natal astrological subject return_subject: Planetary return subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use include_house_comparison: Whether to include house comparison distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Return chart data r5rxrp)rur}r-r.r/r(r)s rlcreate_return_chart_dataz)ChartDataFactory.create_return_chart_datas44 11'()')%= 3(C2  rrc8tj|d||||S)a Convenience method for creating single wheel planetary return chart data. Args: return_subject: Planetary return subject active_points: Points to include in calculations active_aspects: Aspect types and orbs to use distribution_method: Strategy for element/modality weighting custom_distribution_weights: Optional overrides for distribution weights Returns: ChartDataModel: Single wheel return chart data r8rorp)r}r-r.r(r)s rl%create_single_wheel_return_chart_dataz6ChartDataFactory.create_single_wheel_return_chart_data s., 11(*') 3(C 2  rr)__name__ __module__ __qualname____doc__ staticmethodrr rrrrrrrrboolfloatrrstrrrmrqrtryr|rrrTrrrlr%r%Bsi0X\;?-C)-+0~+/@JEI~~57LN__`~!'?AR'R!ST~ %6 78 ~ \* ~ #' ~%)~!~>~&.gc5j.A%B~ ~~@<@-C AKEI  /1FHYYZ %6 78 \* >  &.gc5j.A%B     <<@-C)-+/ % AKEI% /% 0%  %6 78% \* % #' % %) % >% &.gc5j.A%B%  % % N<@-C)- " AKEI" /" 1"  %6 78" \* " #' " >" &.gc5j.A%B"  " " H<@-C AKEI  0 %6 78 \* >  &.gc5j.A%B     <<@-C)- " AKEI" /" )"  %6 78" \* " #' " >" &.gc5j.A%B"  " " H<@-C AKEI  ) %6 78 \* >  &.gc5j.A%B     rrr%__main__)AstrologicalSubjectFactoryz Test Subject)r<z Chart Type: zActive Points: z Aspects: zFire: %zEarth: zAir: zWater: z --- )indentN)Artypingrrrrrrkerykeion.aspectsr 3kerykeion.house_comparison.house_comparison_factoryr $kerykeion.relationship_score_factoryr kerykeion.schemasr r rkerykeion.schemas.kr_modelsrrrrrrrrrr!kerykeion.schemas.settings_modelsrkerykeion.schemas.kr_literalsrkerykeion.utilitiesrr#kerykeion.settings.config_constantsr!kerykeion.settings.chart_defaultsrkerykeion.charts.charts_utilsrr r!r"r#r%r&kerykeion.astrological_subject_factoryrfrom_current_timerNrq natal_dataprintr*lenr-rOrPrDrErFrGmodel_dump_jsonrTrrrlrsx<A@,VIII   SYFOg g T zQ;(::OG!99'BJ L../ 01 OC 8 89: ;< Ic*,,-. /0 F:22BBC1 EF GJ33DDEQ GH E*11@@A CD GJ33DDEQ GH ) * $ $A $ ./#rr