soj>ddlZddlmZddlTddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZGd d eZedS) N)Any)*) BaseModeldb_obj) GameStatus)CrewChestSpendingReason)CrewLevelUpgradeType)CrewRole)DevilFruitAbilityTypeceZdZUdZeejZe eze d<e dZ e e ze d<e dZe e ze d<edZejeze d <ed Zeeze d <ed Zeeze d <ed Zeeze d <edZeeze d<eejZeeze d<e dZe e d<e dZe e d<eejZeeze d<eejdZ e eze d<ed Z!eeze d<e dZ"e e ze d<e dZ#e e ze d<e dZ$e e ze d<e dZ%e e ze d<e dZ&e e ze d<e dZ'e e ze d<dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/Gdd Z0d!efd"Z1d!efd#Z2dLd$ed%ed!e3fd&Z4d!e3fd'Z5e6de d!dfd(Z7e6d)ed!dfd*Z8d!e9fd+Z:d!e9fd,Z;d!e fd-Zd0e?d!efd1Z@d0e?d!e fd2ZAd!e fd3ZBd!e3fd4ZCd!e3eDfd5ZEd6eDd!e3fd7ZFd!eGfd8ZHd9eGd!efd:ZId;ZJe6dMd<ZKd!e fd=ZLd!e fd>ZMd!e fd?ZNdMd@ZOe6dMdAZPd!e9fdBZQd!e9fdCZRd!e9fdDZSd!e9fdEZTdNdFe fdGZUd!e fdHZVd!e fdIZWdNdJe fdKZXdS)OCrewz Crew class ) max_lengthnameT)defaultcan_accept_new_members is_active)null disband_dater chest_amounttotal_gained_chest_amountfrom_reset_chest_amountlevel max_abilitiescan_promote_first_matecan_promote_captain max_members)rr descriptionrequired_bountyallow_view_in_searchallow_join_from_searchFis_fullallow_davy_back_fight_requestauto_accept_davy_back_fightauto_accept_joinNceZdZdZdS) Crew.MetacrewN)__name__ __module__ __qualname__db_table/app/src/model/Crew.pyMetar'5sr.r0returncddlm}|jt|jS)z\ Returns the average bounty of the crew :return: The average bounty rUser)src.model.Userr4 crew_membersselectfnAvgbountyscalarselfr4s r/get_average_bountyzCrew.get_average_bounty8sG (''''' ''t{(;(;<<CCEEEr.cddlm}|jt|jS)zl Returns the average location level of the crew :return: The average location level rr3)r5r4r6r7r8r9location_levelr;r<s r/get_average_location_levelzCrew.get_average_location_levelBsH (''''' ''t/B(C(CDDKKMMMr.limitpagecFddlm}|W|j|j||S|j|jS)z Returns the crew members ordered by join date :param limit: The number of members to return. If None, returns all members :param page: The page to return. If None, returns the first page :return: The crew member rr3)r5r4r6r7order_bycrew_join_dateascpaginate)r=rBrCr4s r/ get_memberszCrew.get_membersLs ('''''  !((**$-113344$&&   ''))2243F3J3J3L3LMMMr.cJt|dS)zp Returns the crew members ordered by Davy Back Fight priority :return: The crew members c.|jdu|j|jfS)N)crew_davy_back_fight_priorityrF)xs r/zDCrew.get_members_order_by_davy_back_fight_priority..hs /47/ r.)key)sortedrIr=s r/-get_members_order_by_davy_back_fight_priorityz2Crew.get_members_order_by_davy_back_fight_priority`s4          r.crttj|ztjdkzS)zt Returns the crew by name, case-insensitive :param name: The name :return: The crew T)r get_or_nonerr)rs r/get_by_name_if_activezCrew.get_by_name_if_activeos*DT^t5K LMMMr.crew_idctttj|ktjdkzS)z{ Returns the Crew if it exists and is active :param crew_id: The crew id :return: The crew T)r rTidr)rVs r/ logical_getzCrew.logical_getys+G!3$8N OPPPr.cddlm}|j|jt jkS)zL Returns the crew captain :return: The crew captain rr3) r5r4r6r7where crew_roler CAPTAINgetr<s r/ get_captainzCrew.get_captainsO (''''' ''))//(BR0RSSWWYYYr.cddlm}|j|jt jkS)zR Returns the crew first mate :return: The crew first mate rr3) r5r4r6r7r[r\r FIRST_MATErTr<s r/get_first_matezCrew.get_first_matesS ('''''   $ $ & & , ,T^x?R-R S S _ _ a a r.c.|duS)zn Returns True if the crew has a first mate :return: True if the crew has a first mate N)rbrQs r/has_first_matezCrew.has_first_mates ""$$D00r.cX|jS)z` Returns the number of crew members :return: The number of crew members )r6r7countrQs r/get_member_countzCrew.get_member_counts%  ''))//111r.cddlm}|j|jS)zd Returns the crew members ordered by contribution :return: The crew members r)CrewMemberChestContribution)%src.model.CrewMemberChestContributionricrew_member_chest_contributionsr7rEamountdesc)r=ris r/get_chest_contributionszCrew.get_chest_contributionssP VUUUUU3::<<EE ' . 3 3 5 5   r.reasonc|tjur&tj|jzStjd|jdz zzS)zz Returns the next powerup price :param reason: The reason :return: The next powerup price r)rLEVEL_UPEnvCREW_POWERUP_BASE_PRICEget_intr)r=ros r/get_powerup_pricezCrew.get_powerup_pricesV 09 9 9.66884:E E*2244dj1n8MNNr.cJddlm}|||S)z Returns the next powerup price formatted :param reason: The reason :return: The next powerup price formatted rget_belly_formatted)src.utils.string_utilsryrv)r=rorys r/get_powerup_price_formattedz Crew.get_powerup_price_formatteds5 ?>>>>>""4#9#9&#A#ABBBr.c.ddlm}||jS)z\ Returns the crew chest formatted :return: The crew chest formatted rrx)rzryrr=rys r/get_crew_chest_formattedzCrew.get_crew_chest_formatteds* ?>>>>>""4#4555r.cddlm}|j|jt jkS)z^ Returns the crew active abilities :return: The crew active abilities r CrewAbility)src.model.CrewAbilityrcrew_abilitiesr7r[expiration_datedatetimenow)r=rs r/get_active_abilitieszCrew.get_active_abilitiessW 655555"))++11  '(*;*?*?*A*A A   r.crttd|DS)zj Returns the crew active abilities types :return: The crew active abilities types c6g|]}|Sr-)get_ability_type).0 crew_abilitys r/ z3Crew.get_active_abilities_types..s$aaa\..00aaar.)listsetrrQs r/get_active_abilities_typeszCrew.get_active_abilities_typess<  aaTE^E^E`E`aaa b b   r. ability_typecddlm}|j|jt jk|j|j kzS)z Returns the crew active ability of the given type :param ability_type: The ability type :return: The crew active ability rr) rrrr7r[rrrrvalue)r=rrs r/get_active_abilityzCrew.get_active_abilitysj 655555"))++11  (8+<+@+@+B+B B'<+== ?   r.c:tj|jdzS)zb Returns the next level upgrade type :return: The next level upgrade type r)r get_by_levelrrQs r/get_next_level_upgrade_typez Crew.get_next_level_upgrade_types $0a@@@r. upgrade_typec:|tjur|jS|jS)z Returns the number of upgrades of the given type :param upgrade_type: The upgrade type :return: The number of upgrades )r MEMBERrrr=rs r/get_upgrade_type_countzCrew.get_upgrade_type_counts% /6 6 6# #!!r.c&|}|tjur,|xjdz c_d|_|n|tjur|xjdz c_|xjdz c_| dS)z9 Level up the crew :return: None rTN) rr rrr set_is_fullABILITYrrsavers r/level_upz Crew.level_ups 7799 /6 6 6    !  *.D '        19 9 9   ! #   a  r.c tjdz tjktjftjdz dkdfg}t d|tjdz }t|ttjt tjdz dz ztj t tjdz dz zt dS)z< Reset the crew level :return: None rqrN)r)rr)r rrsCREW_MAX_LEVEL_AFTER_RESETruCaseupdateexecuteCREW_MAX_MEMBERSr8ROUNDCREW_MAX_ABILITIESFLOORset_is_full_for_all_crews) conditions case_stmts r/ reset_levelzCrew.reset_level#s,a3#A#I#I#K#KK.6688  j1n !1 %. z4:>::  ) $$,,... -5577"((DJQRNVWCW:X:XX199;;bhh UVZ[G[>\>\\    '))) &&(((((r.c.ddlm}||jS)zV Returns the crew name escaped :return: The crew name escaped rescape_valid_markdown_chars)src.service.message_servicerrr=rs r/get_name_escapedzCrew.get_name_escapedBs* LKKKKK**49555r.c.ddlm}||jS)zd Returns the crew description escaped :return: The crew description escaped rr)rrrrs r/get_description_escapedzCrew.get_description_escapedLs+ LKKKKK**4+;<<>>>>""4#7888r.c||jkrd|_nd|_||dS)z< Set the crew as full :return: None TFN)rgrr"rset_davy_back_fight_priorityrQs r/rzCrew.set_is_full`sU  " "d&6 6 6DLL DL  ))+++++r.cVd}tj|dS)zJ Set the crew as full for all crews :return: None zxupdate crew set is_full = IF(((select count(*) from user where crew_id = crew.id) < crew.max_members), 0, 1) where true;N)rget_db execute_sql) raw_querys r/rzCrew.set_is_full_for_all_crewsos.    ##I.....r.cddlm}||j|k|j|kz|jtjk|jtj kzz S)zb Returns the current Davy Back Fight :return: The current Davy Back Fight r DavyBackFight) src.model.DavyBackFightrr7r[challenger_crew opponent_crewstatusrCOUNTDOWN_TO_START IN_PROGRESSrTr=rs r/get_active_davy_back_fightzCrew.get_active_davy_back_fight~s :99999  " " U/47M(/:?BD'4<(/:>AC !1H4E4I4I4K4KK M  []] r. add_levelc(ddlm}|||S)z Returns the crew name with deeplink :param add_level: Whether to add the level to the name :return: The crew name with deeplink r)get_crew_name_with_deeplink)r)src.service.crew_servicer)r=rrs r/get_name_with_deeplinkzCrew.get_name_with_deeplinks- IHHHHH**49EEEEr.c.|duS)z Returns True if the crew has an active Davy Back Fight :return: True if the crew has an active Davy Back Fight N)rrQs r/has_active_davy_back_fightzCrew.has_active_davy_back_fights ..00<rArrIrR staticmethodrUrYrr_rbrdrgrnrrvr{r~rr rrr rrrrrrrrrrrrrrrrrr-r.r/r r s&I1I1Q1Q1S1STTTD# /TTT2>,t2L2L2LD</LLL%1\$%?%?%?Itl"???6Cm6N6N6NL(#m3NNN*9/!*D*D*DL#'DDD7Fq7Q7Q7Qs_4QQQ5D_Q5O5O5OS?2OOO , Q 7 7 7E3 777(4 S=S=[=[=]=](^(^(^M3%^^^#/<#=#=#=D=== , T : : ::::&2l3;O;W;W;Y;Y&Z&Z&ZK|#ZZZ#,92::<<4$$$Ky.=_Q-G-G-GOS?*GGG0< T0J0J0J$-JJJ2>,t2L2L2LD</LLL#/<#>#>#>GTL >>>9Ed9S9S9S!4,#6SSS7C|E7R7R7R !4RRR,8L,F,F,Fd\)FFFN"&&*##' #' !%LFCFFFFNCNNNNNNN3NtNNNN(  t     NCNFNNN\NQSQVQQQ\QZSZZZZ       111112#2222        O(? OC O O O O C2I Cc C C C C6#6666  d      D1F,G      /D       A-AAAAA "3G "C " " " "&)))\)<6#6666=====9s9999 , , , , / / /\ / C    &     "  S      S    4 F F F F F F=D====>T>>>>r.r )rtypingrpeeweeresources.Environment Environmentrssrc.model.BaseModelrrsrc.model.enums.GameStatusr,src.model.enums.crew.CrewChestSpendingReasonr)src.model.enums.crew.CrewLevelUpgradeTyper src.model.enums.crew.CrewRoler 1src.model.enums.devil_fruit.DevilFruitAbilityTyper r create_tabler-r.r/rs######11111111111111PPPPPPJJJJJJ222222SSSSSShhhhh9hhhVr.