osushi-bot/finance.py at master · akazs/osushi-bot · 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
import datetime
from functools import lru_cache

from alpha_vantage.foreignexchange import ForeignExchange
from alpha_vantage.timeseries import TimeSeries

import config
from util import float_format


def retrieve_stock(update, context):
    if len(context.args) < 1:
        context.bot.send_message(chat_id=update.message.chat.id,
                                 text='Usage: /stock <symbol> [YYYY-MM-DD]')
        context.bot.send_message(chat_id=update.message.chat.id,
                                 text='ex: /stock AAPL [2019-12-24]')
        return
    if len(context.args) > 1:
        day = context.args[1]
    else:
        day = datetime.date.today().strftime('%Y-%m-%d')

    ts = TimeSeries(config.finance.api_key)
    try:
        data, meta = ts.get_daily(symbol=context.args[0])
    except ValueError:
        context.bot.send_message(chat_id=update.message.chat.id,
                                text='No such symbol.')
        return

    try:
        result = data[day]
    except KeyError:
        if len(context.args) > 1:
            context.bot.send_message(
                chat_id=update.message.chat.id,
                text='Date invalid, fallback: ' + meta['3. Last Refreshed'])

        # TODO: find previous business day - need to determine country
        result = data[meta['3. Last Refreshed']]

    context.bot.send_message(chat_id=update.message.chat.id,
                             text=str(result))


def foreign_exchange(update, context):
    if len(context.args) < 2:
        context.bot.send_message(chat_id=update.message.chat.id,
                                 text='Usage: /fx <from> <to> [amount]')
        context.bot.send_message(chat_id=update.message.chat.id,
                                 text='ex: /fx BTC JPY 10')
        return

    fx = ForeignExchange(key=config.finance.api_key)
    try:
        data, _ = fx.get_currency_exchange_rate(
            from_currency=context.args[0],
            to_currency=context.args[1])
    except ValueError:
        context.bot.send_message(
            chat_id=update.message.chat.id,
            text='One or both inputted currencies are unsupported.')
        return
    answer = None

    if len(context.args) > 2:
        try:
            amount = float(context.args[2])
            answer = 'You can get {} {} with {} {}.'.format(
                float_format(float(data['5. Exchange Rate']) * amount),
                data['4. To_Currency Name'],
                context.args[2],
                data['2. From_Currency Name'])
        except ValueError:
            pass

    if not answer:
        answer = 'The rate of {} to {} is {}.'.format(
            data['2. From_Currency Name'],
            data['4. To_Currency Name'],
            float_format(data['5. Exchange Rate']))

    context.bot.send_message(chat_id=update.message.chat.id, text=answer)