find_buys_yf.py

import csv
import algo_yf as algo
import uuid
import pickle
from email_sender import send_email
from tqdm import tqdm


def read_exchange(file):
    with open(file) as f:
        reader = csv.reader(f)
        symbols = []
        for line in reader: 
            symbols.append(line[0])
        f.close()
    return list(sorted(symbols[1:]))



def main():
    name = input('Name: ')
    email = input('Email to send to: ')
    stock_price = int(input('Stock Price Under: '))
    multithread = bool(input('Multithread (True/False): '))
    exchanges = ['./exchanges/nasdaq.csv', './exchanges/nyse.csv']
    all_symbols = []
    print('Reading Exchange Data')
    for exchange in exchanges:
        symbols = read_exchange(exchange)
        all_symbols += symbols
    all_symbols = list(sorted(all_symbols))
    print('Using {0} Symbols'.format(len(all_symbols)))

    buys = []
    print('Downloading Stock Data')
    data = algo.download_yf_data(all_symbols, multithread=multithread)
    with open('yahoo-data.p', 'wb') as f:
        pickle.dump(data, f)
        f.close()
    for symbol in tqdm(all_symbols):
        try:
            if (algo.buy_signal(data[symbol], under_value=stock_price)):
                buys.append(symbol)
        except Exception as e:
            continue

    if len(buys) > 0:
        print('Getting Fundamental Data for Buy Signal Stocks...')
        rows = [['Symbol', 'Business Summary', 'Sector', 'Forward EPS', 'Forward P/E', 'P/B']]
        for buy in tqdm(buys):
            buy_info = algo.get_stock_info(buy)
            row = [buy] + buy_info
            rows.append(row)
        new_id = uuid.uuid1()
        file_name = './runs/{0}.csv'.format(new_id)
        with open(file_name, 'w', encoding='utf-8', newline='') as f:
            writer = csv.writer(f)
            writer.writerows(rows)
            f.close()
        print('Sending Email to {}...'.format(email))
        send_email(email, name, buys, file_name, stock_price)
    else:
        print('No buys were found today...')

if __name__ == "__main__":
    main()