`9Sc@sddlZddlZddlZddlTddlTddlTddlmZddlmZddl TdgZ de fdYZ dS(iN(t*(tgui(ttxttHttpRequestDialogcBseZdZddZdZdZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZRS(sDialog that makes it possible to download a file from the Internet using either GET or POST HTTP request. If 'setDestinationFilePath' is called prior to 'exec_', a temporary file is created during the download with name obtained by adding '.tmp' to the name of the destination file. If the download completes successfully, temporary file is renamed to the destination one, possibly overwriting the target. If the download fails or is aborted by the user, temporary file is deleted. If 'setDestinationFilePath' is not called, then no files are created during the download, and 'getDownloadData' should be used to obtain contents of the downloaded file instead.c Cs=tj||t||_||_d|_d|_d|_d|_ t ||_ d|_ t }t|_|jj|jj||j|jt|jjtjtjB}td||_|jjt|j|jt|_|jjt|j|jt|_|jj|jt j!|j|j"t#d|_$|j$j%t|j$j&j'|j(t)}|j"|j|j$|j*||j+||j,||j-t j.d|j/|j0t1j2@|j3j'|j4t5j6d|j7dS(stUse 'requestData' different from 'None' to send a POST request, and the default value to send a GET request.ts(%s)...tCanceltworldiN(8tQDialogt__init__tQUrlturlt requestDatatNonetdestinationFilePatht tmpFilePathttmpFiletdownloadedDatatQNetworkAccessManagertnetworkAccessManagert networkReplyt QVBoxLayouttStatusLabelWidgett statusLabeltdisableWordWrappingt setProgresst addWidgettunicodettoStringtRemoveUserInfot RemoveQuerytQLabelturlLabelt setWordWraptFalset progressLabelt QProgressBart progressBarthidetaddItemRtUnhidableWidgetItemt addStretcht QPushButtont cancelButtontsetAutoDefaulttclickedtconnecttrejectt QHBoxLayoutt addLayoutt setLayouttsetWindowTitlet setWindowIcontloadIcontsetWindowFlagst windowFlagstQttWindowContextHelpButtonHinttrejectedt onRejectedtQTimert singleShottonInitComplete( tselftparentt windowTitletprogressMessageR R tlayoutt urlStringtbuttonBoxLayout((s code\common\HttpRequestDialog.pyR1sP                   cCs8|jdkst||_|d|_d|_dS(sCall this prior to 'exec_' to instruct the dialog to store downloaded data in the given file on disk instead of a local variable.s.tmpN(RR tAssertionErrorR RR(R>R ((s code\common\HttpRequestDialog.pytsetDestinationFilePaths  cCs|jS(sReturn an 'str' object holding the contents of the downloaded data, or 'None' if the data have been stored in a file on disk instead of a local variable.(R(R>((s code\common\HttpRequestDialog.pytgetDownloadedDatascCs6|jdk r(|jjd|_n|jdS(N(RR t deleteLatertaccept(R>tevent((s code\common\HttpRequestDialog.pyt closeEvents  cCs|jdk r(|jjd|_n|jdk rCd|_ntj|j|jjWdQX|jjd|_|j j d|j j t jd|jdS(smAbort the currently running download process and initiate a new one with the same request paremeters.RNi(RR tcloseRRt SignalBlockerRtabortRHR"tsetTextR$R%R;R<R=(R>((s code\common\HttpRequestDialog.pytrestartDownloads      cCs_|jdk r|jdks$ttjj|j}xtry'tjj|sgtj |nPWq<t k rt j |j ddtj|t jt jBt j}|t jkrtSq<Xq<WxtrZyt|jd|_PWqtk rVt j |j ddtj|jt jt jBt j}|t jkrWtSqXqWtS(sCreate an empty file at 'self.tmpFilePath' and open it. Return 'True' if the operation succeeds and 'False' otherwise.sFailed to save data to disks8Failed to create directory for the destination file (%s)twbs7Failed to open the temporary data file (%s) for writingN(RR RREtostpathtdirnametTruetisdirtmakedirstOSErrort QMessageBoxtcriticalt parentWidgetRt quotePathtRetrytAbortR!topentIOError(R>t tmpFileDirtanswer((s code\common\HttpRequestDialog.pyt openTmpFiles8$       cCs|jdk r|jdk s$txtry|jj|PWq'tk rtj|j ddt j |jtj tj Btj }|tj krtSq'Xq'WtS(soWrite the given data to 'self.tmpFile'. Return 'True' if the operation succeeds and 'False' otherwise.sFailed to save data to disks,Failed to write the temporary data file (%s)N(RR RRERUtwriteR`RYRZR[RR\R]R^R!(R>tdataRb((s code\common\HttpRequestDialog.pyt writeTmpFiles$   cCs|jdk r|jdks$txtrytj|j|jPWq'tt fk rt j |j ddt j|jt jt jBt j}|t jkrtSq'Xq'W|jtS(sMove the file at 'self.tmpFilePath' to its final location. Return 'True' if the operation succeeds and 'False' otherwise.sFailed to save data to disksAFailed to move the temporary data file to its final location (%s)N(RR RRERUtshutiltcopyfileR R`RXRYRZR[RR\R]R^R!t deleteTmpFile(R>Rb((s code\common\HttpRequestDialog.pyt moveTmpFiles$   cCs|jdk r|jdks$txtrytj|jPWq'tk rtj |j ddt j |jtj tjBtj }|tjkrPqq'Xq'WdS(s&Delete the file at 'self.tmpFilePath'.s!Failed to clean up temporary datas-Failed to delete the temporary data file (%s)N(RR RRERURRtremoveRXRYRZR[RR\R]tIgnore(R>Rb((s code\common\HttpRequestDialog.pyRi&s$  cCst|j}|jdkr6|jj||_n4|jtjd|jj |t |j|_|jj j |j |jjj |j|jjj |jdS(Ns!application/x-www-form-urlencoded(tQNetworkRequestR R R RtgetRt setHeadertContentTypeHeadertpostt QByteArraytdownloadProgressR-tonDownloadProgressterrortonDownloadErrortfinishedtonDownloadFinished(R>trequest((s code\common\HttpRequestDialog.pyR==s  cCsW||kodknr dS|jj|dkry|jjd||f|jjd||jj|n7|jjd||jjdd|jjd|jdk r|jdkr|j s|j dSnt |j j }|jdk r!|j|7_n2|jdk s6t|j|sS|j dSdS(Niis%d bytes of %ds%d bytes(R$tshowR"ROtsetRangetsetValueRR RRcR.tstrRtreadAllRRERf(R>t bytesReceivedt bytesTotalRe((s code\common\HttpRequestDialog.pyRtPs,      cCs[tj|d|jjtjtjBtj}|tjkrM|jn |jdS(NsInternet connection error(RYRZRt errorStringR]R^RPR.(R>tnetworkErrorCodeRb((s code\common\HttpRequestDialog.pyRvys   cCsP|jdk rB|jjd|_|jsB|jdSn|jdS(N(RR RLRjR.RI(R>((s code\common\HttpRequestDialog.pyRxs    cCs\|jdk r2|jjd|_|jntj|j|jjWdQXdS(N(RR RLRiRRMRRN(R>((s code\common\HttpRequestDialog.pyR:s    N(t__name__t __module__t__doc__R RRFRGRKRPRcRfRjRiR=RtRvRxR:(((s code\common\HttpRequestDialog.pyR"s  O   2  !   ) ( RRtos.pathRgt PyQt4.QtCoret PyQt4.QtGuitPyQt4.QtNetworkt common.utilsRRtcommon.StatusLabelWidgett__all__RR(((s code\common\HttpRequestDialog.pyts