regex

[erebus.git] / modules / coins.py

diff --git a/modules/coins.py b/modules/coins.py
index e4338e683ae855b462ed7a94af513cbb82bd5807..5a92fad90e8ecb23bbd4476c87c79582aeda4b41 100644 (file)
--- a/modules/coins.py
+++ b/modules/coins.py
@@ -17,9 +17,18 @@ modstart = lib.modstart
 modstop = lib.modstop
 
 # module code
+import re
 import json
 import requests
 
+coin_regex = (
+       re.compile(r'([0-9.,\s]+)\s(btc|bitcoin|doge|dogecoin|ltc|litecoin)'),
+)
+
+cur_regex = (
+       re.compile(r'([0-9.,\s]+)\s([a-zA-Z]{3})\sin\s([a-zA-Z]{3})'),
+)
+
 url = 'http://www.cryptocoincharts.info/v2/api/tradingPairs'
 
 def get_coin_price(pairs):
@@ -71,3 +80,54 @@ def cmd_gtest(bot, user, chan, realtarget, *args):
                btc_eur = get_coin_price('btc_eur')
                price = str(float(ltc_btc[0]['price']) * float(btc_eur[0]['price']))
                bot.msg(chan, "1 LTC = %s EUR" % price)
+
+@lib.hooknum("PRIVMSG")
+def privmsg_hook(bot, line):
+
+       try:
+               linetx = line.split(None, 3)[3][1:]
+       except IndexError:
+               linetx = ''
+
+       chan = line.split()[2]
+
+       if 'in' in line:
+               for r in cur_regex:
+                       for a, f, t in r.findall(linetx):
+
+                               # https://www.google.com/finance/converter?a=1.2&from=USD&to=EUR
+
+                               a = a.replace(",", ".")
+                               a = a.replace(" ", "")
+                               print a
+                               print f
+                               print t
+
+       if 'btc' in line or 'bitcoin' in line or 'doge' in line or 'dogecoin' in line:
+               for r in coin_regex:
+                       for amount, coin in r.findall(linetx):
+                               amount = amount.replace(",", ".")
+                               amount = amount.replace(" ", "")
+                               if 'btc' in coin or 'bitcoin' in coin:
+                                       try:
+                                               response = get_coin_price('btc_eur')
+                                               price = str(float(response[0]['price']) * float(amount))
+                                               bot.msg(chan, "%s BTC = %s EUR" % (amount, price))
+                                       except:
+                                               pass
+
+                               if 'ltc' in coin or 'litecoin' in coin:
+                                       pass    # Add LTC
+
+                               if 'doge' in coin or 'dogecoin' in coin:
+                                       try:
+                                               doge_btc = get_coin_price('doge_btc')
+                                               btc_eur = get_coin_price('btc_eur')
+                                               price = str(float(doge_btc[0]['price']) * float(btc_eur[0]['price']) * float(amount))
+                                               bot.msg(chan, "%s DOGE = %s EUR" % (amount, price))
+                                       except:
+                                               bot.msg(chan, "Invalid amount.")
+                                               
+                                       
+                               print amount
+                               print coin