P1i&ddlmZddlmZddlmZddlZeZeje Z ddZ dZ ddZ dZdd Zdd ZGd d Zd Zy))get_user_model) Permission) ContentTypeNc tjj|||||dd}tjj t}t jj dd|d\}}|jj||S)a. Create an admin user with report notification permissions. This function creates a staff user with the 'can_receive_ai_report_emails' permission required to receive AI analysis report notifications. Args: phone (str): Phone number for the admin user email (str): Email address for the admin user (required for notifications) first_name (str): First name last_name (str): Last name password (str): Password for the user Returns: User: Created admin user with permissions T)phoneemail first_name last_namepasswordis_staff is_activecan_receive_ai_report_emails%Can receive AI analysis report emailsname content_typecodenamedefaults Userobjects create_userr get_for_modelr get_or_createuser_permissionsadd) rrr r r admin_useruser_content_type permissioncreateds E/home/cursorai/projects/iching/iching/tests/../../tests/test_utils.py$create_admin_with_report_permissionsr# s0))*J$++99$?%,,::/;- ;J##J/ c i}tjjd}|j|d<g|d<|D]W}|dj |j |j |j|jt|jdYtjjdd}|j|d<g|d <|D]}|jj}|d j |j |j|j|j |j"|j|jjdj%d d d lm}||d <|S)z Debug function to understand permission system issues. Returns: dict: Debug information about permissions and users r)rpermissions_foundpermission_details)idrrcontent_type_idrT)r r staff_users_countstaff_user_details)r(rrr r permissions_counthas_report_permissionr)get_report_recipients recipients)rrfiltercountappendr(rrr)strrrrallrrr r exists api.utilsr.) debug_info permissionsperm staff_usersuser user_permsr.s r"debug_permission_systemr=AsoJ$$++5S+TK&1&7&7&9J"#')J#$'(//'' II#33 1 12 1  ,,%%tt%DK&1&7&7&9J"#')J#$**..0 '(//''ZZZZ !+!1!1!3%)%:%:%A%A7&B&fh 1 046J| r$c ntjj|||||ddd}tjj t}t jj dd|d\}}t jj dd|d\} }|jj|| |S) a Create an admin user with report management permissions. This function creates a staff user with both report notification and management permissions. Args: phone (str): Phone number for the admin user email (str): Email address for the admin user first_name (str): First name last_name (str): Last name password (str): Password for the user Returns: User: Created admin user with management permissions T)rrr r r r r is_superuserrrrrcan_manage_ai_reports*Can manage and resolve AI analysis reportsr) rrr r r rrreport_permission_manage_permissions r"(create_admin_with_management_permissionsrErs0))* J$++99$?&--;;/;- <q&--;;(@- <q##$57HI r$ctjjt}i}tjj dd|d\|d<}tjj dd|d\|d<}|S) a" Set up all necessary permissions for testing. This function ensures all required permissions exist in the test database. Call this in test setUp methods if you need to ensure permissions exist. Returns: dict: Dictionary containing created permissions rrrr report_emailsr@rAmanage_reports)rrrrrr)rr8rCs r"setup_test_permissionsrIs$++99$?K'1&8&8&F&F/;- 'G'#K !(2'9'9'G'G(@- (H($K !1 r$cddlm}ddlm}m}|j j ||ddd|ddd ||rd Sd S) z Create a test BaZi person for testing purposes. Args: user (User): User to associate with the person (optional) name (str): Name for the person Returns: Person: Created BaZi person instance r)Person)datetimeiMTF)r birth_date birth_timegender created_byowner) bazi.modelsrKdatetimerLrMrcreate)r;rrKrLrMs r"create_test_bazi_personr[s]## >> a$B<d !  !& ! r$c ddlm}ddlm}|jj ||j |ddddddd S) z Create a test LiuYao for testing purposes. Args: user (User): User to associate with the LiuYao (optional) question (str): Question for the divination Returns: liuyao: Created LiuYao instance r)liuyao)timezone10000z{}) questionqdater;y1y2y3y4y5y6data) liuyao.modelsr] django.utilsr^rrZnow)r;rbr]r^s r"create_test_liuyaornsJ%% >> lln         !  r$c"eZdZdZdZdZdZy)ReportTestMixinz Mixin class for tests that need report functionality. Provides common setup methods for testing report features. cDtd|_t|_y)z5Set up admin user with report permissions for testingtest_admin@example.comrN)r#rrIr8selfs r"setUp_report_adminz"ReportTestMixin.setUp_report_admin s>* 23r$cDtd|_t|_y)z9Set up admin user with management permissions for testingtest_manager@example.comrsN)rE manager_userrIr8rts r"setUp_report_managerz$ReportTestMixin.setUp_report_managersD, 23r$ctjjddddd|_t |jd|_t |jd |_y ) z$Set up test data for reporting tests 13800000100ztest_user@example.comTestrtestpassword123)rrr r r zTest Person for Reports)r;rzTest Question for Reports)r;rbN)rrr test_userr[ test_personrn test_liuyaorts r"setUp_test_datazReportTestMixin.setUp_test_datas`11)& 2 3*  .0 r$N)__name__ __module__ __qualname____doc__rvrzrr$r"rprps 44 r$rpcdtjjddgjy)zw Clean up test permissions after testing. Call this in test tearDown methods to clean up permissions. rr@) codename__inN)rrr0deleterr$r"cleanup_test_permissionsr3s- 46MN fhr$) 13800000001rrr}Adminr~) 13800000002rxr}Managerr~)Nz Test Person)Nz Test Question)django.contrib.authrdjango.contrib.auth.modelsr"django.contrib.contenttypes.modelsrloggingr getLoggerrloggerr#r=rErIr[rnrprrr$r"rs/1:   8 $  "  1h.d  $  :z F08* * Zr$