Untitled

 avatar
unknown
plain_text
2 years ago
1.1 kB
5
Indexable

import tkinter as tk
import requests
import pandas as pd
import os

# Download the stock data
def download_data():
    url = "https://finance.yahoo.com/quote/SQQQ/history?p=SQQQ"
    filename = "SQQQ_raw_data.csv"
    path = "C:\\Users\\Public\\Test_machine\\Data\\"
    if not os.path.exists(path):
        os.makedirs(path)
    response = requests.get(url)
    with open(os.path.join(path, filename), "wb") as file:
        file.write(response.content)

# Pre-process the data
def preprocess_data():
    path = "C:\\Users\\Public\\Test_machine\\Data\\"
    filename = "SQQQ_raw_data.csv"
    data = pd.read_csv(os.path.join(path, filename))
    data = data[["Date", "Close"]]
    preprocessed_filename = "SQQQ_stock_data.csv"
    data.to_csv(os.path.join(path, preprocessed_filename), index=False)

root = tk.Tk()
download_data_button = tk.Button(root, text="Download Data", command=download_data)
download_data_button.pack()
preprocess_data_button = tk.Button(root, text="Pre-process Data", command=preprocess_data)
preprocess_data_button.pack()
root.mainloop()

Editor is loading...