From cd29b9251912aeec95605eadc85637919640d31d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sergio=20=C3=81lvarez?= <xergio@gmail.com>
Date: Thu, 20 Jul 2017 12:58:54 +0200
Subject: [PATCH] achievements!

---
 guild.py | 51 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 49 insertions(+), 2 deletions(-)

diff --git a/guild.py b/guild.py
index 0e48430..24a6bb2 100755
--- a/guild.py
+++ b/guild.py
@@ -13,11 +13,15 @@ wh = webhook.Webhook(conf.url_discord_webhook)
 
 r = redis.StrictRedis(host='localhost', charset="utf-8", decode_responses=True)
 
+bnet_achs = conf.battle_net_url("https://{0}.api.battle.net/wow/data/guild/achievements?locale=es_ES&apikey={1}")
+
 bnet_guild = conf.battle_net_url("https://{0}.api.battle.net/wow/guild/dun%20modr/vagrant%20story?fields=news,members,achievements&locale=es_ES&apikey={1}")
 
+a = requests.get(url=bnet_achs).json()
 g = requests.get(url=bnet_guild).json()
 
 
+
 if "members" not in g:
 	print("'members' not in g")
 	sys.exit()
@@ -25,7 +29,7 @@ if "members" not in g:
 members = r.smembers("bot:members") # members database, used by other scripts
 chars = set()
 
-# how join the guild, how leaves
+# who join the guild, who leaves
 
 for member in g["members"]:
 	chars.add(member["character"]["name"])
@@ -40,6 +44,7 @@ for kick in members.difference(chars):
 	wh.send(":outbox_tray: **{0}** ha salido a la guild :confused:".format(kick))
 
 
+
 if "news" not in g:
 	print("'news' not in g")
 	sys.exit()
@@ -62,5 +67,47 @@ for news in g["news"]:
 		continue
 
 	wh.send(push)
-	
+
 	time.sleep(2) # prevent rate limit, for example with boss FK
+
+
+
+if "achievements" not in g:
+	print("'achievements' not in g")
+	sys.exit()
+
+def ach_to_list(data):
+	ret = []
+
+	for d in data:
+		if "categories" in d:
+			ret.extend(ach_to_list(d["categories"]))
+
+		elif "achievements" in d:
+			ret.extend(ach_to_list(d["achievements"]))
+
+		else:
+			ret.append(d)
+
+	return ret
+
+a = ach_to_list(a["achievements"])
+
+achievements = [int(x) for x in r.smembers("bot:guild-ach")]
+
+for new in set(g["achievements"]["achievementsCompleted"]).difference(achievements):
+	r.sadd("bot:guild-ach", new)
+
+	ach = next((item for item in a if item["id"] == new), None)
+	if not ach or "title" not in ach:
+		continue
+
+	title = ach["title"]
+	desc = ach["description"]
+	url = "http://es.wowhead.com/achievement={0}".format(new)
+	icon = "https://wow.zamimg.com/images/wow/icons/large/{0}.jpg".format(ach["icon"])
+
+	wh.add_embed(webhook.embed(title=ach["title"], url="http://es.wowhead.com/achievement={0}".format(new), description=ach["description"], thumbnail=webhook.image(icon)))
+	wh.send(":clap: Vagrant Story ha ganado un logro!")
+
+	time.sleep(2)