diff options
Diffstat (limited to 'TTS.rb')
| -rwxr-xr-x | TTS.rb | 136 |
1 files changed, 136 insertions, 0 deletions
@@ -0,0 +1,136 @@ +#!/usr/bin/env ruby +require 'rubygems' +require 'date' +require 'highline/import' +require 'hpricot' +require 'net/http' +require 'net/https' +require 'net/http/post/multipart' +require 'uri' + +@parameters={ + "time_in"=>"", + "time_out"=>"", + "time_break"=>"", + "weekend_days"=>[], + "login_url"=>'', + "timesheet_path"=>'', +} + +# You normally don't need to do any change below this line + +def create_timesheet_metadata(data) + editts_csrf_token = (Hpricot(data)/"input"/"#editTimesheet__csrf_token").first[:value] + editts_tsId = (Hpricot(data)/"input"/"#editTimesheet_txtTimesheetId").first[:value] + editts_employeeNumber = (Hpricot(data)/"input"/"#editTimesheet_employeeNumber").first[:value] + editts_yearChoices = (Hpricot(data)/"select"/"#editTimesheet_yearChoices/option[@selected=selected]").first[:value] + editts_monthChoices = (Hpricot(data)/"select"/"#editTimesheet_monthChoices/option[@selected=selected]").first[:value] + + metadata_hash = Hash.new + metadata_hash["editTimesheet[_csrf_token]"] = editts_csrf_token + metadata_hash["editTimesheet[txtTimesheetId]"] = editts_tsId + metadata_hash["editTimesheet[show]"] = "" + metadata_hash["editTimesheet[txtAction]"] = "save" + metadata_hash["editTimesheet[employeeNumber]"] = editts_employeeNumber + metadata_hash["editTimesheet[yearChoices]"] = editts_yearChoices + metadata_hash["editTimesheet[monthChoices]"] = editts_monthChoices + metadata_hash["editTimesheet[submitComment]"] = "" + metadata_hash +end + +def create_repost_data_input_hash(data,field) + i=0 + last_id = field + i.to_s + field_hash = Hash.new + while last_id_elem = (Hpricot(data)/last_id).first + field_hash[last_id_elem[:name]]= last_id_elem[:value] || "" + i+=1 + last_id = field + i.to_s + end + field_hash +end + +def create_repost_data_select_hash(data,field) + i=0 + last_id = field + i.to_s + field_hash = Hash.new + while last_id_elem = (Hpricot(data)/last_id).first + if selected = (last_id_elem/"option[@selected=selected]").first + field_hash[last_id_elem[:name]] = selected[:value] + else + field_hash[last_id_elem[:name]] = "1" + end + i+=1 + last_id = field + i.to_s + end + field_hash +end + +# Gererates the timesheet data (in_time, out_time, break_duration) +def generate_timesheet_data_hash(year,month) + ts_hash = Hash.new + i = 0 + day=Date.new(year,month,1) + while day.month == month + if @parameters["weekend_days"].include? day.wday + ts_hash["in_time[" + i.to_s + "]"] = "" + ts_hash["out_time[" + i.to_s + "]"] = "" + ts_hash["break_duration[" + i.to_s + "]"] = "" + ts_hash["row_total_duration[" + i.to_s + "]"] = "" + i += 1 + day += 1 + next + end + ts_hash["in_time[" + i.to_s + "]"] = @parameters["time_in"] + ts_hash["out_time[" + i.to_s + "]"] = @parameters["time_out"] + ts_hash["break_duration[" + i.to_s + "]"] = @parameters["time_break"] + i+=1 + day += 1 + end + ts_hash +end + + +data_hash = Hash.new +headers = Hash.new +username = ask("Enter your username: ") { |q| q.echo = true } +password = ask("Enter your password: " ) { |q| q.echo = "*"} +uri = URI(@parameters['login_url']) + +# Start session for all the transactions +Net::HTTP.start(uri.host, uri.port, + :use_ssl => uri.scheme == 'https') do |http| + request = Net::HTTP::Get.new uri.request_uri + + # Get the login page and login + response = http.request request + cookie = response['set-cookie'] + login_csrf_token = (Hpricot(response.body)/"input"/"#login__csrf_token").first[:value] + data_hash = Hash.new + data_hash["login[txtUserName]"] = username + data_hash["login[txtUserPassword]"] = password + data_hash["login[_csrf_token]"] = login_csrf_token + headers = { + 'Cookie' => cookie, + 'Referer' => @parameters['login_url'], + 'Content-Type' => 'multipart/form-data' + } + + res = http.request(Net::HTTP::Post::Multipart.new('/', data_hash, headers)) + + # Get the timesheet page and create post data + res = http.get(@parameters['timesheet_path'], headers) + data_hash=data_hash.merge(create_timesheet_metadata(res.body)) + data_hash=data_hash.merge(create_repost_data_input_hash(res.body,"#original_record_ids_")) + data_hash=data_hash.merge(create_repost_data_input_hash(res.body,"#record_id_")) + data_hash=data_hash.merge(create_repost_data_input_hash(res.body,"#date_")) + data_hash=data_hash.merge(create_repost_data_input_hash(res.body,"#medical_checkup_duration_")) + data_hash=data_hash.merge(create_repost_data_input_hash(res.body,"#comment_")) + data_hash=data_hash.merge(create_repost_data_select_hash(res.body,"#classification_id_")) + data_hash=data_hash.merge(generate_timesheet_data_hash(data_hash["editTimesheet[yearChoices]"].to_i,data_hash["editTimesheet[monthChoices]"].to_i)) + + # Save the timesheets + res = http.request(Net::HTTP::Post::Multipart.new((Hpricot(res.body)/"form"/"#editTimesheet").first[:action], data_hash, headers)) +end + +print "The timesheets should have been saved. Please do a manual check before submitting!\nPlease report bugs, comments and improvements at ctrianta@rh.c.\n" |