Add exchange and currency fields to trade imports

app/controllers/import/configurations_controller.rb

@@ -29,6 +29,7 @@ def import_params
         :account_col_label,
         :qty_col_label,
         :ticker_col_label,
+        :exchange_col_label,
         :price_col_label,
         :entity_type_col_label,
         :notes_col_label,

app/controllers/import/rows_controller.rb

@@ -14,7 +14,7 @@ def show
 
   private
     def row_params
-      params.require(:import_row).permit(:type, :account, :date, :qty, :ticker, :price, :amount, :currency, :name, :category, :tags, :entity_type, :notes)
+      params.require(:import_row).permit(:type, :account, :date, :qty, :ticker, :exchange, :price, :amount, :currency, :name, :category, :tags, :entity_type, :notes)
     end
 
     def set_import_row

app/helpers/imports_helper.rb

@@ -20,6 +20,7 @@ def import_col_label(key)
       notes: "Notes",
       qty: "Quantity",
       ticker: "Ticker",
+      exchange: "Exchange",
       price: "Price",
       entity_type: "Type"
     }[key]

app/models/import.rb

@@ -73,6 +73,7 @@ def generate_rows_from_csv
         date: row[date_col_label].to_s,
         qty: sanitize_number(row[qty_col_label]).to_s,
         ticker: row[ticker_col_label].to_s,
+        exchange: row[exchange_col_label]&.strip.to_s,
         price: sanitize_number(row[price_col_label]).to_s,
         amount: sanitize_number(row[amount_col_label]).to_s,
         currency: (row[currency_col_label] || default_currency).to_s,

app/models/import/row.rb

@@ -3,6 +3,7 @@ class Import::Row < ApplicationRecord
 
   validates :amount, numericality: true, allow_blank: true
   validates :currency, presence: true
+  validates :exchange, presence: true, if: -> { import.type == "TradeImport" && import.exchange_col_label.present? }
 
   validate :date_valid
   validate :required_columns

app/models/trade_import.rb

@@ -5,14 +5,24 @@ def import!
 
       rows.each do |row|
         account = mappings.accounts.mappable_for(row.account)
-        security = Security.find_or_create_by(ticker: row.ticker)
+        security = Security.find_or_create_by!(
+          ticker: row.ticker,
+          exchange_mic: row.exchange.presence || "UNKNOWN",
+          exchange_acronym: row.exchange.presence || "Unknown",
+          currency: row.currency.presence || account.currency
+        )
 
         entry = account.entries.build \
           date: row.date_iso,
           amount: row.signed_amount,
           name: row.name,
-          currency: row.currency,
-          entryable: Account::Trade.new(security: security, qty: row.qty, currency: row.currency, price: row.price),
+          currency: row.currency.presence || account.currency,
+          entryable: Account::Trade.new(
+            security: security,
+            qty: row.qty,
+            currency: row.currency.presence || account.currency,
+            price: row.price
+          ),
           import: self
 
         entry.save!
@@ -29,7 +39,7 @@ def required_column_keys
   end
 
   def column_keys
-    %i[date ticker qty price currency account name]
+    %i[date ticker exchange currency qty price account name]
   end
 
   def dry_run
@@ -41,10 +51,10 @@ def dry_run
 
   def csv_template
     template = <<-CSV
-      date*,ticker*,qty*,price*,currency,account,name
-      05/15/2024,AAPL,10,150.00,USD,Trading Account,Apple Inc. Purchase
-      05/16/2024,GOOGL,-5,2500.00,USD,Investment Account,Alphabet Inc. Sale
-      05/17/2024,TSLA,2,700.50,USD,Retirement Account,Tesla Inc. Purchase
+      date*,ticker*,exchange,currency,qty*,price*,account,name
+      05/15/2024,AAPL,XNAS,USD,10,150.00,Trading Account,Apple Inc. Purchase
+      05/16/2024,GOOGL,XNAS,USD,-5,2500.00,Investment Account,Alphabet Inc. Sale
+      05/17/2024,TSLA,XNAS,USD,2,700.50,Retirement Account,Tesla Inc. Purchase
     CSV
 
     CSV.parse(template, headers: true)

app/views/import/configurations/_trade_import.html.erb

@@ -12,6 +12,8 @@
   </div>
 
   <%= form.select :ticker_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Ticker" } %>
+  <%= form.select :exchange_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Exchange" } %>
+  <%= form.select :currency_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Currency" } %>
   <%= form.select :price_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Price" } %>
   <%= form.select :account_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Account (optional)" } %>
   <%= form.select :name_col_label, import.csv_headers, { include_blank: "Leave empty", label: "Name (optional)" } %>

db/migrate/20250207061453_add_exchange_and_currency_columns_to_imports.rb

@@ -0,0 +1,5 @@
+class AddExchangeAndCurrencyColumnsToImports < ActiveRecord::Migration[7.2]
+  def change
+    add_column :imports, :exchange_col_label, :string
+  end
+end

db/migrate/20250207061552_add_exchange_and_currency_to_securities.rb

@@ -0,0 +1,6 @@
+class AddExchangeAndCurrencyToSecurities < ActiveRecord::Migration[7.2]
+  def change
+    add_column :securities, :currency, :string
+    add_index :securities, :currency
+  end
+end

db/migrate/20250207062248_add_exchange_to_import_rows.rb

@@ -0,0 +1,5 @@
+class AddExchangeToImportRows < ActiveRecord::Migration[7.2]
+  def change
+    add_column :import_rows, :exchange, :string
+  end
+end