o h,@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z dZhdZejeddejd ddd d d ddd addZdddZddZddZedkrddlmZeeZde_eeededejdddddSdS) zH TAS License Checker - Simple Web UI Upload file and solve CAPTCHA only N)render_templaterequestjsonify)secure_filenameuploads>csvxlsxlsxT)exist_ok templatesFidleReady to process files is_processingstatusmessageprogress captcha_urlcCs d|vo|dddtvS)z"Check if file extension is allowed.)rsplitlowerALLOWED_EXTENSIONS)filenamer'/var/www/html/captchaScrapper/web_ui.py allowed_file!s rc Cszzdtd<dtd<d|dtd<dtt}d |d }|d kr,d }n |d kr3d}n|dkr:d}n|dkrAd}n |dkrHd}nd }d }tdtjd|d|d|d| tj tj tj t }t j tj|||||||g|d} | durzOtj|rt|d:} t| } | ddtd<| d|dtd<d| vr| dtd<d| vr| dtd<dtd<Wdn1swYWnty} z td| WYd} ~ nd} ~ wwtd| dus| } td | | d!krEd"td<|d#td<ztj|r)t|td$|Wn)tyD} z td%| WYd} ~ nd} ~ wwd&td<|d'| td<ztj|rht|td(|Wnty} z td)| WYd} ~ nd} ~ wwWnTty} zGd&td<|d*t| td<ztj|rt|td+|Wnty}z td,|WYd}~nd}~wwWYd} ~ n d} ~ wwWd-td<dSWd-td<dSd-td<w).z0Run the appropriate scraper script based on typeTr processingrz Starting z data processing...rjob_status_z.jsontaszcaptchaScrapper/tas_web.pynswzcaptchaScrapper/nsw_web.pyqldzcaptchaScrapper/qld_web.pyviczcaptchaScrapper/vic_web.pyactzcaptchaScrapper/act_web.pyz Running:  )cwdNr processing...rrz8CAPTCHA detected! Please solve it in the browser window.zError reading status file: z$Process completed with return code: r completedz# processing completed successfully!Cleaned up uploaded file: z!Error cleaning up uploaded file: errorz% processing failed with return code: Cleaned up status file: zError cleaning up status file: z processing failed: z&Cleaned up uploaded file after error: z-Error cleaning up uploaded file after error: F)processing_statusupperinttimeprintsys executableospathdirnameabspath__file__ subprocessPopenpollexistsopenjsonloadget Exceptionsleepwaitremovestr) file_path scraper_typeuserId entityTypeuserRolejob_id status_file script_name project_rootprocessf status_datae return_code cleanup_errorrrrrun_scraper_script%s (              rWcCs t|dS)zBRun the tas_web.py script with the uploaded file (legacy function)r!)rW)rHrrrrun_tas_scripts rXcCs|ddd}|jddgddd}|d d d }|d d d}|ddd}|ddd}|jddgddd}dS)z.Register web UI routes with the main Flask appz/scrapercSstdS)z+Main page - File data upload and processingz upload.html)rrrrrindexz%register_web_ui_routes..indexz/uploadPOST)methodsc Sstdr tddidfSdtjvrtddidfStjdd}tjd d }tjd d }tjd d }td |td|td|tjd}|jd krXtddidfSt |jsetddidfSz?t |j}t j t|}||ddd|ddatjt|||||fd}d|_|tdd|d|dWSty}ztddt|idfWYd}~Sd}~ww)z)Handle file upload with scraper selectionrr-z'Already processing a file. Please wait.filezNo file uploadedstater!rJrrKrLzUserId:z Entity Type:z User Role:zNo file selectedz4Invalid file type. Please upload CSV or Excel files.Tuploadedz%File uploaded successfully. Starting r))rrr)targetargszFile uploaded successfully. z processing started.)successrscraperzUpload failed: N)r/rrfilesformrBrr3rrrr6r7join UPLOAD_FOLDERsaver0 threadingThreadrWdaemonstartrCrG) rIrJrKrLr^rrHthreadrTrrr upload_filesH          &z+register_web_ui_routes..upload_filez/statuscSsttS)zGet current processing status)rr/rrrr get_statusrZz*register_web_ui_routes..get_statusz/resetcSsddddddatddd S) zReset processing statusFr r rNrTz Status resetrcr)r/rrrrr reset_statussz,register_web_ui_routes..reset_statusz/cleanupc Sz+td}d}|D]}zt||d7}td|Wq Yq td|dWStyF}ztdt|dWYd }~Sd }~ww) zClean up status files status_*.jsonrrr.Trc cleaned_filesFrcr-Nglobr6rFr3rrCrG) status_files cleaned_countr^rTrrrcleanup_status_files  z4register_web_ui_routes..cleanup_status_filesz/cleanup-uploadsc Srt) z$Clean up any leftover uploaded filesz uploads/*.csvrrr,TrvFrxNry) upload_filesr|r^rTrrrcleanup_uploaded_filesr~z6register_web_ui_routes..cleanup_uploaded_filesz/captcha-solvedc Sszctd}|stddidfWSt|tjjd}t|d }t|}Wdn1s.wYd|d <d |d <d |d <t|d}tj ||ddWdn1sWwYtdddWSt y}ztddt |idfWYd}~Sd}~ww)zMark CAPTCHA as solved manuallyrur-zNo active processing foundr])keyr(Nrrz(CAPTCHA solved, continuing processing...rFcaptcha_detectedwr*)indentTzCAPTCHA marked as solvedrrz"Failed to mark CAPTCHA as solved: re) rzrmaxr6r7getctimer?r@rAdumprCrG)r{latest_status_filerRrSrTrrrmark_captcha_solveds&    &z3register_web_ui_routes..mark_captcha_solvedN)route)apprYrprqrsr}rrrrrregister_web_ui_routess  6    r__main__)Flasktas_license_checker_secretz&Starting TAS License Checker Web UI...z2Open your browser and go to: http://localhost:5000z0.0.0.0i)debughostport)rrr)__doc__r6r4r;rkr2r@rzflaskrrrwerkzeug.utilsrrirmakedirsr/rrWrXr__name__rr secret_keyr3runrrrrsD  d