Skip to content

Commit

Permalink
Moved HODL to its own lego per review in #17
Browse files Browse the repository at this point in the history
  • Loading branch information
Drew Pearce committed Jan 21, 2018
1 parent 1c26870 commit aaec5e0
Show file tree
Hide file tree
Showing 3 changed files with 204 additions and 158 deletions.
185 changes: 27 additions & 158 deletions legos/cryptocurrency.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,28 +2,15 @@
import requests
import logging
import json
import configparser

logger = logging.getLogger(__name__)


class Cryptocurrency(Lego):
def __init__(self, baseplate, lock):
super().__init__(baseplate, lock)
self.crypto_index = configparser.ConfigParser()
self.crypto_index.read('crypto_index.ini')
if 'crypto_index' not in self.crypto_index.sections():
self.crypto_index['crypto_index'] = {}
self.crypto_index['crypto_index']['fsyms'] = 'BTC'
self.crypto_index['crypto_index']['tsyms'] = 'USD,BTC'
with open('crypto_index.ini', 'w') as configfile:
self.crypto_index.write(configfile)

def listening_for(self, message):
if message['text'] is not None:
try:
command = message['text'].split()[0]
return command == '!crypto' or command == '!hodl'
return message['text'].split()[0] == '!crypto'
except Exception as e:
logger.error('''Stocks lego failed to check message text:
{}'''.format(e))
Expand All @@ -37,125 +24,12 @@ def handle(self, message):
logger.error('''Could not identify message source in message:
{}'''.format(message))

message_list = message['text'].split()
if message_list[0] == '!hodl':
self.reply(message, self._parse_multi_commands(message_list), opts)
else:
try:
query = message['text'].split()[1]
except:
self.reply(message, "Invalid query", opts)
self.reply(message, self._lookup_symbol(query), opts)

def _parse_multi_commands(self, message_list):
if len(message_list) == 1:
return self._lookup_multi()
elif len(message_list) > 1:
case_dict = {}
case_dict['list'] = self._list_hodl_symbols
case_dict['add'] = self._add_symbols
case_dict['drop'] = self._drop_symbols
return case_dict[message_list[1]](message_list)

def _list_hodl_symbols(self, message_list):
fsyms = self.crypto_index['crypto_index']['fsyms']
tsyms = self.crypto_index['crypto_index']['tsyms']
return_val = ''
return_val += 'Convert From: ' + fsyms + '\n'
return_val += 'Convert To: ' + tsyms + '\n'
return return_val

def _add_symbols(self, message_list):
if len(message_list) < 3:
return ('Please supply additional arguments, e.g.:\n'
'!hodl add from BTC,LTC')
elif message_list[2] == 'from':
return self._add_from_symbols(message_list)
elif message_list[2] == 'to':
return self._add_to_symbols(message_list)
else:
return 'There was an issue processing your request.'

def _add_from_symbols(self, message_list):
try:
add_fsyms = message_list[3].split(',')
old_fsyms = self.crypto_index['crypto_index']['fsyms'].split(',')
new_fsyms = list(set(old_fsyms + add_fsyms))
self.crypto_index['crypto_index']['fsyms'] = ','.join(new_fsyms)
with open('crypto_index.ini', 'w') as configfile:
self.crypto_index.write(configfile)
return self._list_hodl_symbols(message_list)
except:
return ('Please supply additional arguments, e.g.:\n'
'!hodl add from BTC,LTC')

def _add_to_symbols(self, message_list):
try:
add_tsyms = message_list[3].split(',')
old_tsyms = self.crypto_index['crypto_index']['tsyms'].split(',')
new_tsyms = list(set(old_tsyms + add_tsyms))
self.crypto_index['crypto_index']['tsyms'] = ','.join(new_tsyms)
with open('crypto_index.ini', 'w') as configfile:
self.crypto_index.write(configfile)
return self._list_hodl_symbols(message_list)
except:
return ('Please supply additional arguments, e.g.:\n'
'!hodl add to BTC,LTC')

def _drop_symbols(self, message_list):
if len(message_list) < 3:
return ('Please supply additional arguments, e.g.:\n'
'!hodl drop from BTC')
elif message_list[2] == 'from':
return self._drop_from_symbols(message_list)
elif message_list[2] == 'to':
return self._drop_to_symbols(message_list)
else:
return 'There was an issue processing your request.'

def _drop_from_symbols(self, message_list):
try:
drop_fsyms = message_list[3].split(',')
old_fsyms = self.crypto_index['crypto_index']['fsyms'].split(',')
new_fsyms = set(old_fsyms).symmetric_difference(set(drop_fsyms))
self.crypto_index['crypto_index']['fsyms'] = ','.join(new_fsyms)
with open('crypto_index.ini', 'w') as configfile:
self.crypto_index.write(configfile)
return self._list_hodl_symbols(message_list)
except:
return ('Please supply additional arguments, e.g.:\n'
'!hodl drop from BTC,LTC')

def _drop_to_symbols(self, message_list):
try:
drop_tsyms = message_list[3].split(',')
old_tsyms = self.crypto_index['crypto_index']['tsyms'].split(',')
new_tsyms = set(old_tsyms).symmetric_difference(set(drop_tsyms))
self.crypto_index['crypto_index']['tsyms'] = ','.join(new_tsyms)
with open('crypto_index.ini', 'w') as configfile:
self.crypto_index.write(configfile)
return self._list_hodl_symbols(message_list)
query = message['text'].split()[1]
except:
return ('Please supply additional arguments, e.g.:\n'
'!hodl drop to BTC,LTC')
self.reply(message, "Invalid query", opts)

def _lookup_multi(self):
multi_url = self._build_index_url()
get_multi = requests.get(multi_url)
if get_multi.status_code == requests.codes.ok:
api_response = json.loads(get_multi.text)
return self._parse_multi_price_response(api_response)
else:
return 'There was an error getting the data: ' + get_multi.text

def _parse_multi_price_response(self, api_response):
return_val = ''
for key, value in api_response.items():
return_val += key + ': | '
for key, value in value.items():
return_val += '{} {} | '.format(value, key)
return_val += '\n'
return return_val
self.reply(message, self._lookup_symbol(query), opts)

def _lookup_symbol(self, query):
query = query.upper()
Expand All @@ -166,21 +40,26 @@ def _lookup_symbol(self, query):
params['tsyms'] = 'USD' # tsyms, the SYMbolS to convert To
else:
params['tsyms'] = 'USD,BTC' # tsyms, the SYMbolS to convert To
request_url = self._build_index_url()
api_response = requests.get(request_url, params=params)
if api_response.status_code == requests.codes.ok:
if 'There is no data for the symbol' in api_response['Message']:
api_response = json.loads(api_response.text)
if 'Message' in api_response and \
'There is no data for the symbol' in api_response['Message']:
matched_items = self._search_symbol(query)
params['fsym'] = matched_items[0]['symbol']
api_response = requests.get(request_url, params=params)
if api_response.status_code == requests.codes.ok:
api_response = json.loads(api_response.text)
query = matched_items[0]['symbol']
meta = 'Did you mean {}?'.format(matched_items[0]['name'])
return self._parse_api_response(
api_response, query, meta=meta)
if len(matched_items) > 0:
params['fsym'] = matched_items[0]['symbol']
api_response = requests.get(request_url, params=params)
if api_response.status_code == requests.codes.ok:
api_response = json.loads(api_response.text)
query = matched_items[0]['symbol']
meta = 'Did you mean {}?'.format(
matched_items[0]['name'])
return self._parse_api_response(
api_response, query, meta=meta)
else:
return 'We had trouble getting that ticker price.'
else:
return 'We had trouble getting that ticker price.'
return 'Could not match "{}" to a symbol.'.format(query)
else:
return self._parse_api_response(api_response, query)
else:
Expand All @@ -197,7 +76,8 @@ def _search_symbol(self, query):
matched_items = []
for coin in api_list['Data']:
full_name = api_list['Data'][coin]['FullName']
if query in full_name.lower():
logger.debug('FULL NAME: ' + full_name)
if query.lower() in full_name.lower():
matched_items.append({"symbol": coin, "name": full_name})

return matched_items
Expand All @@ -206,31 +86,20 @@ def _search_symbol(self, query):
return 'There was an error fetching the list'

def _parse_api_response(self, api_response, query, **kwargs):
return_val = ''
if 'meta' in kwargs:
return_val = kwargs['meta'] + '\n'
return_val = return_val + query + ': | '
return_val += kwargs['meta'] + '\n'
return_val += query + ': | '
for key, value in api_response.items():
return_val += '{} {} | '.format(value, key)
if query == 'DOGE':
return_val = 'WOW! {} TO THE MOON!!!'.format(return_val)
return return_val

def _build_index_url(self):
baseurl = 'https://min-api.cryptocompare.com/data/pricemulti'
fsyms = self.crypto_index['crypto_index']['fsyms']
tsyms = self.crypto_index['crypto_index']['tsyms']
return baseurl + '?fsyms=' + fsyms + '&tsyms=' + tsyms

def get_name(self):
return 'crypto'

def get_help(self):
return ('Lookup a crypto symbol\'s value. Usage: !crypto <symbol>.\n'
'List of symbols here: '
'https://min-api.cryptocompare.com/data/all/coinlist.\n'
'Usage for !hodl:\n'
'!hodl -- returns prices for multiple coins at once.\n'
'!hodl list -- returns the list of coins to convert from/to.\n'
'!hodl <add/drop> <to/from> <comma separated symbols list>'
' -- adds or drop list of symbols to the convert to or convert'
' from list.')
'List of symbols here:\n'
'https://min-api.cryptocompare.com/data/all/coinlist.')
Loading

0 comments on commit aaec5e0

Please sign in to comment.