a 'Ni@s*dZddlZddlZddlZddlmZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/ZdFd3d4ZdGd5d6Zd7d8Zd9d:Zd;d<ZdHd>d?Z d@dAZ!dBdCZ"dDdEZ#dS)Ia Character Manager Module This module contains all character creation and management functions for BaoLife. Extracted from functions.py to improve code organization and maintainability. Functions in this module handle: - Character creation (players, NPCs, family members, friends, classmates, coworkers) - Character data retrieval and manipulation - Relationship management - Avatar generation - Bio and description generation N) timedeltacCs|jd|jd|jd}|jD]}|d|d7}q&|jdkrt|drb|d|jd 7}t|d rt|jd r|d |jj d7}|jd krt|dr|d|j d7}t|dr|j rt|j d r|d|j j d7}|jdkrDt|dr|j r|dd |j 7}t|drD|jrD|dd |jd7}|jdkrt|d r~t|jdr~|d|jjd7}t|dr|jdkr|d|jd7}|S) z Generate a human-readable description of a person based on their attributes. Args: person: personClass instance Returns: str: Formatted description of the person  z is a .z They are your z. rageYearsz They are years oldjobtitlez, and work as a  educationz They have a z level of educationcollegez, and graduated from (likesz. Their interests include , dislikesz, but they dislike Pincomez. They earn approximately z per month from their jobmoneyz, and have about z in savings.) firstnamelastnamesexlower relationships familiarityhasattrrrrr r r joinrrrperson descriptionir7}||_ |S)?z Generate and set avatar URL for a person using DiceBear API. Args: person: personClass instance Returns: str: Avatar URL hairNONE facial_hair accessory transparentZcircleincludeexclude)ZlongHairZ shortHairZeyepatchZhatZhijabZturband)blackZbluegrayZheatherpastelZpinkredZwhite)auburnr,blondebrownr.platinumr/r-)ZkurtZprescription01Zprescription02roundZ sunglassesZ wayfarers )mediumlightZmagesticZfancyZmagnum)r0r,r1r2r3r/)ZblazerZsweaterZshirtZhoodieZoverall)Z3c4f5cZ65c9ffZ262e33Z5199e4Z25557cZ929598Za7ffc4Zb1e2ffZe6e6e6Zff5c5cZff488eZffafb9Zffffb1Zffffff) ZbatZbearZcumbiaZdeerZdiamondZholaZpizzaZresistZskullZ skullOutline)angrydefaultZflatZraisedsadZunibrowZup)ZtannedZyellowZpaler7r2Z darkBrownr,)Z concernedr9Z disbeliefr:ZseriousZsmileZtwinkle)Z614335Zae5d29Zd08b5bZedb98aZf8d25cZfd9841Zffdbb4Black2c1b18namehex Dark Brown4a312cChestnut Brown724133Auburna55728 Golden Brownb58143Redc93305Blonded6b370 Light Greye8e1e1 Dark GreyecdcbfCopperf59797) r8Z angryNaturalr9ZdefaultNaturalZ flatNaturalZ frownNaturalZ raisedExcitedZraisedExcitedNaturalZ sadConcernedZsadConcernedNaturalZunibrowNaturalZupDown)closedZcryr9ZhappyZsideZsquintZ surprised) beardLightZ beardMajestic beardMediumZmoustacheFancymoustacheMagnumz;https://api.dicebear.com/7.x/avataaars/svg?seed={person.id}z&topProbability=0z&top=z&facialHairProbability=0z &facialHair=z&accessoryProbability=0z &accessory=z &hairColor=z &clothing=z&clothesColor=z&clothingGraphic=z &skinColor=z&mouth=z &eyes=defaultz&eyebrows=default) ravatar_settingsr$r&r' hair_colorZclothingrandomchoice skin_colorZmouthZimageURL)rstylemodetopZ topChanceZhatColorZ hairColor accessoriesZaccessoriesChanceZ facialHairZfacialHairChanceZfacialHairColorZclothesZ clothesColorZclothingGraphicZeyebrowZskinZ mouthOptionsZ skinColors haircolorsZeyebrowsZeyesZ facial_hairsbase_urlrrr set_avatarsst       rac sDddddddddddd dd d dd d dddddddddddddg fdd}t||jjdvr|d|d|d|dg}nX|jjdkr|d|d|dg}n4|jjdvr|d|dg}n|d|d |d g}|jdkr:|d|vr||dgd|d|vr:||dgdt|S)z Get appropriate hair color based on person's age and skin tone. Args: person: personClass instance Returns: str: Hex color code for hair r;r<r=r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQcs&D]}|d|kr|dSqdS)Nr>r?r)Z color_nameZcolorr_rr get_hexs zget_hair_color..get_hex)ZPALEZLIGHTZBROWN)Z DARK_BROWNZBLACK2)printrVrZrextendrXrY)rrcZ hair_colorsrrbr get_hair_colors4      rhcCs^dg}|jdkrd}n:|jdkr4|jdkr4ddg}n|jdkrHgd}ngd}t|}|S)z Get appropriate hair type based on person's age and sex. Args: person: personClass instance Returns: str: Hair type identifier r%rdMalesides)ZdreadsZdreads01Zdreads02froZshaggyZ shaggyMulletZ shortFlatZ theCaesarZtheCaesarAndSidePartZ shavedSidesrkZ shortWavedZfrizzleZ shortRoundZ shortCurly) ZbigHairZbobZbunZcurlyrlZcurvyZfroBandZlongButNotTooLongZ miaWallaceZ straight01Z straight02ZstraightAndStrand)rrrXrY)rZ hair_typesZ hair_typerrr get_hair_type s      rmcCs6|jdkr.|jdkr.|jdkr.tgd}nd}|S)z Get appropriate facial hair based on person's age and sex. Args: person: personClass instance Returns: str: Facial hair type identifier rjrd)rSrTrUr%)rrrXrY)rr&rrr get_facial_hair$s rocCs`gd}|jdkrgd}n,d|jkr2dkr@nn gd}ngd}tj||ddd }|S) z Get appropriate accessory (glasses) based on person's age. Args: person: personClass instance Returns: str: Accessory type identifier )r%r4r4r4rn)g?x&?rprp')g333333?/$?rrrr)g?皙?rsrsri)weightskr)rrXchoices)rr^Zprobsr'rrr get_accessory6s    rwcCsddl}|jdddkr0gd}t|Sddlm}|}zJ|}| d| }|dd W|rz| | S| n|r| | 0dS) zd Get a random last name from the database. Returns: str: Capitalized last name rNBAOLIFE_TEST_MODEfalsetrue) ZSmithZJohnsonZWilliamsZBrownZJonesZGarciaZMillerZDavisZ RodriguezZMartinezget_database_connectionzWSELECT name FROM lastnames_race WHERE lastnames_race.rank < 500 order by RAND() limit 1 osenvirongetrrXrYdatabase.db_operationsr|cursorZexecuteZfetchall capitalizeclose)r~Ztest_lastnamesr|mydbmycursormyresultrrr get_lastnameVs*    rc Csddl}|jdddkrB|dkr0gd}ngd}t|Sdd lm}|}zb|}d }|dkrnd }d }| ||f| }|dd W|r| | S| n|r| | 0dS) z Get a random first name from the database based on gender. Args: gender: "Male" or "Female" Returns: str: Capitalized first name rNrxryrzFemale) ZEmmaZOliviaZAvaZIsabellaZSophiaZMiaZ CharlotteZAmeliaZHarperZEvelyn) ZLiamZNoahZOliverZElijahZJamesZWilliamZBenjaminZLucasZHenryZ Alexanderr{MFzWSELECT Name FROM firstnames WHERE Cnt > 1000000 AND Gender = %s ORDER BY RAND() LIMIT 1r}) Zgenderr~Z test_namesr|rrZgenderSelectorZsqlrrrr get_firstnamers6     rcCsD|jr@ddlm}m}||jd}|||jd}|d|_dS)z Set a person's birthday based on their age in days and current date. Args: person: personClass instance player: playerClass instance r)datetimerz%m-%d)daysN)ageDaysrrstrptimedatestrftimebirthday)rplayerrrZ current_daterrrr setBirthdays rcCs0ddlm}|jd||jjdkrPd|jd_|jdd|jd_n&d|jd_|jdd|jd_|jdj dt |jdj|jd_ t |jd_ |jjtdd d |jd_|jdjd tdd |jd_|jdj|jd_t||jd|jd<|jdj|j_|S) z Create a partner (boyfriend/girlfriend) for the player character. Args: player: playerClass instance Returns: playerClass: Updated player object r personClassrjrZ GirlfriendZ Boyfriendpartnerrim) core.modelsrrinsertcr setPronounrrappendrrrrrrXrandintrr setValuesidrrrrrr get_partners$      $rcCsddlm}|jd||jdjd|jdjd|jdjdkr^d|jd_n d|jd_d|jd_t |jdj|jd_ |j j |jd_ d|jd_ d|jd_t||jd|jd<|jdS) z Create a child for the player character. Args: player: playerClass instance Returns: personClass: The newly created child rrchildfamilyrjZSonZDaughterri)rrrrrrrr familyLevelrrrrrrrrrrr add_childs     rcCs<d}|jD],}d|jvr |s"|}q |r |j|jkr |}q |S)z Get the youngest child from player's relationships. Args: player: playerClass instance Returns: personClass or False: The youngest child or False if no children Fr)rrr)rZyoungestrrrr get_youngestChilds   rcCsddlm}m}ddlm}m}|}d|_||j d|j dd|_ d|_ t |j|_ |jj|_|jjtd d |_|jd tdd |_|jjj|j_t||}|jd||}d |_||j d |j dd|_ d|_ t |j|_ |jj|_|jjtd d |_|jd tdd |_|jjj|j_t||}|jd|||j|j|dd}|j|_|j|_|ddd|_|ddd|_|j ||S)z Create mother and father for the player character. Args: player: playerClass instance Returns: playerClass: Updated player object rrrelationshipClass)generate_random_daterandrjfatherrFatherrirn"rrmotherMotherMarriedHappily married for years.rd)rrr utils.helpersrrrrrrrrrrrrrrXrrrVrZrrrr relationshipaffinityrelData)rrrrrrrZparents_relationshiprrr add_parentssJ            rcCsddlm}tdtdddkrtd|}|jd|jd|jdkr^d |_nd |_d|_ t |j|_ | |j j|_|j jtdd |_|jd d tdd |_|j jj|j_t||}|jd|tdddkrtd|}|jd|jd|jdkr,d |_nd |_d|_ | t |j|_ |j j|_|jdjtdd |_|jd d tdd |_|j jj|j_t||}|jd||S)z Create older siblings for the player character (with some randomness). Args: player: playerClass instance Returns: playerClass: Updated player object rrzadding older siblingsrizadding older siblingsiblingrrjZBrotherZSisterrr zadding another older sibling)rrrfrXrrrrrrrrrrrrrrVrZrrr)rrrrrr add_older_siblings4sJ             rc Csddlm}m}ddlm}|}d|_||jd|jdd|_ d|_ t |j|_ |j j|_t|d jtd d |_|jd td d |_|j jj|j_t||}|jd||}d|_||jd|jdd|_ d|_ t |j|_ |j j|_t|d jtd d |_|jd td d |_|j jj|j_t||}|jd|||j|j|dd}|j|_|j|_|j||}d|_||jd|jdd|_ d|_ t |j|_ t|_t|djtd d |_|jd td d |_|j jj|j_t||}|jd||}d|_||jd|jdd|_ d|_ t |j|_ |jd j|_t|djtd d |_|jd td d |_|j jj|j_t||}|jd|||j|j|dd}|j|_|j|_|j||S)z Create all four grandparents for the player character. Args: player: playerClass instance Returns: playerClass: Updated player object rr)rrZpaternal_grandmotherrZ GrandmotherrerrnrrrirjZpaternal_grandfatherZ GrandfatherrrZmaternal_grandfatherrZmaternal_grandmother)rrrrrrrrrrrrrrrget_relationshiprrXrrrVrZrrrrrrr) rrrrZ grandmother1Z grandfather1Zg_relZ grandfather2Z grandmother2rrr add_grandparentsls                      rcCsddlm}|}||_||jdd|_tdd|_ t |j|_ t |_ |dkrt|jjtdd |_n^|d kr|jjtdd d |_n:|d kr|jjtdd d |_n|jjtdd |_t||}|jd||S) z Create a friend for the player character. Args: player: playerClass instance sex: "Male" or "Female" age: "similar", "younger", "older", or other Returns: playerClass: Updated player object rrfriendFriendFr+similarrirZyoungerrZolder)rrrrrrrrXrrrrrrrrrrr)rragerrrrr add_friends&    rcCsg}tdt|jD]$}d|j|jvr||j|qt|dkrt|tdt|jD]$}d|j|jvr^||j|q^|tdt|dS)z Get a random classmate from player's relationships. Creates classmates if none exist. Args: player: playerClass instance Returns: personClass: A random classmate r classmateri)rangelenrrrcreate_classmatesrXr)rZ classmatesrrrr get_random_classmates  rcCsbg}tdt|jD]$}d|j|jvr||j|qt|dkrJdS|tdt|dS)z Get a random friend from player's relationships. Args: player: playerClass instance Returns: personClass or False: A random friend or False if no friends rrFri)rrrrrrXr)rfriendsrrrr get_random_friends  rcCs|jtdt|jdS)z Get a random character from player's relationships. Args: player: playerClass instance Returns: personClass: A random character rri)rrXrr)rrrr get_random_characters rcCsbg}tdt|jD]$}|j|jdkr||j|qt|dkrJdS|tdt|dS)z Get a random immediate family member (familyLevel == 1). Args: player: playerClass instance Returns: personClass: A random family member or None if no family rriN)rrrrrrXrrrrrrr get_random_family%s  rcCs>g}tdt|jD]$}|j|jdkr||j|q|S)z Get all immediate family members (familyLevel == 1). Args: player: playerClass instance Returns: list: List of all immediate family members rri)rrrrrrrrr get_allFamily8s rrXrrcCsddlm}|}|dkr0|j|||_|dkr>||_|t|j|_ t |_ |dkrz|j j tdd|_ n6|dkrdtdd |_ n|d krd tdd |_ |j d tdd|_t||}|dkr|S|jd||jdSdS)aW Create a generic character with customizable attributes. Args: player: playerClass instance sex: "Male", "Female", or "random" age: "similar", "random_adults", "random_teens_adults", or other relationship: Relationship type or "none" Returns: personClass: The newly created character rrnonerXrri random_adultsrnr random_teens_adults -rrN)rrrrrrrrrrrrrrrXrrrrr)rrrrrZ new_characterrrr create_characterMs,     rcCs\ddlm}|dkrXtddD]}t|ddqt|ddd }t||jd d id|_dS) z Create 10 classmates and 1 teacher for the player. Args: player: playerClass instance activity: Optional activity parameter (unused currently) r)find_where_testNr5r)rZteacherrrrrzElementary School Teacher)rrrrlist occupationsr)ractivityrrrrrr r~s  rcCsNddlm}td|ddD]}t|ddd}||_qt|dd d}||_d S) z Create coworkers and a boss for the player. Args: player: playerClass instance occupation: Job occupation object rrZcoworkerrrZbossrN)rrrrr)r occupationrrrrrr create_coworkerss  rcstfdd|jDdS)z Get a person by their ID from player's relationships. Args: player: playerClass instance id: Person's ID Returns: personClass or None: The person if found, None otherwise c3s|]}|jkr|VqdSNr.0itemrrr zget_person..Nnextr)rrrrr get_persons rcstfdd|jDdS)z Get a person by their relationship type. Args: player: playerClass instance type: Relationship type (e.g., "father", "mother", "friend") Returns: personClass or None: The person if found, None otherwise c3s|]}|jvr|VqdSr)rrtyperr rrz#get_relationship..Nr)rrrrr rs rFcCsVt|jD]F\}}t|jvr |j|j||j|j||r ||j|_q |S)aC Update a relationship type for all matching characters. Args: player: playerClass instance oldRelationship: Current relationship type newRelationship: New relationship type title: Optional new title for the relationship Returns: playerClass: Updated player object ) enumeraterrrremoverr)rZoldRelationshipZnewRelationshiprindexrrrr update_relationships  rcCsddlm}m}m}m}ddlm}d}d}|jjdkrrrirrrr+rjZHeZShehomezGhttps://cdn.midjourney.com/fe048fc6-f534-4ffe-9fb2-02297a3e1d2d/0_2.pngactiverrZDatingzYou are dating!Z setupComplete)#rrrrrsplitrrrrintrrrrZpronounrarrrlrrrrrr controllerrrrrrrstatus)rdatarrrZ name_partsrelrrr characterSetup%sD         r)rXrr)N)F)$__doc__rrXuuidrr!r#rarhrmrorwrrrrrrrrrrrrrrrrrrrrrrrrrrrr sF )0d4 * 68['  1  $