1 files changed, 74 insertions, 0 deletions
diff --git a/octopart.py b/octopart.py
new file mode 100644
index 0000000..106a69f
--- /dev/null
+++ b/octopart.py
@@ -0,0 +1,74 @@
+
+import json
+import urllib
+from decimal import Decimal
+
+from settings import *
+
+def fetch_bom(bom):
+    # inspired by "bom_quote.py" provided at http://octopart.com/api
+    reply = dict()
+    queries = []
+    for p in bom:
+        pid = "%s|%s" % p
+        queries.append({'mpn': p[1],
+                        'brand': p[0],
+                        'reference': pid})
+        reply[pid] = None
+
+    # do requests in batches of 20
+    results = []
+    for i in range(0, len(queries), OCTOPART_BATCH_SIZE):
+        batched_queries = queries[i:i+OCTOPART_BATCH_SIZE]
+        url = 'http://octopart.com/api/v3/parts/match?queries=%s' \
+            % urllib.quote(json.dumps(batched_queries))
+        url += '&apikey=%s' % OCTOPART_API_KEY
+        #print url
+        data = urllib.urlopen(url).read()
+        response = json.loads(data)
+        results.extend(response['results'])
+   
+    #print "len(results): %d" % len(results)
+
+    for result in results:
+        pid = result['reference']
+        print "len(items[%s]): %d" % (pid, len(result['items']))
+        if len(result['items']) == 0:
+            reply[pid] = None
+        else:
+            reply[pid] = result['items'][0]
+    return reply
+
+def price_info(item, quantity=1000):
+    if not item:
+        return dict(url=None, css='notfound', price='Not Found')
+    info = dict(url=item['octopart_url'])
+    info['css'] = 'unavailable'
+    info['price'] = 'No Offers'
+    for offer in item['offers']:
+        if not offer['is_authorized']:
+            continue
+        if not offer['prices'].has_key('USD'):
+            continue
+        price = None
+        for price_pair in offer['prices']['USD']:
+            if price_pair[0] <= quantity:
+                if not price or price_pair[1] < price:
+                    price = price_pair[1]
+        if not price:
+            print "WARNING: not a price: %s" % price
+            continue
+        if not info['price'] or price < info['price']:
+            info['price'] = Decimal(price)
+            if offer['in_stock_quantity'] > 0:
+                info['css'] = 'available'
+            else:
+                info['css'] = 'outofstock'
+    return info
+
+def url_info(item):
+    if item:
+        return item['octopart_url']
+    else:
+        return None
+