fix display.rotation display is None exception

NetEaseGame · Jun 6, 2017 · 5a715e0 · 5a715e0
1 parent 0c9991d
commit 5a715e0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/atx/adbkit/device.py b/atx/adbkit/device.py
@@ -127,7 +127,8 @@ def rotation(self):
         Return:
             - int [0-4]
         """
-        return self.display.rotation
+        if self.display:
+            return self.display.rotation
 
     def properties(self):
         '''

diff --git a/atx/drivers/android.py b/atx/drivers/android.py
@@ -481,11 +481,11 @@ def dump_view(self):
         warnings.warn("deprecated, source() instead", DeprecationWarning)
         return self._uiauto.dump()
 
-    def source(self):
+    def source(self, *args, **kwargs):
         """
         Dump page xml
         """
-        return self._uiauto.dump()
+        return self._uiauto.dump(*args, **kwargs)
 
     def _escape_text(self, s, utf7=False):
         s = s.replace(' ', '%s')