From 1fd5271ebd12cff537e8ce68d9f8d706e531eb51 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Thu, 1 Aug 2019 17:13:07 -0500
Subject: Updated aliases to get all species aliases and remove duplicates

---
 wqflask/base/trait.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

(limited to 'wqflask/base')

diff --git a/wqflask/base/trait.py b/wqflask/base/trait.py
index a0679041..58169b5c 100644
--- a/wqflask/base/trait.py
+++ b/wqflask/base/trait.py
@@ -135,9 +135,20 @@ class GeneralTrait(object):
 
         alias = 'Not available'
         if self.symbol:
-            response = requests.get("http://gn2.genenetwork.org/gn3/gene/aliases/" + self.symbol)
-            alias_list = json.loads(response.content)
-            alias = "; ".join(alias_list)
+            human_response = requests.get("http://gn2.genenetwork.org/gn3/gene/aliases/" + self.symbol.upper())
+            mouse_response = requests.get("http://gn2.genenetwork.org/gn3/gene/aliases/" + self.symbol.capitalize())
+            other_response = requests.get("http://gn2.genenetwork.org/gn3/gene/aliases/" + self.symbol.lower())
+            alias_list = json.loads(human_response.content) + json.loads(mouse_response.content) + json.loads(other_response.content)
+
+            filtered_aliases = []
+            seen = set()
+            for item in alias_list:
+                if item in seen:
+                    continue
+                else:
+                    filtered_aliases.append(item)
+                    seen.add(item)
+            alias = "; ".join(filtered_aliases)
 
         return alias
 
-- 
cgit v1.2.3