U Ih<@sdddlmZddlmZmZddlmZmZddlZddl m Z ddl Z e e Z GdddZdS)) MongoClient)ConnectionFailureServerSelectionTimeoutError) MONGODB_URI MONGODB_DBN)datetimec@s8eZdZddZd ddZddZdd Zdd d ZdS) MongoHandlerc Csztk r}ztd.t|WYdSd}~XYnXdS)/Nrz No document found with url_key: Fz/Updating status for test case with identifier 'z' in document ./$setzstatus.Tz-Updated central status dictionary for title: rz Shared view update detected for Titlez"Found shared view match by title: z test_data.z.StatuszNo test case found with title 'z' in shared view document r _rzExtracted UI identifier: titleContentcontentzFound match in title: ztest_data.test_cases.z.statusz/Successfully updated status by title match for zFound match for UI identifier z in title: z7Successfully updated status by UI identifier match for zFound match in contentz1Successfully updated status by content match for test_case_idz Test Case IDzFound direct ID match at index zNo test case found matching 'z Document z has no test casesz!Error updating test case status: )r find_oner rr update_one isinstancelist enumerategetwarningsplitlenZmodified_countrr)rrr/rdocZ title_foundis_shared_viewr foundidxtcr,resultZ ui_identifierpartsr.rrrrupdate_test_case_status*s              z$MongoHandler.update_test_case_statusc Csnz*|jd|i}|s&td||WStk rh}z tdt|tdW5d}~XYnXdS)z"Retrieve test case data by URL keyr No test case found for URL key: zError retrieving test case: z*Failed to retrieve test case from databaseN)r r0r r6rrr)rrr>rrrr get_test_caseszMongoHandler.get_test_caseFc Csz|jd|i}|s.td|WdSd|krLtd|dn tdd|krt|dtrt|dD]B\}}|dd }|d d }|rxtd |d |d |dqxn|d|kr:d|dkr:t|ddD]R\}}|d|dd }|d |dd }|rtd|d |d |dqd|krx|drx|sxtdt |dd|dWSi}d|krt|dtrtd|dD]6}d|kr|dd }|d d }|r|||<qnld|krNd|dkrNtd|ddD]<}|d|dd }|d |dd }|r|||<q|rtdt |d||j d|idd|iitdt |d||WSt k r} zt dt | WYdSd} ~ XYnXdS)zRetrieve all status values for test cases in a document Args: url_key: The unique URL key for the document force_refresh: If True, forces a direct database query to get fresh data rrANrzSTATUS DICT in MongoDB: z"NO STATUS DICT in MongoDB documentrr&r'StatuszSHARED VIEW TC[z ]: Title='z ', Status=''r r,z MAIN VIEW TC[zFound z# status values in status dictionaryz.Building status values from shared view formatz,Building status values from main view formatz%UPDATING status dict in MongoDB with z values: r%z Returning z status values for z*Error retrieving test case status values: )r r0r r6rr2r3r4r5r8r1rrr) rr force_refreshr>ir=r,r status_valuesrrrrget_test_case_status_valuessd   "          z(MongoHandler.get_test_case_status_values)N)F)__name__ __module__ __qualname__rr"r@rBrHrrrrr s  ) r)ZpymongorZpymongo.errorsrrconfig.settingsrrrrlogging getLoggerrIr rrrrrs