gallicagram 1.0.6 → 1.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 550bf9ebdcf99a41e01e157ba09dc5e864ef4fc531036b4afdd0501a4d5c4ad3
-  data.tar.gz: 6a18e0c6719baef37699c5035c7abc53008cca58040368fdbcf56ccb687de9f1
+  metadata.gz: fc701174f91b6d0af5092efa9944eccd1abdb6981fb9ac396240b62e4457cc8f
+  data.tar.gz: 027772ca247ea0be65812aed3607c2c705a07a424bfa2e5d21ad4eefc6854462
 SHA512:
-  metadata.gz: bbdb13303e02606ac3eea5911787815e5f52a0081adc59f5a7c96cfcfa962ca4edfb49dc2a93c020f973771bd18facd8b10812cb9dddad7b24f0404d9ae82c74
-  data.tar.gz: 428ddb548775277156091035b04f9d72f3b4549ea39a9842bbc7294a2a93e081e6ce65db402dba921dd119235e4b3a3d9f8e3e9aeab415907a9876d243a91af2
+  metadata.gz: 5818bd7812626868cc83bfd9f7c251f5a40eb1a4850823528e676815f7780aeab820ef02c10d55f6086559af17555c9cf5a59abf98c9c376f4f7b11a9063afc1
+  data.tar.gz: eae5a2a87b17d7b293e408eb27c4ad9bda093fdcc818a07ac084bc5f49ad939077413d8493c4caf91dcb5b0aa8a6d3f4f49170a2a2110664750d4256e6a1add0

data/lib/gallicagram.rb

@@ -6,25 +6,27 @@ class Gallicagram
   
 	# query = string or array of strings
 	# corpus = ["lemonde","livres","presse"]
-	# resolution = ["year", "month"]
-	def self.search(query,corpus="lemonde",start_date="1900",end_date="2000",resolution="month",sum=false)
+	# resolution = ["mois", "annee"]
+	def self.search(query,corpus="lemonde",start_date="1900",end_date="2000",resolution="annee",sum=false)
 
 		query = [query] unless query.kind_of?(Array)
-		output = ""
+		data = ""
 		query.each_with_index do |word, index|
 			query = format_query(word, sum)
-			response = call_api(word, corpus, start_date, end_date)
+			response = call_api(word, corpus, start_date, end_date,resolution)
 			unless index == 0
 				response = response.gsub("n,gram,annee,mois,jour,total", "\n").strip
 			end
-			output << response
+			data << response
 		end
 
-		if corpus == "livres" && resolution == "month"
-			resolution = "year"
+		if corpus == "livres" && resolution == "mois"
+			resolution = "annee"
 		end
 
-		data = group_by_resolution(output, resolution)
+		if resolution == "mois"
+			data = group_by_resolution(data)
+		end
 
 		return data
 	end
@@ -40,33 +42,24 @@ def format_query(query, sum)
 
 end
 
-def call_api(query, corpus, start_date, end_date)
+def call_api(query, corpus, start_date, end_date, resolution)
 	# We search through the Gallicagram API
-	url = "https://shiny.ens-paris-saclay.fr/guni/query?corpus=#{corpus}&mot=#{CGI.escape(query)}&from=#{start_date}&to=#{end_date}"
+	url = "https://shiny.ens-paris-saclay.fr/guni/query?corpus=#{corpus}&mot=#{CGI.escape(query)}&from=#{start_date}&to=#{end_date}&resolution=#{resolution}"
 
 	response = URI.open(url)
 
 	return response.read
 end
 
-def group_by_resolution(data, resolution)
+def group_by_resolution(data)
 	csv_parsing = CSV.parse(data, :headers => true)
 	clean_data_array = []
-	case resolution 
-	when "year"
-		csv_parsing.group_by { |word| [word["annee"], word["gram"]] }.each do |cp|
-			sum = cp[1].sum { |word| word["total"].to_i }
-			new_row = cp[0]
-			new_row << sum
-			clean_data_array << new_row
-		end
-	when "month"
-		csv_parsing.group_by { |word| [word["annee"], word["mois"], word["gram"]] }.each do |cp|
-			sum = cp[1].sum { |word| word["total"].to_i }
-			new_row = cp[0]
-			new_row << sum
-			clean_data_array << new_row
-		end
+
+	csv_parsing.group_by { |word| [word["annee"], word["mois"], word["gram"]] }.each do |cp|
+		sum = cp[1].sum { |word| word["total"].to_i }
+		new_row = cp[0]
+		new_row << sum
+		clean_data_array << new_row
 	end
 
 	return clean_data_array

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gallicagram
 version: !ruby/object:Gem::Version
-  version: 1.0.6
+  version: 1.0.7
 platform: ruby
 authors:
 - Nicolas Le Roux