Merge branch 'master' of github.com:filipp/py-gsxws

Conflicts:
	gsxws/products.py

1 files changed, 10 insertions, 1 deletions

diff --git a/gsxws/products.py b/gsxws/products.py
index ea2e648..c62021b 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,10 +51,19 @@ 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)',...
         """
         if hasattr(self, "alternateDeviceId"):
             if not self.serialNumber:
                 self.activation()
+        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