From a01a92491adaf3bd063d55125701e81d01b58534 Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <f@230.to>
Date: Mon, 10 Jun 2013 20:47:30 +0300
Subject: Added parts argument to Product.warranty()

---
 gsxws/products.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'gsxws')

diff --git a/gsxws/products.py b/gsxws/products.py
index 3042109..252d97e 100644
--- a/gsxws/products.py
+++ b/gsxws/products.py
@@ -39,7 +39,7 @@ class Product(GsxObject):
         self.configCode = result.configCode
         return result
 
-    def warranty(self):
+    def warranty(self, parts=None):
         """
         The Warranty Status API retrieves the same warranty details
         displayed on the GSX Coverage screen.
@@ -51,7 +51,17 @@ class Product(GsxObject):
         u'Out Of Warranty (No Coverage)'
         >>> Product('DGKFL06JDHJP').warranty().estimatedPurchaseDate
         datetime.date(2011, 6, 2)
+        >>> Product('WQ8094DW0P1').warranty([(u'661-5070', u'Z26',)]) # doctest: +ELLIPSIS
+        {'warrantyStatus': 'Out Of Warranty (No Coverage)',...
         """
+        try:
+            self.partNumbers = []
+            for k, v in parts:
+                part = GsxObject(partNumber=k, comptiaCode=v)
+                self.partNumbers.append(part)
+        except Exception:
+            pass
+
         self._submit("unitDetail", "WarrantyStatus", "warrantyDetailInfo")
         self.warrantyDetails = self._req.objects
         return self.warrantyDetails
-- 
cgit v1.2.3