main.py

import os
import glob
import shutil
import datafunctions as df
from tkinter import filedialog
import concurrent.futures

# from colorama import Fore, Style

# ------------------------------------ Setup

timeStart = df.get_time("Starting Date (dd/mm/yyyy): ")
while True:
    timeEnd = df.get_time("Ending Date (dd/mm/yyyy): ")
    if timeEnd > timeStart:
        break
    print("Ending date must be after starting date. Please try again.")

print("\nSelect tab-delimited text file with tickers")

ticker_filename = filedialog.askopenfilename(filetypes=[("Text files", "*.txt")])
with open(ticker_filename, "r") as file:
    tickerList = [line.strip().upper() for line in file if line.strip() != ""]

print("\nGenerating URLs for " + str(len(tickerList)) + " tickers.\n")

# Multithreaded data fetching
def fetch_ticker_data(ticker):
    print(f"Fetching data for ticker: {ticker}")
    df.fetchData(ticker, timeStart, timeEnd)

with concurrent.futures.ThreadPoolExecutor() as executor:
    executor.map(fetch_ticker_data, tickerList)

# ------------------------------------ Merge Data

# Get the csv files in the /temp directory
csv_files = glob.glob("./temp/*.csv")

combinedDataframe = df.mergeData(csv_files)

# Validate dateOrder input
while True:
    dateOrder = input("\nEnter 'a' or 'd' to choose ascending or descending order: ")
    if dateOrder.lower() in ['a', 'd']:
        break
    print("Invalid input. Please enter 'a' for ascending or 'd' for descending order.")

# Validate closeType input
while True:
    closeType = input("Enter 'a' or 'c' to choose adjusted close or close: ")
    if closeType.lower() in ['a', 'c']:
        break
    print("Invalid input. Please enter 'a' for adjusted close or 'c' for close.")

# print("\n" + Fore.BLUE + "Merging" + Style.RESET_ALL + " .csv files...")
print("\n" + "Merging" + " .csv files...")

df.pivotData(combinedDataframe, closeType, ticker_filename, dateOrder)

if os.path.exists("./temp"):
    shutil.rmtree("./temp")

# print("\n" + Fore.GREEN + "Finished!" + Style.RESET_ALL + " Output saved to ./out.csv \n")
print("\n" + "Finished!")