o H@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z dd lm Z dd l mZdd lmZddlZGd d d eejeZGdddeZGdddeejeZGdddeZdS)z!This module helps with prompting.)absolute_import)division)unicode_literalsN)config)log) properties)yaml) console_io)filesc@s|eZdZdZdddZddZddZejd d Z ejd d Z d dZ ddZ e ddZe ddZejddZdS)PromptRecordBasezBase class to cache prompting results. Attributes: _cache_file_path: cache file path. dirty: bool, True if record in the cache file should be updated. Otherwise, False. last_prompt_time: Last time user was prompted. NcC||_d|_dS)NF)_cache_file_path_dirty)selfcache_file_pathrC/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/prompt_helper.py__init__-s zPromptRecordBase.__init__cCstj|jSN)ospathisfiler rrrrCacheFileExists1sz PromptRecordBase.CacheFileExistscCsR|jsdSt|j}tj||dWdn1swYd|_dS)z"Serializes data to the cache file.N)streamF)r file_utils FileWriterr rdump _ToDictionary)rfrrrSavePromptRecordToFile4s  z'PromptRecordBase.SavePromptRecordToFilecCdSrrrrrrr<zPromptRecordBase._ToDictionarycCr!rrrrrrReadPromptRecordFromFile@r"z)PromptRecordBase.ReadPromptRecordFromFilecCs|Srrrrrr __enter__DszPromptRecordBase.__enter__cCs |dSr)r )rexc_typeexc_valexc_tbrrr__exit__G zPromptRecordBase.__exit__cC|jSr)rrrrrdirtyJzPromptRecordBase.dirtycCr*r_last_prompt_timerrrrlast_prompt_timeNr,z!PromptRecordBase.last_prompt_timecCr )NT)r.r)rvaluerrrr/Rs r)__name__ __module__ __qualname____doc__rrr abcabstractmethodrr#r$r(propertyr+r/setterrrrrr #s"     r cs0eZdZdZfddZddZddZZS)OptInPromptRecordzOpt-in data usage record.cs&tt|jtjd||_dS)N)r)superr9rrPathsopt_in_prompting_cache_pathr#r.r __class__rrr[s zOptInPromptRecord.__init__cCsi}|jr |j|d<|S)Nr/r-)rresrrrr`s zOptInPromptRecord._ToDictionarycCsp|sdSz"t|j }t|}Wdn1swY|ddWSty7t dYdSw)Nr/z?Failed to parse opt-in prompt cache. Using empty cache instead.) rr FileReaderr rloadget Exceptionrdebug)rrdatarrrr#fs   z*OptInPromptRecord.ReadPromptRecordFromFile)r1r2r3r4rrr# __classcell__rrr=rr9Xs  r9c@s(eZdZejddZejddZdS) BasePromptercCr!rrrrrrPromptvr"zBasePrompter.PromptcCr!rrrrrr ShouldPromptzr"zBasePrompter.ShouldPromptN)r1r2r3r5r6rHrIrrrrrGts  rGc@s0eZdZdZdZdZddZddZdd Zd S) OptInPrompterz!Prompter to opt-in in data usage.iOaTo help improve the quality of this product, we collect anonymized usage data and anonymized stacktraces when crashes are encountered; additional information is available at . This data is handled in accordance with our privacy policy . You may choose to opt in this collection now (by choosing 'Y' at the below prompt), or at any time in the future by running the following command: gcloud config set disable_usage_reporting false cCs t|_dSr)r9recordrrrrrr)zOptInPrompter.__init__cCstsdS|js!|j }d|_Wdn1swY|rVtj|jddddd}|r;t tj j j d|j}t |_WddS1sOwYdSdS)z'Asks users to opt-in data usage report.NrzDo you want to opt-inF)message prompt_stringdefaultthrow_if_unattended cancel_on_noFalse)rIsDefaultUniverserKrr/rIr PromptContinueMESSAGEPersistPropertyVALUEScoredisable_usage_reportingtime)rpranswerrrrrHs,   "zOptInPrompter.PromptcCsBtjr tjs dS|jj}t}|r|||jkrdSdS)z Checks whether to prompt or not.FT)routisattyerrrKr/rYPROMPT_INTERVAL)rr/nowrrrrIszOptInPrompter.ShouldPromptN) r1r2r3r4r_rTrrHrIrrrrrJs  rJ)r4 __future__rrrr5rrYgooglecloudsdk.corerrrrgooglecloudsdk.core.consoler googlecloudsdk.core.utilr rsixwith_metaclassABCMetaobjectr r9rGrJrrrrs$         5