StockPred/loaddata.py at main · sauravshub1/StockPred · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
"""
loaddata.py - Loads historical (and financial) data from investpy

@authors: Pratiksha Jain, Shubham Saurav
"""
# --------------------------------- #

# Imports needed
import time
import numpy as np
import pandas as pd
import investpy
from os import listdir
from os.path import isfile, join
import warnings

warnings.filterwarnings("ignore")

# --------------------------------- #

#### TESTING USAGE ONLY ####
'''
#df = investpy.get_stock_historical_data(stock='HDBK',
                                        country='India',
                                        from_date='01/01/2010',
                                        to_date='01/01/2020')
#print(df.head())


#info = investpy.get_stock_company_profile(stock='HDBK',
                                        country='India', language='english')

#print(info)
#df2 = investpy.get_stock_recent_data(stock='HDBK',
                                        country='India',
                                        interval='Weekly')

print(df2)

data = investpy.get_stock_financial_summary(stock='HDBK',
                                        country='India', summary_type='income_statement', period='annual')
data2 = investpy.get_stock_financial_summary(stock='HDBK',
                                        country='India', summary_type='cash_flow_statement', period='annual')
data3 = investpy.get_stock_financial_summary(stock='HDBK',
                                        country='India', summary_type='balance_sheet', period='quarterly')
print(data)
print(data2)
print(data3)
'''

# --------------------------------- #

# Loads historical data of stock
def loadHistoricalData(stock):
    df = investpy.get_stock_historical_data(stock=stock,
                                        country='India',
                                        from_date='01/01/2010',
                                        to_date='07/05/2021')

    df = df.drop('Currency', axis=1)
    df.to_csv('stock_details/%s.csv'%(stock))

# loads for multiple stocks
def loadStocks(specific_stocks=[], max_stocks=20):
    stocks = investpy.stocks.get_stocks_list(country='India')
    stocks = [i for i in stocks if not i in specific_stocks]
    stocks = stocks[:(max_stocks-len(specific_stocks))]
    stocks_req = stocks+specific_stocks

    for stock in stocks_req:
        loadHistoricalData(stock)
        print(stock, ' loaded')


# checks if stock is loaded, if it isnt (and exists), then loads it
def loadStock(stock):
    stocks = investpy.stocks.get_stocks_list(country='India')
    stocks_loaded = [f for f in listdir('stock_details') if isfile(join('stock_details', f))]

    if not stock in stocks:
        return 0
    elif stock in stocks_loaded:
        return 1
    else:
        loadHistoricalData(stock)
        print(stock, ' loaded')
        return 2

# --------------------------------- #