summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorChristos Triantafyllidis <ctrianta@redhat.com>2012-11-05 01:13:17 +0100
committerChristos Triantafyllidis <ctrianta@redhat.com>2012-11-05 01:13:17 +0100
commit4c82230e601ca7ba2ff11ceb496938f92ae8d013 (patch)
treee97f6876c26b008c93a84bcf288930ae92ab70fa
downloadTTS-4c82230e601ca7ba2ff11ceb496938f92ae8d013.zip
TTS-4c82230e601ca7ba2ff11ceb496938f92ae8d013.tar.gz
TTS-4c82230e601ca7ba2ff11ceb496938f92ae8d013.tar.xz
Initial release
-rwxr-xr-xTTS.rb136
1 files changed, 136 insertions, 0 deletions
diff --git a/TTS.rb b/TTS.rb
new file mode 100755
index 0000000..4d61161
--- /dev/null
+++ b/TTS.rb
@@ -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"