o 1h$@sFddlZddlZddlmZmZmZmZddlmZGdddZ dS)N)DictListAnyOptional)Pathc @sxeZdZd&defddZdefddZdd Zd eee fded ed e fd dZ d eded e eee ffddZ ded eeee ffddZd edeee fded e fddZd eded e eee ffddZd eded e fddZd eded e fddZd eded e fddZdeee fded e fdd Zd eded e eee ffd!d"Zded eeee ffd#d$Zd%S)' JSONHandlerstorage base_pathcCst||_|dS)N)rr _create_directories)selfr r D/var/www/eduai.edurigo.com/roleplay/production/utils/json_handler.py__init__s  zJSONHandler.__init__is_admincCs2|dkrdnd}|j|}|d|d|ddS)z$Get storage paths based on user typeadminlearner scenarios conversations assessments)rrr)r )r r user_type user_pathr r r _get_storage_paths s  zJSONHandler._get_storage_pathscCs4dD]}||}|D] }|jdddq qdS)z5Create storage directories for both admin and learner)rrT)parentsexist_okN)rvaluesmkdir)r rpathspathr r r r s   zJSONHandler._create_directories scenario_data session_idreturnc CszKtd|dt|d||}d|d}td|d|d|}t|d d d }tj||d d dWdWdS1sDwYWdStye}ztd|WYd}~d Sd}~ww)z+Save scenario to JSON file using session_idz-DEBUG JSONHandler.save_scenario: session_id=' ' (type: ) scenario_.jsonz+DEBUG JSONHandler.save_scenario: filename=''rwutf-8encodingFindent ensure_asciiNTzError saving scenario: )printtyperopenjsondump Exception) r rrr rfilenamefilepathfer r r save_scenarios"    zJSONHandler.save_scenarioc Cz8||}d|d}|d|}|sWdSt|ddd}t|WdWS1s1wYWdStyR}ztd|WYd}~dSd}~ww) z-Load scenario from JSON file using session_idr$r%rNrr(r)zError loading scenario: rexistsr1r2loadr4r/r r rrr5r6r7r8r r r load_scenario/   (zJSONHandler.load_scenarioc Csg}zF||}|ddD]7}t|ddd&}t|}||d|jdd|d |d |d d Wd n1sAwYqWnt ya}z t d|WYd }~nd }~wwt |ddddS)z*List all available scenarios for user typerzscenario_*.jsonr;r(r)r r$category objective created_at)r rCrDrENzError listing scenarios: cS|dSNrEr xr r r Pz,JSONHandler.list_scenarios..Tkeyreverse) rglobr1r2r>appendgetstemreplacer4r/sorted)r rrrr6r7scenarior8r r r list_scenarios?s(    zJSONHandler.list_scenariosconversation_datac Csz]td|dt|d||}d|d}td|d|d|}td |dtd |dt|d d d }tj||dddWdWdS1sVwYWdStyw}ztd|WYd}~dSd}~ww)zSave conversation to JSON filez1DEBUG JSONHandler.save_conversation: session_id='r"r# conversation_r%z/DEBUG JSONHandler.save_conversation: filename='r&rz4DEBUG JSONHandler.save_conversation: full_filepath='z4DEBUG JSONHandler.save_conversation: absolute_path='r'r(r)r+Fr,NTzError saving conversation: )r/r0rabsoluter1r2r3r4) r r rWrrr5r6r7r8r r r save_conversationRs&    zJSONHandler.save_conversationc Cr:) z Load conversation from JSON filerXr%rNr;r(r)zError loading conversation: r<r?r r r load_conversationdrAzJSONHandler.load_conversationc Cz-||}d|d}|d|}|r$|td|WdStd|WdStyG}ztd|WYd }~dSd }~ww) zDelete conversation filerXr%rz"DEBUG: Conversation file deleted: Tz$DEBUG: Conversation file not found: FzError deleting conversation: Nrr=unlinkr/r4r r rrr5r6r8r r r delete_conversationt   zJSONHandler.delete_conversationc Cr\) zDelete scenario filer$r%rzDEBUG: Scenario file deleted: Tz DEBUG: Scenario file not found: FzError deleting scenario: Nr]r_r r r delete_scenariorazJSONHandler.delete_scenarioc Cr\) zDelete assessment file assessment_r%rz DEBUG: Assessment file deleted: Tz"DEBUG: Assessment file not found: FzError deleting assessment: Nr]r_r r r delete_assessmentrazJSONHandler.delete_assessmentassessment_datac CszTtd|dd|||}d|dd}td|d|d|}td |dt|d d d }tj||d ddWdWdS1sMwYWdStyn}ztd|WYd}~dSd}~ww)zSave assessment to JSON filez/DEBUG JSONHandler.save_assessment: session_id='r z ', is_admin=rcr%z-DEBUG JSONHandler.save_assessment: filename='r&rz-DEBUG JSONHandler.save_assessment: filepath='r'r(r)r+Fr,NTzError saving assessment: )r/rr1r2r3r4)r rerrr5r6r7r8r r r save_assessments$   zJSONHandler.save_assessmentc Cr:) zLoad assessment from JSON filercr%rNr;r(r)zError loading assessment: r<r?r r r load_assessmentrAzJSONHandler.load_assessmentc Csg}zF||}|ddD]7}t|ddd&}t|}||d|d|d|d|d |d d Wd n1sAwYqWntya}z td |WYd }~nd }~wwt |ddddS)z4List all assessments with summary info for user typerzassessment_*.jsonr;r(r)r scenario_id overall_scoreperformance_levelrE)r rhrirjrENzError listing assessments: cSrFrGr rHr r r rJrKz.JSONHandler.list_assessments..TrL) rrOr1r2r>rPrQr4r/rT)r rrrr6r7 assessmentr8r r r list_assessmentss*    zJSONHandler.list_assessmentsN)r)__name__ __module__ __qualname__strrintrr rrboolr9rr@rrVrZr[r`rbrdrfrgrlr r r r rs """"""r) r2ostypingrrrrpathlibrrr r r r s