72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
from datetime import datetime, timedelta
|
|
import orjson
|
|
import time
|
|
import sqlite3
|
|
import asyncio
|
|
import aiohttp
|
|
import random
|
|
from tqdm import tqdm
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
load_dotenv()
|
|
api_key = os.getenv('FMP_API_KEY')
|
|
|
|
async def save_json(data):
|
|
with open(f"json/sentiment-tracker/data.json", 'wb') as file:
|
|
file.write(orjson.dumps(data))
|
|
|
|
|
|
async def get_data(session, total_symbols):
|
|
sources = ["twitter", "stocktwits"] # Sources to loop through
|
|
result_data = {} # Dictionary to store results from both sources
|
|
|
|
for source in sources:
|
|
# Construct the API URL with the source parameter
|
|
url = f"https://financialmodelingprep.com/api/v4/social-sentiments/trending?type=bullish&source={source}&apikey={api_key}"
|
|
|
|
try:
|
|
async with session.get(url) as response:
|
|
if response.status == 200:
|
|
data = await response.json()
|
|
if len(data) > 0:
|
|
res_list = []
|
|
for item in data:
|
|
symbol = item['symbol']
|
|
item['sentiment'] = round(item['sentiment']*100, 2)
|
|
item['lastSentiment'] = round(item['lastSentiment']*100, 2)
|
|
if symbol in total_symbols:
|
|
try:
|
|
with open(f"json/quote/{symbol}.json", 'r') as file:
|
|
res = orjson.loads(file.read())
|
|
item['price'] = round(res['price'], 2)
|
|
item['changesPercentage'] = round(res['changesPercentage'], 2)
|
|
item['marketCap'] = round(res['marketCap'], 2)
|
|
res_list.append({**item})
|
|
except:
|
|
pass
|
|
result_data[source] = res_list # Store the result list for the current source
|
|
except Exception as e:
|
|
print(f"Error fetching data from {source}: {e}")
|
|
pass
|
|
|
|
# Save the combined result as a single JSON file
|
|
if result_data:
|
|
await save_json(result_data)
|
|
|
|
|
|
|
|
async def run():
|
|
con = sqlite3.connect('stocks.db')
|
|
cursor = con.cursor()
|
|
cursor.execute("PRAGMA journal_mode = wal")
|
|
cursor.execute("SELECT DISTINCT symbol FROM stocks WHERE symbol NOT LIKE '%.%'")
|
|
total_symbols = [row[0] for row in cursor.fetchall()]
|
|
con.close()
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
await get_data(session, total_symbols)
|
|
|
|
if __name__ == "__main__":
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(run()) |