peoplegroup-connectors 0.1.33 → 0.1.38

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b660ffce96d59df79bda2b48309142606ad7a5df8b684b4121f31a22071ec70
-  data.tar.gz: 79d28873b3b7309cc866278cfe22f14c61c967c7d7c74e4b8b0cce38ab5d96fa
+  metadata.gz: ce7bbe7a63bb597ed928d7333a33c15a9250bb7a78275a5d3221570b6a0d33af
+  data.tar.gz: 8654607403148b2398969daa729051617d4272df8c427ecf7db65e5ecc86b811
 SHA512:
-  metadata.gz: 33e90edb0b68f99fd05a60b67e870cc44361f4dd1824a0574f5d19c51182c03f918d2ac2d3073c56ff6c891decf49e04128ec4818bebdd2a37d10661bcc3709b
-  data.tar.gz: 592c6091a7d632cfa72144057eec962557c7eab55d099e11e87dc88059035bfdfce7b7e562a10374bbae6b64e86014acb91f03f43e946e6f2ffd923232c0f368
+  metadata.gz: 04be80f29f17161b84fe5c101e73d856e513afdea5306546bafbc3e2c4f809cdaf4495210cb0f5b31c549c8eeb291d15690c7d4534a4f3639fe28244b1a1ff0e
+  data.tar.gz: 0fd7ae60b6b1064b8d0780e642a1971148f019ca1910dc93be7615bdcea624782ce3945e374210f6220c25d7ee8fa7f76b133cf1288c402ad105036e63e183dd

data/lib/peoplegroup/connectors.rb

@@ -2,10 +2,12 @@
 
 require_relative 'connectors/version'
 require_relative 'connectors/gitlab'
+require_relative 'connectors/google_sheets'
 require_relative 'connectors/slack'
 require_relative 'connectors/pto_roots'
 require_relative 'connectors/bamboo'
 require_relative 'connectors/greenhouse'
+require_relative 'utils'
 
 module PeopleGroup
   module Connectors

data/lib/peoplegroup/connectors/bamboo.rb

@@ -5,9 +5,8 @@ require 'bamboozled'
 module PeopleGroup
   module Connectors
     class Bamboo
-      MAX_RETRIES = 3
-
-      def initialize
+      def initialize(use_report: false)
+        @use_report = use_report
         @client = Bamboozled.client(subdomain: 'gitlab', api_key: ENV['BAMBOO_API_KEY'])
       end
 
@@ -52,6 +51,11 @@ module PeopleGroup
         active_team_members.find { |tm| tm['id'] == team_member['supervisorEId'] }
       end
 
+      def fetch_manager_for_id(team_member_id)
+        team_member = get_employee_details(team_member_id)
+        active_team_members.find { |tm| tm['id'] == team_member['supervisorEId'] }
+      end
+
       def fetch_second_level_manager(team_member)
         fetch_manager(fetch_manager(team_member))
       end
@@ -77,12 +81,12 @@ module PeopleGroup
       end
 
       def employees
-        retries = 0
-
-        begin
-          @employees ||= @client.report.custom(fields, 'JSON').reject { |employee| employee['lastName'] == 'Test-Gitlab' }
-        rescue Net::ReadTimeout, Bamboozled::GatewayError
-          retry if (retries += 1) < MAX_RETRIES
+        Utils.retry_on_error(errors: [Net::ReadTimeout, Bamboozled::GatewayError]) do
+          if @use_report
+            @employees ||= @client.report.find(@use_report, 'JSON', true).reject { |employee| employee['lastName'] == 'Test-Gitlab' }
+          else
+            @employees ||= @client.report.custom(fields, 'JSON').reject { |employee| employee['lastName'] == 'Test-Gitlab' }
+          end
         end
       end
 
@@ -171,6 +175,14 @@ module PeopleGroup
         @client.employee.add_table_row(employee_id, 'customAdditionalInformation1', data)
       end
 
+      def additional_data(employee_id)
+        @client.employee.table_data(employee_id, 'employmentStatus')
+      end
+
+      def currency_conversion(employee_id)
+        @client.employee.table_data(employee_id, 'customCurrencyConversion')
+      end
+
       def add_bonus(team_member_id, comment)
         data = {
           customBonusdate: Date.today.to_s,
@@ -182,6 +194,10 @@ module PeopleGroup
         @client.employee.add_table_row(team_member_id, 'customBonus', data)
       end
 
+      def has_direct_reports?(id)
+        active_team_members.any? { |team_member| team_member['supervisorEId'] == id.to_s }
+      end
+
       private
 
       def meta_fields

data/lib/peoplegroup/connectors/gitlab.rb

@@ -7,8 +7,6 @@ module PeopleGroup
     class GitLab
       attr_accessor :client
 
-      MAX_RETRIES = 3
-
       API_URL = ENV['GITLAB_API_V4_URL'] || 'https://gitlab.com/api/v4'
 
       def initialize(token: ENV['GITLAB_API_TOKEN'])
@@ -144,14 +142,8 @@ module PeopleGroup
 
       private
 
-      def retry_on_error
-        retries = 0
-
-        begin
-          yield
-        rescue Gitlab::Error::InternalServerError
-          retry if (retries += 1) < MAX_RETRIES
-        end
+      def retry_on_error(&block)
+        Utils.retry_on_error(errors: [Gitlab::Error::InternalServerError], &block)
       end
 
       def find_epic(group_id, title)

data/lib/peoplegroup/connectors/google_sheets.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'google_drive'
+
+module PeopleGroup
+  module Connectors
+    class GoogleSheets
+      def spreadsheet(title)
+        retry_on_error { session.spreadsheet_by_title(title) }
+      end
+
+      def worksheet(spreadsheet_title, worksheet_title)
+        worksheets = retry_on_error { spreadsheet(spreadsheet_title).worksheets }
+        worksheets.find { |worksheet| worksheet.title == worksheet_title }
+      end
+
+      private
+
+      def service_account_key
+        StringIO.new(Base64.strict_decode64(ENV['GOOGLE_DRIVE_SERVICE_ACCOUNT_KEY']))
+      end
+
+      def session
+        @session ||= retry_on_error { GoogleDrive::Session.from_service_account_key(service_account_key) }
+      end
+
+      def retry_on_error(&block)
+        Utils.retry_on_error(errors: [Google::Apis::ServerError], delay: 1, &block)
+      end
+    end
+  end
+end

data/lib/peoplegroup/connectors/greenhouse.rb

@@ -19,20 +19,19 @@ module PeopleGroup
       def hired_candidates(updated_since)
         page = 1
         candidates = []
-        retries = 0
+        on_error = -> { p [updated_since, page] }
 
         loop do
-          begin
-            results = @client.candidates(nil, updated_after: updated_since, page: page)
-            break if results.empty?
-
-            results.each do |candidate|
-              candidates << candidate if hired_non_active?(candidate)
-            end
-          rescue GreenhouseIo::Error
-            p [updated_since, page]
-            retry if (retries += 1) < MAX_RETRIES
+          results = Utils.retry_on_error(errors: [GreenhouseIo::Error], on_error: on_error) do
+            @client.candidates(nil, updated_after: updated_since, page: page)
           end
+
+          break if results.empty?
+
+          results.each do |candidate|
+            candidates << candidate if hired_non_active?(candidate)
+          end
+
           page += 1
         end
 

data/lib/peoplegroup/connectors/version.rb

@@ -2,6 +2,6 @@
 
 module PeopleGroup
   module Connectors
-    VERSION = '0.1.33'
+    VERSION = '0.1.38'
   end
 end

data/lib/peoplegroup/utils.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module PeopleGroup
+  class Utils
+    class << self
+      def retry_on_error(max_attempts: 3, delay: 0, errors: [StandardError], on_error: -> {})
+        attempts = 0
+
+        begin
+          yield
+        rescue *errors => e
+          on_error.call
+          sleep delay
+          retry if (attempts += 1) < max_attempts
+          raise e
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: peoplegroup-connectors
 version: !ruby/object:Gem::Version
-  version: 0.1.33
+  version: 0.1.38
 platform: ruby
 authors:
 - lien van den steen
-autorequire:
+autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-03-29 00:00:00.000000000 Z
+date: 2021-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gitlab
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: google_drive
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: slack-ruby-client
   requirement: !ruby/object:Gem::Requirement
@@ -163,10 +177,12 @@ files:
 - lib/peoplegroup/connectors.rb
 - lib/peoplegroup/connectors/bamboo.rb
 - lib/peoplegroup/connectors/gitlab.rb
+- lib/peoplegroup/connectors/google_sheets.rb
 - lib/peoplegroup/connectors/greenhouse.rb
 - lib/peoplegroup/connectors/pto_roots.rb
 - lib/peoplegroup/connectors/slack.rb
 - lib/peoplegroup/connectors/version.rb
+- lib/peoplegroup/utils.rb
 homepage: https://gitlab.com/gitlab-com/people-group/peopleops-eng/connectors-gem
 licenses:
 - MIT
@@ -174,7 +190,7 @@ metadata:
   homepage_uri: https://gitlab.com/gitlab-com/people-group/peopleops-eng/connectors-gem
   source_code_uri: https://gitlab.com/gitlab-com/people-group/peopleops-eng/connectors-gem
   changelog_uri: https://gitlab.com/gitlab-com/people-group/peopleops-eng/connectors-gem
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -190,7 +206,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.1.4
-signing_key:
+signing_key: 
 specification_version: 4
 summary: Library for our shared connectors.
 test_files: []