U "h*@s|ddlZddlmZmZmZddlmZmZddlmZddl m Z ddl Z ddl Z GdddZ dd Zed krxedS) N)datetimetimezone timedelta) create_enginetext) DATABASE_URL)Configc@sBeZdZddZddZddZddd Zdd d Zdd dZd S)ActivityWatchSynccCstj|_tj|_tt|_dSN)rACTIVITYWATCH_HOSTaw_hostLOCAL_DEVELOPER_NAMEdeveloper_namerrengine)selfr5/var/www/html/timesheet/backend/activitywatch_sync.py__init__ szActivityWatchSync.__init__c CszDtj|jddd}|jdkr,|WStd|jiWSWn8tk r|}ztd|iWYSd}~XYnXdS)z#Get available ActivityWatch bucketsz/api/0/buckets )timeoutzError getting buckets: z#Error connecting to ActivityWatch: N)requestsgetr status_codejsonprint Exception)rresponseerrrget_activitywatch_bucketss   z+ActivityWatchSync.get_activitywatch_bucketsc Cszbtj|jd|d||dddd}|jdkrD|WStd|d |jgWSWn>tk r}z td|d |gWYSd }~XYnXd S) z$Get window events from ActivityWatchz/api/0/buckets/z/eventsi)startendlimit)paramsrrzError getting events from : N)rrr isoformatrrrr)r bucket_name start_timeend_timerrrrrget_window_eventss    z#ActivityWatchSync.get_window_eventscs||tfdddDr*dStfdddDrDdStfddd Dr^d Stfd dd Drxd StfdddDrdStfdddDrdStfdddDstfdddDrdSdS)z5Categorize application based on name and window titlec3s|]}|kVqdSr r).0Zdev_app app_lowerrr 9sz;ActivityWatchSync.categorize_application..) vscodecodecursorpycharmintellijsublimeatomvimemacsz notepad++ZIDEc3s|]}|kVqdSr r)r,browserr-rrr/@s)chromefirefoxsafariedgebraveoperaZBrowserc3s|]}|kVqdSr r)r,db_toolr-rrr/Fs)datagrippgadminZ phpmyadminmysql postgresqlmongodbDatabasec3s|]}|kVqdSr r)r,Z comm_toolr-rrr/Ls)slackteamsdiscordtelegramwhatsappzoomskype Communicationc3s|]}|kVqdSr r)r,Z prod_toolr-rrr/Rs) notionobsidiantrelloasanajira confluenceexcelword powerpointoutlook Productivityc3s|]}|kVqdSr r)r,Z file_toolr-rrr/Ys) filezillaZwinscpexplorerfinderterminalcmd powershellSystemc3s|]}|kVqdSr r)r,Zent_appr-rrr/_s)spotifyyoutubenetflixtwitchZsteamrIc3s|]}|kVqdSr r)r,Zent_word) title_lowerrrr/as)musicvideogameZ EntertainmentOther)lowerany)rapp_name window_titler)r.rercategorize_application3s&z(ActivityWatchSync.categorize_applicationNc Cs.td|jd|s0ttjjddddd}|s@ttj}|}|sXtddStdt|dd }| D]}d | krx|}qqx|std dStd || |||}|std dStdt|d|j (}|td|j||d|W5QRXd}|j }|D]} z| di} | dd} | dd} | dd} t| ddd}| dkrWq.|| | }|td|j| | || |d|d7}Wn<tk r}ztd|WYq.W5d }~XYnXq.|W5QRXtd |d!|jd"S)#z(Sync data from ActivityWatch to databaseu$🔄 Syncing ActivityWatch data for z...rhourminutesecond microsecondu#❌ No ActivityWatch buckets found!Fu 📊 Found z ActivityWatch bucketsNwindowu#❌ No window watcher bucket found!u📋 Using bucket: u%❌ No events found in ActivityWatch!u📥 Retrieved z events from ActivityWatchz DELETE FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date dev_id start_dateend_datedataappUnknowntitler+duration timestampZz+00:00ra  INSERT INTO activity_records (developer_id, application_name, window_title, category, duration, timestamp) VALUES (:dev_id, :app, :title, :category, :duration, :timestamp) )rvrzr|categoryr}r~zError processing event: u ✅ Stored z activity records for T)rrrnowrutcreplacerlenkeysrjr*rconnectexecutercommitr fromisoformatrnr)rrwrxbucketsZ window_bucketr'eventsconnZ stored_counteventryrlrmr}r~rrrrrsync_activitywatch_datagsx                z)ActivityWatchSync.sync_activitywatch_datac Cs|sttjjddddd}|s.ttj}|j}|td|j ||d }g}ddddd d g}t |D]8\}}| |d|d d |d ||t |dqt|td|j ||d} | dpdd } | d pd} || | ||ddW5QRSQRXdS)z"Get activity summary from databaserroa SELECT category, SUM(duration) as total_duration, COUNT(*) as activity_count FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date GROUP BY category ORDER BY total_duration DESC ruz#8884d8z#82ca9dz#ffc658z#ff7300z#8dd1e1z#d084d0rg @)rr}countcoloraN SELECT SUM(duration) as total_duration, COUNT(DISTINCT application_name) as project_count FROM activity_records WHERE developer_id = :dev_id AND timestamp >= :start_date AND timestamp <= :end_date )r r!)ry total_timeactive_projects date_rangeN)rrrrrrrrrrfetchall enumerateappendrfetchoner&) rrwrxrresult activitiescolorsirowZ total_resultrZ project_countrrr get_summarysF       zActivityWatchSync.get_summary)r+)NN)NN) __name__ __module__ __qualname__rrr*rnrrrrrrr s   4 `r cCst}ttjjddddd}|tdd}|||}|r|||}t d|j dt d|dd d t d |d t d t |d|dD]$}t d|dd|dd dqdS)z Manual sync function for testingrror)daysu 📊 Summary for :z Total time: rz.2fz hoursz Active projects: rz Categories: ryz - rr%r}hN) r rrrrrrrrrrr)synctodayZtomorrowsuccesssummaryactivityrrr manual_syncs   r__main__)rrrr sqlalchemyrrdatabaserconfigrrtimer rrrrrrs