sncf 0.1.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c0ebb101165159e3c8179e93f18a4891e222b655
-  data.tar.gz: 96b4e851249d3686a9dc2ae3a1c269512e29c3bc
+  metadata.gz: 543bad488723801d92911a47e70904a3d63b0fa1
+  data.tar.gz: 68db2da4c6a30470aa09128fcc27c348a18ae659
 SHA512:
-  metadata.gz: a142907f567191d2d6828043feb4ec8f9cf172821368b4bf156c6b671d9d600413946c326063aea9f4642f3ec94dc3f88571feb40f3782447ba6edbc183543a2
-  data.tar.gz: c2f5c562650e5a118f769dde5a5d7a2955518ce1ff736c2913f25ee89b888682903689b7abb91bafc65c238287645ae74c2c33047e0524c53bc8544f55ef5256
+  metadata.gz: 0290f41f58c8a1c97e19bff538e6436f66f8371dd407e32c0df22eff59755ed6759d67b65c6d445cd1ad42ba6ed3df87e938cfa724a716332ec145cd82d2f9c5
+  data.tar.gz: 3bc2960d5ffc40813b9ceb4aa68de1b18f6087f67ad5292aeea1900f3c26b98db0b7df941ebcc931da914514e36fbf7f01017ca639597a63d33e2269a2a90af5

data/lib/sncf/api_client.rb

@@ -42,13 +42,23 @@ module Sncf
       end
     end
 
+    def fetch_stations(place)
+      raise ArgumentError, "The fetch_place argument should be a string, example: 'Paris Nord'." if place.to_s == ''
+
+      stations_response = fetch('coverage/sncf/places', {
+        q: place
+      })
+
+      Sncf::Parsers::Stations.new(stations_response).get_stations_list
+    end
+
     protected
 
     def construct_formated_url(path, additional_params)
       if additional_params.nil?
         "#{BASE_URL}/#{VERSION_PATH}/#{path}"
       else
-        "#{BASE_URL}/#{VERSION_PATH}/#{path}?#{additional_params.to_query}"
+        "#{BASE_URL}/#{VERSION_PATH}/#{path}?#{URI.encode_www_form(additional_params)}"
       end
     end
   end

data/lib/sncf/models/generator.rb

@@ -0,0 +1,36 @@
+module Sncf
+  module Models
+    MODEL_NAMES = %w[Station AdministrativeRegion]
+    MODEL_ATTRIBUTES = {
+      'Station' => [:id, :coord, :quality, :name, :label, :timezone, :administrative_regions],
+      'AdministrativeRegion' => [:insee, :level, :coord, :name, :label, :id, :zip_code]
+    }
+
+    def self.generate_models
+      Sncf::Models::MODEL_NAMES.each do |klass_name|
+        klass_vars  = Sncf::Models::MODEL_ATTRIBUTES[klass_name.to_s]
+        klass       = generate_class klass_vars
+
+        const_set klass_name, klass
+      end
+    end
+
+    def self.generate_class(klass_vars)
+      Class.new do
+        klass_vars.each do |field|
+          define_method field.intern do
+            instance_variable_get("@#{field}")
+          end
+          define_method "#{field}=".intern do |arg|
+            instance_variable_set("@#{field}", arg)
+          end
+        end
+        define_method :initialize do |args|
+          klass_vars.each do |field|
+            instance_variable_set("@#{field}", args[field])
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sncf/parsers/default.rb

@@ -0,0 +1,15 @@
+module Sncf
+  module Parsers
+    class Default
+      def initialize(api_response)
+        @api_response = api_response
+      end
+
+      protected
+
+      def create_model(model_name, attributes)
+        Object.const_get("Sncf::Models::#{model_name}").new attributes
+      end
+    end
+  end
+end

data/lib/sncf/parsers/stations.rb

@@ -0,0 +1,37 @@
+module Sncf
+  module Parsers
+    class Stations < Sncf::Parsers::Default
+      def get_stations_list
+        stations, administrative_regions = [], []
+
+        @api_response.content['places'].each do |place|
+          if place['embedded_type'] == 'stop_area'
+            place['stop_area']['administrative_regions'].each do |administrative_region|
+              administrative_regions << create_model('AdministrativeRegion', {
+                id: administrative_region['id'],
+                insee: administrative_region['insee'],
+                level: administrative_region['level'],
+                coord: administrative_region['coord'],
+                name: administrative_region['name'],
+                label: administrative_region['label'],
+                zip_code: administrative_region['zip_code']
+              })
+            end
+
+            stations << create_model('Station', {
+              id: place['id'],
+              coord: place['stop_area']['coord'],
+              quality: place['quality'],
+              name: place['stop_area']['name'],
+              label: place['stop_area']['label'],
+              timezone: place['stop_area']['timezone'],
+              administrative_regions: administrative_regions
+            })
+          end
+        end
+
+        stations
+      end
+    end
+  end
+end

data/lib/sncf/version.rb

@@ -1,3 +1,3 @@
 module Sncf
-  VERSION = "0.1.1"
+  VERSION = "0.2.2"
 end

data/lib/sncf.rb

@@ -1,6 +1,11 @@
 require 'sncf/version'
 require 'sncf/api_client'
 require 'sncf/api_response'
+require "sncf/models/generator"
+require "sncf/parsers/default"
+require "sncf/parsers/stations"
 
 module Sncf
 end
+
+Sncf::Models.generate_models

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sncf
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Rémi Delhaye
@@ -155,6 +155,9 @@ files:
 - lib/sncf.rb
 - lib/sncf/api_client.rb
 - lib/sncf/api_response.rb
+- lib/sncf/models/generator.rb
+- lib/sncf/parsers/default.rb
+- lib/sncf/parsers/stations.rb
 - lib/sncf/version.rb
 - sncf.gemspec
 homepage: https://github.com/rdlh/sncf