From f327752947ec4ce9445fcf51baddc9ca72443152 Mon Sep 17 00:00:00 2001
From: Filipp Lepalaan <f@230.to>
Date: Mon, 10 Jun 2013 14:32:12 +0300
Subject: Cleanup

---
 gsxws/repairs.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

(limited to 'gsxws/repairs.py')

diff --git a/gsxws/repairs.py b/gsxws/repairs.py
index ee649d8..13bf4e5 100644
--- a/gsxws/repairs.py
+++ b/gsxws/repairs.py
@@ -1,3 +1,5 @@
+# -*- coding: utf-8 -*-
+
 "gsxws/repairs.py"
 import re
 import sys
@@ -82,8 +84,7 @@ class Repair(GsxObject):
         if hasattr(self, "dispatchId"):
             self.repairConfirmationNumber = self.dispatchId
 
-        self._submit("repairData", "UpdateSerialNumber", "repairConfirmation")
-        return self._req.objects[0]
+        return self._submit("repairData", "UpdateSerialNumber", "repairConfirmation")
 
     def update_kgb_sn(self, sn):
         """
@@ -102,10 +103,8 @@ class Repair(GsxObject):
         self.serialNumber = sn
         self.repairConfirmationNumber = self.dispatchId
 
-        self._submit("UpdateKGBSerialNumberRequest", "UpdateKGBSerialNumber",
-                     "UpdateKGBSerialNumberResponse")
-
-        return self._req.objects[0]
+        return self._submit("UpdateKGBSerialNumberRequest", "UpdateKGBSerialNumber",
+                            "UpdateKGBSerialNumberResponse")
 
     def lookup(self):
         """
@@ -136,9 +135,8 @@ class Repair(GsxObject):
         repair confirmation numbers to be submitted to GSX to be marked as complete.
         """
         self.repairConfirmationNumbers = numbers or self.dispatchId
-        self._submit("MarkRepairCompleteRequest", "MarkRepairComplete",
-                     "MarkRepairCompleteResponse")
-        return self._req.objects[0]
+        return self._submit("MarkRepairCompleteRequest", "MarkRepairComplete",
+                            "MarkRepairCompleteResponse")
 
     def status(self, numbers=None):
         """
-- 
cgit v1.2.3