stockcruncher 1.3.2 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 56cbc10941fcfdc6ccee0273cf5b2f43f4a98c1f3b53b1554a53ea45020c604e
-  data.tar.gz: ce451b4498697f9c6148167771296219c4f83ffdef3f1fbefb4c8a12f7a747e4
+  metadata.gz: d575ec081ba8f154b940a03a7eda7d79606bc78ff9c01132f2da78170b641367
+  data.tar.gz: 5949b3b5108d9d0d67907e49711ae9ecf8dbdfebfba669b9ed952f8592cfa95a
 SHA512:
-  metadata.gz: 5b3bcfdbe1805e95c4729e33ad26b4ea676d63a5e638150fca0a26d75682a96d6d4d71a16e9458ef75423f82fc4ba0a4f7442e838865ac9d27b57fbb817dbaed
-  data.tar.gz: fbe77bb52c0b0d51d6f480246c9467ca4838f9afd385e573c35342b40ada8973f0f5f52c65d159ae85f332d8e0a38e244cf5149b386dc10872323d5990726451
+  metadata.gz: 6f3dbd7ade6b0ad21e39842111ba21aa21ddf38ac0d984a24cf786a9bcd503c5427b85c5e5c500a0acc278fda9e83a08472144117148a8f624c49c724269884c
+  data.tar.gz: 98690a9871f4d68c2a199cdbcce134690a12b594aba5c4842f4c36346dcd143b8ea15079ff69e7996af19958a1feec5859340e0ea6657d3c649c24cc66007832

data/lib/stockcruncher/cli.rb

@@ -45,12 +45,20 @@ module StockCruncher
       default: false,
       desc: 'Full data size.'
     )
+    option(
+      :catchup,
+      aliases: ['-u'],
+      type: :boolean,
+      default: false,
+      desc: 'Catch up the missing data only.'
+    )
     def daily(symbol)
       opts = options.dup
       config = YAML.load_file(opts['config'])
       cruncher = StockCruncher::AlphaVantage.new(config, opts['insecure'])
       data = cruncher.crunch_daily(symbol, opts['full'])
-      StockCruncher::InfluxDB.new(config).export_history(symbol, data)
+      influx = StockCruncher::InfluxDB.new(config)
+      influx.export_history(symbol, data, opts['catchup'])
       puts JSON.pretty_generate(data) unless opts['quiet']
     end
 
@@ -63,10 +71,18 @@ module StockCruncher
       default: false,
       desc: 'Recalculate all MA historical values.'
     )
+    option(
+      :catchup,
+      aliases: ['-u'],
+      type: :boolean,
+      default: false,
+      desc: 'Catch up the missing data only.'
+    )
     def movingaverages(symbol)
       opts = options.dup
       config = YAML.load_file(opts['config'])
-      StockCruncher::InfluxDB.new(config).moving_averages(symbol, opts['all'])
+      influx = StockCruncher::InfluxDB.new(config)
+      influx.moving_averages(symbol, opts['all'], opts['catchup'])
     end
 
     desc('quote SYMBOL [options]',
@@ -76,7 +92,8 @@ module StockCruncher
       config = StockCruncher::Config.load(opts['config'])
       cruncher = StockCruncher::AlphaVantage.new(config, opts['insecure'])
       data = cruncher.crunch_quote(symbol)
-      StockCruncher::InfluxDB.new(config).export_last_day(data)
+      influx = StockCruncher::InfluxDB.new(config)
+      influx.export_last_day(data)
       puts JSON.pretty_generate(data) unless opts['quiet']
     end
   end

data/lib/stockcruncher/influxdb.rb

@@ -20,15 +20,22 @@ module StockCruncher
       data['columns'].zip(data['values'].transpose).to_h
     end
 
+    def get_ma_values(symbol, fullsize)
+      values = %w[ema200]
+      data = query('ema', symbol, values, fullsize)
+      data['columns'].zip(data['values'].transpose).to_h
+    end
+
     # Method to calculate moving averages based on last day values
-    def moving_averages(symbol, fullsize)
+    def moving_averages(symbol, fullsize, catchup)
       series = get_daily_values(symbol, fullsize)
+      mas = catchup ? get_ma_values(symbol, fullsize) : {}
       tags = create_tags(symbol)
       series['close'].each_index do |i|
         date = series['time'][i]
         serie = series['close'][i, 201]
         weights = series['volume'][i, 201]
-        write_moving_averages(tags, serie, weights, date)
+        write_moving_averages(tags, serie, weights, date) unless mas.key? date
         break unless fullsize
       end
     end
@@ -39,8 +46,12 @@ module StockCruncher
     end
 
     # Method to export historical data to database
-    def export_history(symbol, timeseries)
+    def export_history(symbol, timeseries, catchup)
       tags = create_tags(symbol)
+      if catchup
+        series = get_daily_values(symbol, fullsize)
+        series['time'].each { |date| timeseries.delete(date) }
+      end
       timeseries.each_pair do |date, values|
         write('daily', tags, values, date)
       end

data/lib/stockcruncher/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module StockCruncher
-  VERSION = '1.3.2'
+  VERSION = '1.4.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stockcruncher
 version: !ruby/object:Gem::Version
-  version: 1.3.2
+  version: 1.4.0
 platform: ruby
 authors:
 - Richard Delaplace
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-07-20 00:00:00.000000000 Z
+date: 2021-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler