-
Notifications
You must be signed in to change notification settings - Fork 2
/
bot-borsa.py
69 lines (55 loc) · 1.95 KB
/
bot-borsa.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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import random
import os
from telegram.ext import Updater, CommandHandler
from iexfinance.stocks import Stock, get_historical_intraday
import matplotlib
import matplotlib.pyplot as plt
def start(update, context):
context.bot.send_message(
chat_id=update.effective_chat.id,
text="Hola! Usa les comandes /preus i /grafica")
def preus(update, context):
try:
for simbol in context.args:
quote = Stock(simbol)
preu = quote.get_price()
nom = quote.get_company()['companyName']
missatge = "%s %s %s\n" % (simbol, nom, preu)
context.bot.send_photo(
chat_id=update.effective_chat.id,
photo=quote.get_logo()['url'])
context.bot.send_message(
chat_id=update.effective_chat.id,
text=missatge)
except Exception as e:
print(e)
context.bot.send_message(
chat_id=update.effective_chat.id,
text='💣')
def grafica(update, context):
try:
fitxer = "%d.png" % random.randint(1000000, 9999999)
dataframe = get_historical_intraday(
context.args[0],
output_format='pandas')
dataframe = dataframe.filter(items=['close'])
dataframe.plot()
plt.savefig(fitxer, bbox_inches='tight')
context.bot.send_photo(
chat_id=update.effective_chat.id,
photo=open(fitxer, 'rb'))
os.remove(fitxer)
except Exception as e:
print(e)
context.bot.send_message(
chat_id=update.effective_chat.id,
text='💣')
# hack pel Mac
matplotlib.use('Agg')
TOKEN = open('token.txt').read().strip()
updater = Updater(token=TOKEN, use_context=True)
dispatcher = updater.dispatcher
dispatcher.add_handler(CommandHandler('start', start))
dispatcher.add_handler(CommandHandler('preus', preus))
dispatcher.add_handler(CommandHandler('grafica', grafica))
updater.start_polling()