fix bug import csv

app/models/company.rb

@@ -2,5 +2,6 @@
 
 
 # Description/Explanation of Person class
 # Description/Explanation of Person class
 class Company < ApplicationRecord
 class Company < ApplicationRecord
+  COMPANY_SECURITY = 1
   has_many :jobs
   has_many :jobs
 end
 end

config/application.rb

@@ -11,6 +11,7 @@ module Venjob
     # Initialize configuration defaults for originally generated Rails version.
     # Initialize configuration defaults for originally generated Rails version.
     config.load_defaults 5.2
     config.load_defaults 5.2
 
 
+    config.autoload_paths << Rails.root.join('lib/service')
     # Settings in config/environments/* take precedence over those specified here.
     # Settings in config/environments/* take precedence over those specified here.
     # Application configuration can go into files in config/initializers
     # Application configuration can go into files in config/initializers
     # -- all .rb files in that directory are automatically loaded after loading
     # -- all .rb files in that directory are automatically loaded after loading

lib/service/ftp.rb → lib/service/ftp_sever.rb

 # frozen_string_literal: true
 # frozen_string_literal: true
 
 
-require 'service/unzip'
 require 'net/ftp'
 require 'net/ftp'
 require 'logger'
 require 'logger'
 require 'csv'
 require 'csv'
@@ -30,9 +29,11 @@ class FtpSever
   def download_csv
   def download_csv
     Net::FTP.open(CONTENT_SERVER_DOMAIN_NAME, CONTENT_SERVER_USER_NAME, CONTENT_SERVER_USER_PASSWORD) do |ftp|
     Net::FTP.open(CONTENT_SERVER_DOMAIN_NAME, CONTENT_SERVER_USER_NAME, CONTENT_SERVER_USER_PASSWORD) do |ftp|
       ftp.getbinaryfile(NAME_CSV)
       ftp.getbinaryfile(NAME_CSV)
-      Unzip.extract_zip((jobs_csv_path), Rails.root.join('lib', 'csv'))
+      Unzip.extract_zip(jobs_csv_path, Rails.root.join('lib', 'csv'))
       File.delete(jobs_csv_path) if File.exist?(jobs_csv_path)
       File.delete(jobs_csv_path) if File.exist?(jobs_csv_path)
       logger.info 'Download & extract success'
       logger.info 'Download & extract success'
+    rescue StandardError => e
+      logger.error "Donwload csv have error: #{e}"
     end
     end
   end
   end
 end
 end

lib/service/import_data.rb

 # frozen_string_literal: true
 # frozen_string_literal: true
 
 
-require 'service/ftp'
-
 # Import data from CSV
 # Import data from CSV
 class ImportData
 class ImportData
-  COMPANY_SECURITY = 1
+  attr_accessor :csv
+
+  def initialize(csv)
+    @csv = csv
+  end
 
 
   def import_data
   def import_data
-    csv_data = FtpSever.new.data_csv
-    import_industries_from(csv_data)
-    import_cities_from(csv_data)
-    import_companies_from(csv_data)
-    import_jobs_from(csv_data)
+    import_companies
+    import_jobs
   end
   end
 
 
   def logger
   def logger
@@ -20,31 +19,7 @@ class ImportData
 
 
   private
   private
 
 
-  def import_industries_from(csv)
-    industries = []
-    industries += csv['category'].map(&:strip)
-    industries.each do |val|
-      val.gsub!(',', '/') if val.include?(',')
-      val.gsub!('/', ' / ')
-      Industry.find_or_create_by(name: val)
-    end
-  rescue StandardError => e
-    logger.error "Import_industries: #{e}"
-  end
-
-  def import_cities_from(csv)
-    city_names = csv['work place'].uniq.select(&:present?)
-    cities = city_names.map { |val| val.delete('[]\"') }
-    cities.each do |val|
-      next if val.blank?
-
-      City.find_or_create_by(name: val, area: City.areas['domestic'])
-    end
-  rescue StandardError => e
-    logger.error "Import_cities: #{e}"
-  end
-
-  def import_companies_from(csv)
+  def import_companies
     csv['company name'].each_with_index do |name, index|
     csv['company name'].each_with_index do |name, index|
       Company.find_or_create_by(name: name.strip) do |company|
       Company.find_or_create_by(name: name.strip) do |company|
         company.address           =  csv['company address'][index]
         company.address           =  csv['company address'][index]
@@ -55,29 +30,30 @@ class ImportData
     end
     end
   end
   end
 
 
-  def import_jobs_from(csv)
-    csv['name'].each_with_index do |name, index|
-      desc = "#{csv['requirement'][index]} #{(csv['description'][index])}"
-      company = Company.find_by name: csv['company name'][index].to_s.strip
-      company_id = company.try(:id) || COMPANY_SECURITY
-      begin
-        job = Job.create(name: name,
+  def import_jobs
+    csv.each do |val|
+      desc = "#{val['requirement']} #{(val['description'])}"
+      # Get ID Company to create job
+      company = Company.find_by name: val['company name'].strip
+      company_id = company.try(:id) || Company::COMPANY_SECURITY
+      # Get data city do create relationship city_jobs
+      city_names = val['work place'].delete('[]\"')
+      city = City.find_or_create_by(name: city_names.strip) { |record| record.area = City.areas['domestic'] }
+      # Get data Industry do create relationship industry_jobs
+      industry_name = val['category'].gsub(',', '/').gsub('/', ' / ')
+      industry = Industry.find_or_create_by(name: industry_name.strip)
+
+      Job.find_or_create_by(name: val['name'],
                             company_id: company_id,
                             company_id: company_id,
-                         level: csv['level'][index],
-                         salary: csv['salary'][index],
+                            level: val['level'],
+                            salary: val['salary'],
                             create_date: Time.now,
                             create_date: Time.now,
-                         description: desc)
-        city_names = csv['work place'][index].to_s.delete('[]\"')
-        city = City.find_or_create_by(name: city_names.strip) { |record| record.area = DOMESTIC }
+                            description: desc) do |job|
         job.cities << city
         job.cities << city
-
-        industry_name = csv['category'][index].to_s.gsub(',', '/').gsub('/', ' / ')
-        industry = Industry.find_or_create_by(name: industry_name.strip)
         job.industries << industry
         job.industries << industry
-        puts index
+      end
     rescue StandardError => e
     rescue StandardError => e
       logger.error "Import_jobs: #{e}"
       logger.error "Import_jobs: #{e}"
     end
     end
   end
   end
-  end
 end
 end

lib/tasks/csv.rake

 # frozen_string_literal: true
 # frozen_string_literal: true
 
 
-require 'service/import_data'
-
-# Description/Explanation of Person class
+# Import data from csv
 namespace :csv do
 namespace :csv do
   task import_csv: :environment do
   task import_csv: :environment do
-    csv = ImportData.new
-    csv.import_data
+    data_csv = FtpSever.new.data_csv
+    ImportData.new(data_csv).import_data
   end
   end
 end
 end