-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
54 lines (43 loc) · 1.67 KB
/
main.py
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
import requests
import json
import csv
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
sns.set()
def parsing_recommendations(JSON_data):
recommendation_data = pd.read_json(JSON_data)
recommendation_data.to_csv('out.csv', index=True)
df_columns_name = recommendation_data.columns
df_symbol = recommendation_data['symbol'][0]
recommendation_data = recommendation_data.drop(columns=['symbol'])
#print(df_columns_name)
#print(df_symbol)
recommendation_data['period'] = pd.to_datetime(recommendation_data['period'])
#print(recommendation_data['period'].sum())
print(recommendation_data)
plotting_data(recommendation_data)
def get_recommendation(stock):
try:
response = requests.get('https://finnhub.io/api/v1/stock/recommendation?symbol=' + stock + '&token=bsok7avrh5r8ktijv08g')
print('The server has responded with a following status code:', response.status_code)
response = response.json()
return response
except requests.exceptions.RequestException as error:
print('An error has occurred:', error)
def fetching_companies(listOfSymbols):
arrayOfRecommendations = []
for symbol in listOfSymbols:
arrayOfRecommendations += get_recommendation(symbol)
parsedArray = json.dumps(arrayOfRecommendations)
parsing_recommendations(parsedArray)
def plotting_data(df):
print(list(df[:2]))
period_column = df['period']
df = df.drop(columns=['period'])
print(period_column)
plt.bar(period_column, df['buy'], color='maroon')
plt.show()
#listOfStocks = ['TSM', 'BA', 'AMD', 'SNE']
listOfStocks = ['TSM']
fetching_companies(listOfStocks)