pokotarou 1.1.8 → 1.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 166487c039cbcf7a6de3ccf579ca5a543c3fa28c154e94710df616e2db6a8fd7
-  data.tar.gz: 356cdfc5ea3b7e6bcf1d202b55f243056692c5cb597955ead60f7ee060b8e003
+  metadata.gz: 1a8dbadc1f72b24fc4ae994e3334a8e8c78725536a2dbc5437d7203ab77a69c5
+  data.tar.gz: 58f19fcfcd76b2f7df6c0065b3816ef5062735a92390dcdb94804bb9608ca699
 SHA512:
-  metadata.gz: 3624da9475a99fbdbb1a0bcbbe3bc037f4e5ef1c65f847999f1f1b6644da1634be87e0801d79f4da4a0cadc5e4a31d229ea192ac6299ebbd40a203cd9282a5e5
-  data.tar.gz: 1abcc3e584a777db0a277a9a5802f680bf967f302452a096d734ac505c352fd7ef388d1b327093c21aade51f0c8c2398f2273a91a294ef545094089de9479048
+  metadata.gz: 2d74d989306dfc6800a7409ee940ed171de101f8e26ddfe0c7e2a657eba695d92a054022b7ddaabc80dda234ca1ecca05da2f1154b1c6622507e94eea35e6626
+  data.tar.gz: ef45d33242aed49e59a23c216d3964e468414ab0d6d62d230229407673b6cb74a3709ad272a1767815c10e23c123ce44223e392d077bfba68272eb338efae078

data/lib/pokotarou.rb

@@ -1,112 +1,102 @@
-Dir[File.expand_path('../pokotarou', __FILE__) << '/*.rb'].each do |file|
-  require file
-end
-
 require "activerecord-import"
+require "pokotarou/handler_factory"
+require "pokotarou/operator"
 
 module Pokotarou
-  class NotFoundLoader < StandardError; end
-
-  class << self
-    def execute input
-      init_proc()
-
-      # if input is filepath, generate config_data
-      return_val =
-        if input.kind_of?(String)
-          DataRegister.register(gen_config(input))
-        else
-          DataRegister.register(input)
-        end
+  class V2 
+    def initialize filepath
+      @handler = Pokotarou::HandlerFactory.gen_handler(filepath)
 
-      AdditionalMethods.remove_filepathes_from_yml()
-
-      return_val
+      return self
     end
 
-    def pipeline_execute input_arr
-      init_proc()
-
-      return_vals = []
-      input_arr.each do |e|
-        handler = gen_handler_with_cache(e[:filepath])
-
-        if e[:change_data].present?
-          e[:change_data].each do |block, config|
-            config.each do |model, seed|
-              seed.each do |col_name, val|         
-                handler.change_seed(block, model, col_name, val)
-              end
-            end
-          end
-        end
-        
-        e[:args] ||= {}
-        e[:args][:passed_return_val] = return_vals.last
-        set_args(e[:args])
-
-        return_vals << Pokotarou.execute(handler.get_data())
-        AdditionalMethods.remove_filepathes_from_yml()
-      end
-
-      return_vals
+    def make
+      @handler.make()
     end
 
-    def import filepath
-      init_proc()
+    def delete_block sym_block
+      @handler.delete_block(sym_block)
 
-      AdditionalMethods.import(filepath)
+      self
     end
 
-    def set_args hash
-      Arguments.import(hash)
-    end
+    def delete_model sym_block, sym_class
+      @handler.delete_model(sym_block, sym_class)
 
-    def reset
-      AdditionalMethods.remove()
-      Arguments.remove()
-      AdditionalVariables.remove()
-      @handler_chache = {}
+      self
     end
 
-    def gen_handler filepath
-      init_proc()
+    def delete_col sym_block, sym_class, sym_col
+      @handler.delete_col(sym_block, sym_class, sym_col)
 
-      PokotarouHandler.new(gen_config(filepath))
+      self
     end
 
-    def gen_handler_with_cache filepath
-      init_proc()
+    def change_loop sym_block, sym_class, n
+      @handler.change_loop(sym_block, sym_class, n)
+
+      self
+    end
 
-      @handler_cache ||= {}
-      @handler_cache[filepath] ||= PokotarouHandler.new(gen_config(filepath))
+    def change_seed sym_block, sym_class, sym_col, arr
+      @handler.change_seed(sym_block, sym_class, sym_col, arr)
 
-      @handler_cache[filepath].deep_dup
+      self
     end
 
-    private
-    def init_proc
-      AdditionalMethods.init()
+    def get_data
+      @handler.get_data
     end
 
-    def gen_config filepath
-      contents = load_file(filepath)
-      set_const_val_config(contents)
-      DataStructure.gen(contents)
+    def set_data data
+      @handler.set_data(data)
     end
+ 
+    def set_randomincrement sym_block, sym_class, status
+      @handler.set_randomincrement(sym_block, sym_class, status)
 
-    def set_const_val_config contents
-      AdditionalVariables.set_const(contents)
+      self
     end
 
-    def load_file filepath
-      case File.extname(filepath)
-      when ".yml"
-        return YmlLoader.load(filepath)
-      else
-        raise NotFoundLoader.new("not found loader")
-      end
+    def set_autoincrement sym_block, sym_class, status
+      @handler.set_autoincrement(sym_block, sym_class, status)
+
+      self
     end
+  end
+
+  # データ作成メソッドはmakeで統一させる
+  # executeはレガシー扱いに設定する
+  def self.make input
+    Operator.execute(input)
+  end
+
+  def self.execute input
+    Operator.execute(input)
+  end
+
+  def self.pipeline_execute input_arr
+    Operator.pipeline_execute(input_arr)
+  end
+
+  def self.import filepath
+    Operator.import(filepath)
+  end
+
+  def self.set_args hash
+    Operator.set_args(hash)
+  end
+
+  def self.reset
+    Operator.reset()
+  end
 
+  def self.gen_handler filepath
+    HandlerFactory.gen_handler(filepath)
   end
-end
+
+  def self.gen_handler_with_cache filepath
+    HandlerFactory.gen_handler_with_cache(filepath)
+  end
+
+end

data/lib/pokotarou/additional_arguments/main.rb

@@ -0,0 +1,24 @@
+module Pokotarou
+  class MisMatchArgType < StandardError; end
+  class AdditionalArguments
+    module Main
+      class << self
+        @args = nil
+        attr_reader :args
+      
+        def import hash_data
+          unless hash_data.kind_of?(Hash)
+            raise MisMatchArgType.new("Please use Hash for args")
+          end
+  
+          @args = hash_data
+        end 
+  
+        def remove
+          @args = nil
+        end
+  
+      end  
+    end
+  end
+end

data/lib/pokotarou/additional_methods/main.rb

@@ -0,0 +1,49 @@
+module Pokotarou
+  module ::Pokotarou::AdditionalMethods
+    module Main
+      class << self
+        attr_reader :filepathes
+        attr_reader :filepathes_from_yml
+  
+        @filepathes = []
+        @filepathes_from_yml = []
+  
+        def init
+          @filepathes ||= []
+          @filepathes_from_yml = []
+        end
+  
+        def import filepath
+          add(@filepathes, filepath)
+        end
+  
+        def import_from_yml filepath
+          add(@filepathes_from_yml, filepath)
+        end
+  
+        def remove
+          @filepathes = []
+          @filepathes_from_yml = []
+        end
+  
+        def remove_filepathes_from_yml
+          @filepathes_from_yml = []
+        end
+  
+        def load
+          @filepathes.each do |filepath|; require filepath end
+          @filepathes_from_yml.each do |filepath|; require filepath end
+        end
+  
+        private
+        def add filepath_arr, filepath
+          if filepath.instance_of?(Array)
+            filepath_arr.concat(filepath)
+          else
+            filepath_arr.push(filepath)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/pokotarou/additional_variables/main.rb

@@ -0,0 +1,28 @@
+module Pokotarou
+  class AdditionalVariables
+    module Main
+      class << self
+        CONST_KEY = :"const'"
+        @const = {}
+        attr_reader :const
+  
+        def set_const data
+          return {} unless data.has_key?(CONST_KEY)
+          @const = data[CONST_KEY]
+  
+          # parse expression configlation
+          @const.each do |key, val|
+            @const[key] = ConstParser.parse(val)
+          end
+  
+          data.delete(CONST_KEY)
+        end
+  
+        def remove
+          @const = {}
+        end
+      
+      end
+    end  
+  end
+end

data/lib/pokotarou/handler.rb

@@ -0,0 +1,66 @@
+module Pokotarou
+
+  class Handler
+    def initialize data
+      @data = data
+    end
+
+    def make
+      data = get_data()
+
+      AdditionalMethods::Main.init()
+  
+      # if input is filepath, generate config_data
+      return_val =
+        if data.kind_of?(String)
+          SeedDataRegister::Main.register(gen_config(data))
+        else
+          SeedDataRegister::Main.register(data)
+        end
+
+      AdditionalMethods::Main.remove_filepathes_from_yml()
+
+      return_val
+    end
+
+    def delete_block sym_block
+      @data.delete(sym_block)
+    end
+
+    def delete_model sym_block, sym_class
+      @data[sym_block].delete(sym_class)
+    end
+
+    def delete_col sym_block, sym_class, sym_col
+      exists_content = ->(key){ @data[sym_block][sym_class][key].present? }
+
+      @data[sym_block][sym_class][:col].delete(sym_col) if exists_content.call(:col)
+      @data[sym_block][sym_class][:option].delete(sym_col) if exists_content.call(:option)
+      @data[sym_block][sym_class][:convert].delete(sym_col) if exists_content.call(:convert)
+    end
+
+    def change_loop sym_block, sym_class, n
+      @data[sym_block][sym_class][:loop] = n
+    end
+
+    def change_seed sym_block, sym_class, sym_col, arr
+      @data[sym_block][sym_class][:col][sym_col] = arr
+    end
+
+    def get_data
+      @data.deep_dup
+    end
+
+    def set_data data
+      @data = data
+    end
+ 
+    def set_randomincrement sym_block, sym_class, status
+      @data[sym_block][sym_class][:randomincrement] = status
+    end
+
+    def set_autoincrement sym_block, sym_class, status
+      @data[sym_block][sym_class][:autoincrement] = status
+    end
+  end
+end

data/lib/pokotarou/handler_factory.rb

@@ -0,0 +1,29 @@
+require "pokotarou/handler"
+require "pokotarou/registration_config_maker/main"
+
+module Pokotarou
+
+  class HandlerFactory
+    class << self
+      def gen_handler filepath
+        AdditionalMethods::Main.init()
+  
+        Handler.new(gen_config(filepath))
+      end
+  
+      def gen_handler_with_cache filepath
+        AdditionalMethods::Main.init()
+  
+        @handler_cache ||= {}
+        @handler_cache[filepath] ||= Handler.new(gen_config(filepath))
+  
+        @handler_cache[filepath].deep_dup
+      end
+
+      private
+      def gen_config filepath
+        RegistrationConfigMaker::Main.gen(filepath)
+      end
+    end
+  end
+end

data/lib/pokotarou/operator.rb

@@ -0,0 +1,79 @@
+require "pokotarou/handler_factory"
+require "pokotarou/registration_config_maker/main"
+require "pokotarou/additional_arguments/main"
+require "pokotarou/additional_methods/main"
+require "pokotarou/additional_variables/main"
+require "pokotarou/seed_data_register/main.rb"
+require "pokotarou/parser/const_parser.rb"
+
+module Pokotarou
+  class Operator
+    class NotFoundLoader < StandardError; end
+    class << self
+      def execute input
+        AdditionalMethods::Main.init()
+  
+        # if input is filepath, generate config_data
+        return_val =
+          if input.kind_of?(String)
+            SeedDataRegister::Main.register(gen_config(input))
+          else
+            SeedDataRegister::Main.register(input)
+          end
+  
+        AdditionalMethods::Main.remove_filepathes_from_yml()
+  
+        return_val
+      end
+  
+      def pipeline_execute input_arr
+        AdditionalMethods::Main.init()
+  
+        return_vals = []
+        input_arr.each do |e|
+          handler = HandlerFactory.gen_handler_with_cache(e[:filepath])
+  
+          if e[:change_data].present?
+            e[:change_data].each do |block, config|
+              config.each do |model, seed|
+                seed.each do |col_name, val|         
+                  handler.change_seed(block, model, col_name, val)
+                end
+              end
+            end
+          end
+          
+          e[:args] ||= {}
+          e[:args][:passed_return_val] = return_vals.last
+          set_args(e[:args])
+  
+          return_vals << Pokotarou.execute(handler.get_data())
+          AdditionalMethods::Main.remove_filepathes_from_yml()
+        end
+  
+        return_vals
+      end
+  
+      def import filepath
+        AdditionalMethods::Main.init()
+        AdditionalMethods::Main.import(filepath)
+      end
+  
+      def set_args hash
+        AdditionalArguments::Main.import(hash)
+      end
+  
+      def reset
+        AdditionalMethods::Main.remove()
+        AdditionalArguments::Main.remove()
+        AdditionalVariables::Main.remove()
+        @handler_chache = {}
+      end
+  
+      def gen_config filepath
+        RegistrationConfigMaker::Main.gen(filepath)
+      end
+      
+    end
+  end
+end