a w^i6@sdZddlZddlZddlmZddlmZddlmZGdddZ dd Z d d Z d d Z ddZ ddZddZddZddZddZdS)a Relationship Manager Module This module contains all relationship management functions for the BaoLife game. It handles romantic relationships, dating activities, affinity updates, and relationship data management. Functions: - updateAffinity: Updates affinity score for a person - handleRelationships: Manages relationship events and score updates - getRelData: Fetches relationship data by person ID - romance: Initiates a romantic relationship with a target - getActiveRelationship: Returns the active dating/married relationship - breakUp: Ends a relationship with a partner - partnerGift: Gives a gift to partner to increase affinity - dateNight: Executes a date night activity with partner Classes: - DateIdea: Represents a date activity with costs and rewards Imports: - relationshipClass: Imported from core.models - get_person: Imported from character.character_manager Author: BaoLife Development Team N)messageFunction)relationshipClass) get_personc@seZdZdZdddZdS)DateIdeaad Represents a date activity option with associated costs and benefits. Attributes: name: Name of the date activity energy_cost: Energy points required for this activity money_cost: Money cost for this activity message: Description message shown to player image: URL to image representing this date type NcCs"||_||_||_||_||_dS)N)name energy_cost money_costmessageimage)selfrrrr r r C/var/www/lichun.app/lichun/ws/relationships/relationship_manager.py__init__8s zDateIdea.__init__)N)__name__ __module__ __qualname____doc__rr r r r r-s rcCs6|jD]*}|j|ks(|j|ks(|j|kr|SqdS)z Fetch relationship data based on person1 or person2 id. Args: player: The player object person_id: The ID of the person you are searching for in the relationships Returns: Relationship data if found, else None N)relDataperson1person2id)playerZ person_idrelr r r getRelDataDs  rcCs,|jD] }|jdks|jdkr|SqdS)z Get the player's current active romantic relationship. Args: player: The player object Returns: Active relationship (Dating or Married status) if found, None otherwise DatingMarriedN)rrelationshipStatus)rrr r r getActiveRelationshipUs  rcCs:t|jD]*\}}|j|kr |j|j||j|_q |S)a9 Update the affinity score for a specific person in the player's relationships. Args: player: The player object id: The ID of the person whose affinity to update value: The amount to change affinity by (can be positive or negative) Returns: Updated player object ) enumeraterraffinity)rrvalueindexitemr r r updateAffinityis  r$cCstdd}t||j}|s dS|dkrtgd}|j||dkrZ|jd7_n&|dkrr|jd8_n|jd 7_|jd7_|jd kr|jd 7_n|jd kr|jd 8_d S)a Handle relationship events and score updates for a person. This function: - Has a 5% chance of triggering random relationship events - Updates relationship score weekly - Applies correction mechanism to keep scores balanced Args: player: The player object person: The person object to handle relationship with Returns: False if no relationship data exists, otherwise modifies relationship in place dF_)Went on a romantic tripHad a disagreementzMet with friendsr( r)(<N)randomrandintrrchoice eventsLogappendrelationshipScore)rpersonZ event_chancereventr r r handleRelationships{s"     r7cCst||}|r|jdkrt|jj|j|jdd}|j||j|j_|j|_|j d|j d|j dddl m }||d d Sd S) a\ Attempt to start a romantic relationship with a target person. Checks if the target has sufficient affinity (>= 50) and creates a new relationship if successful. Args: player: The player object partner: The ID of the person to romance Returns: True if romance attempt successful, False otherwise 2ZProspectzYou are now dating.z You have attempted a romance of  .rapply_event_modifiersZ start_datingTF)rr rcrdaterr3 relationship messageQueue firstnamelastnamemessaging_styler<)rpartnerZromanticTargetr?r<r r r romances       rEcCsP|jD]D}|j|krd|_d|j_|jdddlm}||ddSqdS) z End a romantic relationship with a partner. Args: player: The player object partner: The ID of the partner to break up with Returns: True if breakup successful, False otherwise zBroke UpNzYou dumped your partner.rr;ZbreakupTF) rrrr=r?r@r3rCr<)rrDrr<r r r breakUps     rFc Cs|jD]}|j|kr|jjdkr|jjd8_dtdd}t||}|j|7_dt|d}d}|j |t d||d d |d|d Sd Sqd S)a2 Give a gift to your partner to increase affinity. Costs $100 and increases affinity by 6-15 points. Args: player: The player object partner: The ID of the partner to give gift to Returns: messageEvent if successful, False if player doesn't have enough money r&r+r%r*z,You bought a gift for your partner, gaining z affinity.z1https://lichun.app/assets/images/partnerGift.jpeg partnerGiftTZGift)titler moneyCostaffinityChangeFN) rrr=moneyr/r0rr strr@r3r)rrDrZgainedr Z imageLinkr r r rGs     rGcCstdddddtdddd d td d d ddtdddddtdddddtdd dddtddddd td!dd"d#d$td%d d&d'd(td)dd*d+d,g }|S)-z Get all available date activity options. Returns: List of DateIdea objects representing different date activities Picnic in the Parkr-*You enjoyed a relaxing picnic in the park.z/https://lichun.app/assets/images/picnicDate.pngMovie Night at Homer%r#You had a cozy movie night at home.z.https://lichun.app/assets/images/dateNight.pngHiking Adventurer*1You embarked on an exhilarating hiking adventure.z/https://lichun.app/assets/images/hikingDate.pngFine Dining Experiencer&*You had a romantic fine dining experience.z3https://lichun.app/assets/images/fineDiningDate.pngVisit a Museum or Art Galleryr8$You visited a museum or art gallery.z/https://lichun.app/assets/images/museumDate.pngCooking Class TogetherF"You took a cooking class together.z0https://lichun.app/assets/images/cookingDate.pngAmusement Park Dayx(You had a fun day at the amusement park.z6https://lichun.app/assets/images/amusementParkDate.png Beach DayYou spent the day at the beach.z.https://lichun.app/assets/images/beachDate.pngAttend a Concert or ShowYou attended a concert or show.z0https://lichun.app/assets/images/concertDate.pngSpa Day for RelaxationYou enjoyed a relaxing spa day.z,https://lichun.app/assets/images/spaDate.png)r)Z date_ideasr r r getDateIdeas s ric CstD]}|j|kr|jj|jkr|jj|jkrt|}t dd}|j |7_ |j d|jd|jj|j8_|jj|j8_|jj d7_ d|jd}|dkrd}n|dkrd }nx|d krd }nj|d krd }n\|dkrd}nN|dkrd}n>|dkrd}n.|dkr&d}n|dkr6d}n|dkrDd}|d|jd|jd7}|d|d7}|j |td||d d!|j|jd"|jd" Sd#Sqd$S)%a Execute a date night activity with the player's partner. Checks if player has sufficient energy and money, then: - Increases relationship score - Deducts energy and money costs - Adds prestige - Creates a message event Args: player: The player object idea: The name of the date idea to execute Returns: messageEvent if successful, False if player lacks resources r%r*zYou went on a 'z' date.r+rMrOrPrQrRrTrUrVrWrXrYr[r\r_r`rbrcrerfrhz You spent z energy and z money.z& Your relationship score increased by z points. dateNightTz Date NightrFN)rirr=energyrrKrrr/r0r4r2r3prestiger@rr )rZideaZdateIdear?Z score_changer r r r rjsH          $rj)rr/uuideventsr core.modelsrcharacter.character_managerrrrrr$r7rErFrGrirjr r r r s   -"$