From 6e059c30eff8a66e568a2c63f81bcae8ff7d9f7a Mon Sep 17 00:00:00 2001
From: clothespin <ian@unstu.cc>
Date: Sat, 6 Mar 2021 16:09:51 -0800
Subject: [PATCH] Adds export command

---
 terminal.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/terminal.py b/terminal.py
index 2b48a773121..af2dfcd34cc 100644
--- a/terminal.py
+++ b/terminal.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 
 import argparse
+from sys import stdout
 import pandas as pd
 from alpha_vantage.timeseries import TimeSeries
 
@@ -225,6 +226,37 @@ def view(l_args, s_ticker, s_start, s_interval, df_stock):
     plot_view_stock(df_stock.iloc[:, ln_col_idx], ns_parser.s_ticker)
 
 
+# ----------------------------------------------------- EXPORT -----------------------------------------------------
+def export(l_args, df_stock):
+    parser = argparse.ArgumentParser(prog='export', description='Exports the current dataframe to a file or stdout')
+    parser.add_argument('-f', "--filename", type=str, dest="s_filename", default=stdout, help="Where to save the export (stdout if unspecified)")
+    parser.add_argument('-F', "--format", dest="s_format", type=str, default="csv", help="Export historical data into following formats: csv, json, excel, clipboard")
+
+    try:
+        (ns_parser, l_unknown_args) = parser.parse_known_args(l_args)
+        s_format = ns_parser.s_format
+        s_filename = ns_parser.s_filename
+    except SystemExit:
+        print("")
+        return
+
+    if l_unknown_args:
+        print(f"The following args couldn't be interpreted: {l_unknown_args}")
+
+    if df_stock.empty:
+        print("No data loaded yet to export.")
+        return
+
+    if s_format == "csv":
+        df_stock.to_csv(s_filename)
+    elif s_format == "json":
+        df_stock.to_json(s_filename)
+    elif s_format == "excel":
+        df_stock.to_excel(s_filename)
+    elif s_format == "clipboard":
+        df_stock.to_clipboard()
+
+
 # ----------------------------------------------------- HELP ------------------------------------------------------------
 def print_help(s_ticker, s_start, s_interval, b_is_market_open):
     """Print help"""
@@ -235,6 +267,8 @@ def print_help(s_ticker, s_start, s_interval, b_is_market_open):
     print("   clear       clear a specific stock ticker from analysis")
     print("   load        load a specific stock ticker for analysis")
     print("   view        view and load a specific stock ticker for technical analysis")
+    if s_ticker:
+        print("   export      export the currently loaded dataframe to a file or stdout")
 
     s_intraday = (f"Intraday {s_interval}", "Daily")[s_interval == "1440min"]
     if s_ticker and s_start:
@@ -303,6 +337,7 @@ def main():
             "clear",
             "load",
             "view",
+            "export",
             "disc",
             "mill",
             "sen",
@@ -360,6 +395,9 @@ def main():
         elif ns_known_args.opt == "view":
             view(l_args, s_ticker, s_start, s_interval, df_stock)
 
+        elif ns_known_args.opt == "export":
+            export(l_args, df_stock)
+
         elif ns_known_args.opt == "disc":
             b_quit = dm.disc_menu()