~singpolyma/jmp-pay

ref: abb452aa583d96a6317e8fbe3de9204067d2d83d jmp-pay/bin/months_to_balance -rwxr-xr-x 1.3 KiB
abb452aaStephen Paul Weber hotfix 1 year, 6 months ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#!/usr/bin/ruby
# frozen_string_literal: true

# This is meant to be a one-time script that takes any "extra"
# months an account is activated for and rolls them back into balance instead
# Mostly as a fix for all the time that web activation spent not setting balance

require "pg"

DB = PG.connect(dbname: "jmp")
DB.type_map_for_results = PG::BasicTypeMapForResults.new(DB)
DB.type_map_for_queries = PG::BasicTypeMapForQueries.new(DB)

rows = DB.query(<<~SQL)
	SELECT customer_id, expires_at,
		((((DATE_PART('year', expires_at) - DATE_PART('year', LOCALTIMESTAMP)) * 12 +
		(DATE_PART('month', expires_at) - DATE_PART('month', LOCALTIMESTAMP))) - 1) *
			case plan_name
			when 'usd_beta_unlimited-v20210223' then 2.99
			when 'cad_beta_unlimited-v20210223' then 3.59
			end)::NUMERIC(12,4) AS amount
	FROM customer_plans WHERE expires_at > (DATE_TRUNC('month', LOCALTIMESTAMP) + '2 months') ORDER BY expires_at;
SQL

rows.each do |r|
	cid = r["customer_id"]
	DB.transaction do
		DB.exec(<<~SQL, [cid, "#{cid}-months_to_balance", r["amount"]])
			INSERT INTO transactions (customer_id, transaction_id, amount, note)
			VALUES ($1, $2, $3, 'months_to_balance')
		SQL
		DB.exec(<<~SQL, [cid, r["expires_at"]])
			UPDATE plan_log SET date_range = date_range - tsrange(LOCALTIMESTAMP + '1 month', UPPER(date_range))
			WHERE customer_id=$1 AND UPPER(date_range)=$2
		SQL
	end
end