ehpt 1.0.4 → 1.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f3653e66627846fae073f4173a4f51cf683b76284d1b413095347c9f3462a94e
-  data.tar.gz: f8835fe7726e1f883e9f5608328e7257f06abdf0a8552cd1332662ad64e64ace
+  metadata.gz: bee523dd1511046a8f0aeb28f6587c5489c4d7b674b0fdd71eb178cc1576da86
+  data.tar.gz: dc47c8db8fe50b4d672261be2106a6c0b5f64e71321528b14d0116bafd5b2af4
 SHA512:
-  metadata.gz: ae913c7ffbd558597ca0847638ffe32bdffdc8e5bf412c9352bfdd38faaa107809f8e092fbada9e68080f9704d2fc97f169b525e6cf68b10200978c106720b60
-  data.tar.gz: 63c49d8f401bf28dbdb407c00878f56d1fafe7b522211ea190197764a106b43d33253109df0fb67a95448892341fd5921986a96fb65e8548babacc7f39f5826a
+  metadata.gz: b404e4aa579b442dabda86c2f4654796f24e92715043d6643f39bc16ea045d379c7941520fad7b06263be4aedd245ede884b6b172eba8868125dd61da72948b1
+  data.tar.gz: 8c73fd0d71dc064f8673526998d23d8b4bed5828f9d6cd263a94cffe27b18a07d029ebec2f4c8eca4df1c52554ef339e4358763e8b6607401ca77b03ddf70a65

data/lib/ehpt/base.rb

@@ -1,6 +1,6 @@
 module Ehpt
   class Base
-    attr_reader :data, :errors
+    attr_reader :data, :errors, :warnings
 
     def self.call(*args)
       new(*args).call
@@ -9,6 +9,7 @@ module Ehpt
     def initialize(*args)
       @data = nil
       @errors = []
+      @warnings = []
     end
 
     def success?
@@ -19,6 +20,10 @@ module Ehpt
       !success?
     end
 
+    def warning?
+      !warnings.empty?
+    end
+
     def add_error(error)
       if error.is_a?(Array)
         error.each do |err|
@@ -28,5 +33,15 @@ module Ehpt
         @errors << error
       end
     end
+
+    def add_warning(warning)
+      if warning.is_a?(Array)
+        warning.each do |err|
+          add_warning(err)
+        end
+      else
+        @warnings << warning
+      end
+    end
   end
 end

data/lib/ehpt/create_stories.rb

@@ -1,22 +1,9 @@
-require 'csv'
-require 'ehpt/base'
-require 'ehpt/create_story'
-
 module Ehpt
   class CreateStories < Base
-    attr_reader :csv_file, :project
-
-    ARRAY_TYPE_ATTRIBUTES = %w[ labels tasks pull_requests branches blockers comments reviews ]
-
-    INT_ARRAY_TYPE_ATTRIBUTES = %w[ owner_ids label_ids follower_ids ]
-
-    INT_TYPE_ATTRIBUTES = %w[ project_id requested_by_id before_id after_id integration_id ]
-
-    FLOAT_TYPE_ATTRIBUES = %w[ estimate ]
+    attr_reader :csv_file
 
-    def initialize(csv_file, project)
+    def initialize(csv_file)
       @csv_file = csv_file
-      @project = project
       super
     end
 
@@ -39,7 +26,10 @@ module Ehpt
     def create_stories
       CSV.foreach(csv_file, headers: true) do |row|
         story_attrs = create_story_attributes(row)
-        story_creator = Ehpt::CreateStory.new(project, story_attrs)
+
+        next if error?
+
+        story_creator = Ehpt::CreateStory.new(story_attrs)
         story_creator.call
 
         if story_creator.success?
@@ -54,14 +44,17 @@ module Ehpt
     end
 
     def create_story_attributes(row)
-      story_attrs = row.to_h.compact
-      story_attrs.each do |key, value|
-        story_attrs[key] = value.to_i if INT_TYPE_ATTRIBUTES.include?(key)
-        story_attrs[key] = value.to_f if FLOAT_TYPE_ATTRIBUES.include?(key)
-        story_attrs[key] = value.split(',') if ARRAY_TYPE_ATTRIBUTES.include?(key)
-        story_attrs[key] = value.split(',').map(&:to_i) if INT_ARRAY_TYPE_ATTRIBUTES.include?(key)
+      attr_creator = Ehpt::CreateStoryAttributes.new(row)
+      attr_creator.call
+
+      add_warning(attr_creator.warnings) if attr_creator.warning?
+
+      if attr_creator.error?
+        add_error(attr_creator.errors)
+        return
       end
-      story_attrs
+
+      attr_creator.data
     end
   end
 end

data/lib/ehpt/create_story.rb

@@ -1,11 +1,8 @@
-require 'ehpt/base'
-
 module Ehpt
   class CreateStory < Base
-    attr_reader :project, :story_attrs
+    attr_reader :story_attrs
 
-    def initialize(project, story_attrs)
-      @project = project
+    def initialize(story_attrs)
       @story_attrs = story_attrs
       super
     end
@@ -20,7 +17,7 @@ module Ehpt
     private
 
     def create_story
-      @data = project.create_story(story_attrs)
+      @data = Ehpt.project.create_story(story_attrs)
     end
 
     def prefix_story_name_with_id

data/lib/ehpt/create_story_attributes.rb

@@ -0,0 +1,63 @@
+module Ehpt
+  class CreateStoryAttributes < Base
+    attr_reader :row
+
+    ARRAY_TYPE_ATTRIBUTES = %w[ labels tasks pull_requests branches blockers comments reviews ]
+
+    INT_ARRAY_TYPE_ATTRIBUTES = %w[ owner_ids label_ids follower_ids ]
+
+    INT_TYPE_ATTRIBUTES = %w[ project_id requested_by_id before_id after_id integration_id ]
+
+    FLOAT_TYPE_ATTRIBUES = %w[ estimate ]
+
+    def initialize(row)
+      @row = row
+      super
+    end
+
+    def call
+      story_attrs = row.to_h.compact
+      story_attrs.each do |key, value|
+        story_attrs[key] = value.to_i if INT_TYPE_ATTRIBUTES.include?(key)
+        story_attrs[key] = value.to_f if FLOAT_TYPE_ATTRIBUES.include?(key)
+        story_attrs[key] = value.split(',') if ARRAY_TYPE_ATTRIBUTES.include?(key)
+        story_attrs[key] = value.split(',').map(&:to_i) if INT_ARRAY_TYPE_ATTRIBUTES.include?(key)
+      end
+
+      if story_attrs.has_key?('requested_by')
+        user_id = get_user_id_from(story_attrs.delete('requested_by'))
+        story_attrs['requested_by_id'] = user_id unless user_id.nil?
+      end
+
+      if story_attrs.has_key?('owners')
+        owners = story_attrs.delete('owners').split(',')
+        owner_ids = owners.map { |owner| get_user_id_from(owner.strip) }.compact
+        story_attrs['owner_ids'] = owner_ids unless owner_ids.empty?
+      end
+
+      @data = story_attrs
+    rescue StandardError => e
+      add_error({
+        row: row.to_h,
+        warnings: e.message
+      })
+    end
+
+    private
+
+    def get_user_id_from(initial)
+      user_id_getter = Ehpt::GetUserIdFromInitial.new(initial)
+      user_id_getter.call
+
+      if user_id_getter.error?
+        add_warning({
+          row: row.to_h,
+          warnings: user_id_getter.errors
+        })
+        return
+      end
+
+      user_id_getter.data
+    end
+  end
+end

data/lib/ehpt/get_project.rb

@@ -1,6 +1,3 @@
-require 'tracker_api'
-require 'ehpt/base'
-
 module Ehpt
   class GetProject < Base
     attr_reader :token, :project_id

data/lib/ehpt/get_user_id_from_initial.rb

@@ -0,0 +1,49 @@
+module Ehpt
+  class GetUserIdFromInitial < Base
+    @@memberships = nil
+
+    attr_reader :initial
+
+    def self.memberships
+      @@memberships
+    end
+
+    def self.memberships=(memberships)
+      @@memberships = memberships
+    end
+
+    def initialize(initial)
+      @initial = initial
+      super
+    end
+
+    def call
+      fetch_memberships_from_pt!
+
+      if user.nil?
+        add_error("Not found any user with initial #{initial}")
+      else
+        @data = user.id
+      end
+    rescue StandardError => e
+      add_error(eval(e.message)[:body])
+    end
+
+    private
+
+    def memberships
+      self.class.memberships
+    end
+
+    def fetch_memberships_from_pt!
+      self.class.memberships ||= Ehpt.project.memberships
+    end
+
+    def user
+      member = memberships.find do |membership|
+        membership.person.initials.downcase == initial.downcase
+      end
+      member && member.person
+    end
+  end
+end

data/lib/ehpt.rb

@@ -1,9 +1,22 @@
+require 'tracker_api'
 require 'pp'
 require 'csv'
-require 'ehpt/get_project'
+require 'ehpt/base'
 require 'ehpt/create_stories'
+require 'ehpt/create_story'
+require 'ehpt/create_story_attributes'
+require 'ehpt/get_project'
+require 'ehpt/get_user_id_from_initial'
 
 module Ehpt
+  def self.project=(project)
+    @@project = project
+  end
+
+  def self.project
+    @@project
+  end
+
   def self.call(csv_file, token, project_id)
     project_getter = Ehpt::GetProject.new(token, project_id)
     project_getter.call
@@ -14,10 +27,10 @@ module Ehpt
       return
     end
 
+    self.project = project_getter.data
     puts "Found project: #{project_getter.data.name}"
-    project = project_getter.data
 
-    stories_creator = Ehpt::CreateStories.new(csv_file, project)
+    stories_creator = Ehpt::CreateStories.new(csv_file)
     stories_creator.call
 
     puts "Done"
@@ -26,5 +39,10 @@ module Ehpt
       puts "===== Errors ====="
       pp stories_creator.errors
     end
+
+    if stories_creator.warning?
+      puts "===== Warnings ====="
+      pp stories_creator.warnings
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ehpt
 version: !ruby/object:Gem::Version
-  version: 1.0.4
+  version: 1.0.5
 platform: ruby
 authors:
 - phamhoaivu911
@@ -104,7 +104,9 @@ files:
 - lib/ehpt/base.rb
 - lib/ehpt/create_stories.rb
 - lib/ehpt/create_story.rb
+- lib/ehpt/create_story_attributes.rb
 - lib/ehpt/get_project.rb
+- lib/ehpt/get_user_id_from_initial.rb
 homepage: https://github.com/phamhoaivu911/ehpt
 licenses:
 - MIT