Bladeren bron

Added stockdatapull method and rdiff data method to Quandl.py

Marco Quezada 10 jaren geleden
bovenliggende
commit
ee2e6516f6
2 gewijzigde bestanden met toevoegingen van 30 en 30 verwijderingen
  1. 30 3
      hackathon_starter/hackathon/scripts/quandl.py
  2. 0 27
      hackathon_starter/hackathon/scripts/steam.py

+ 30 - 3
hackathon_starter/hackathon/scripts/quandl.py

@@ -6,7 +6,7 @@ import json
 
 def fetchData(apikey, url):
     '''Returns JSON data of the Dow Jones Average.'''
-    parameters = {'rows' : 1, 'auth_token' : apikey}
+    parameters = {'rows' : 1, 'column' : 1, 'auth_token' : apikey}
     req = requests.get(url, params=parameters)
     data = json.loads(req.content)
     parsedData = []
@@ -16,12 +16,39 @@ def fetchData(apikey, url):
         stockData['description'] = '''The NASDAQ Composite Index measures all
         NASDAQ domestic and international based common type stocks listed \
         on The NASDAQ Stock Market.'''
-        stockData['data'] = data['data']
+        stockData['data'] = data['data'][0][1]
         stockData['code'] = data['code']
     else:
         stockData['name'] = data['name']
         stockData['description'] = data['description']
-        stockData['data'] = data['data']
+        stockData['data'] = data['data'][0][1]
         stockData['code'] = data['code']
     parsedData.append(stockData)
     return parsedData
+
+def fetchstockData(apikey, url):
+    '''Returns Stock related JSON data of the stock url placed there.'''
+    parameters = {'rows' : 1, 'auth_token' : apikey}
+    req = requests.get(url, params=parameters)
+    data = json.loads(req.content)
+    parsedData = []
+    stockData = {}
+    stockData['name'] = data['name']
+    stockData['open'] = data['data'][0][1]
+    stockData['high'] = data['data'][0][2]
+    stockData['low'] = data['data'][0][3]
+    stockData['close'] = data['data'][0][4]
+    stockData['code'] = data['code']
+    parsedData.append(stockData)
+    return parsedData
+
+def rdiffData(apikey, url):
+    '''Returns data of the difference of the stock URL placed there.'''
+    parameters = {'rows' : 1, 'column' : 1, 'transformation': 'rdiff', 'auth_token' : apikey}
+    req = requests.get(url, params=parameters)
+    data = json.loads(req.content)
+    parsedData = []
+    stockData = {}
+    stockData['rdiff'] = data['data'][0][1]
+    parsedData.append(stockData)
+    return parsedData

+ 0 - 27
hackathon_starter/hackathon/scripts/steam.py

@@ -26,30 +26,3 @@ def steamidpulling(steamun, apikey):
     data = json.loads(req.content)
     steamid = data['response']['steamid']
     return steamid
-
-def steamlibrarypull(steamid, apikey):
-    '''Pulls out a CSV of Steam appids.'''
-    steaminfo = {
-        'key': apikey,
-        'steamid': steamid,
-        'format':'JSON',
-        'include_appinfo':'1'
-    }
-    url = 'http://api.steampowered.com/IPlayerService/GetOwnedGames/v0001/'
-    req = requests.get(url, params=steaminfo)
-    data = json.loads(req.content)
-    response = data['response']['games']
-    games = {}
-    for game in response:
-        url = 'http://store.steampowered.com/api/appdetails/?appids=%d&filters=price_overview&cc=us'
-        getprice = requests.get(url % game['appid'])
-        if getprice.status_code == 200:
-            rjson = json.loads(getprice.text)
-            # use the appid to fetch the value and convert to decimal
-            # appid is numeric, cast to string to lookup the price
-            try:
-                price = rjson[str(game['appid'])]['data']['price_overview']['initial'] * .01
-            except KeyError:
-                pass
-            games[game['name']] = {'price': price, 'appid': game['appid']}
-    return games