fix retry when take screenshot, update readthedocs

onesecurity · Feb 18, 2020 · 8f5fdd3 · 8f5fdd3
1 parent 4328a7c
commit 8f5fdd3
Show file tree

Hide file tree

Showing 10 changed files with 30 additions and 40 deletions.
diff --git a/.readthedocs.yml b/.readthedocs.yml
diff --git a/docs/Makefile b/docs/Makefile
@@ -5,8 +5,8 @@
 # from the environment for the first two.
 SPHINXOPTS    ?=
 SPHINXBUILD   ?= sphinx-build
-SOURCEDIR     = source
-BUILDDIR      = build
+SOURCEDIR     = .
+BUILDDIR      = _build
 
 # Put it first so that "make" without argument is like "make help".
 help:

diff --git a/docs/api.rst b/docs/api.rst
@@ -0,0 +1,8 @@
+API
+----------
+
+.. autoclass:: uiautomator2.Device
+   :members:
+
+.. autoclass:: uiautomator2.Session
+   :members:
diff --git a/docs/source/conf.py → docs/conf.py b/docs/source/conf.py → docs/conf.py
@@ -12,7 +12,9 @@
 #
 import os
 import sys
-sys.path.insert(0, os.path.abspath('../..'))
+sys.path.insert(0, os.path.abspath('..'))
+
+import uiautomator2
 
 
 # -- Project information -----------------------------------------------------
@@ -27,24 +29,28 @@
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
-extensions = ["sphinx.ext.autodoc"]
+extensions = [
+    'sphinx.ext.autodoc',
+    'sphinx.ext.autosummary',
+    'sphinx.ext.coverage',
+    'sphinx.ext.viewcode',
+]
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ['_templates']
 
 # List of patterns, relative to source directory, that match files and
 # directories to ignore when looking for source files.
 # This pattern also affects html_static_path and html_extra_path.
-exclude_patterns = []
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
 
 
 # -- Options for HTML output -------------------------------------------------
 
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-#html_theme = 'alabaster'
-html_theme = 'sphinx_rtd_theme'
+html_theme = 'alabaster'
 
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,

diff --git a/docs/source/index.rst → docs/index.rst b/docs/source/index.rst → docs/index.rst
@@ -1,5 +1,5 @@
 .. uiautomator2 documentation master file, created by
-   sphinx-quickstart on Tue Feb 18 09:46:00 2020.
+   sphinx-quickstart on Tue Feb 18 15:32:23 2020.
    You can adapt this file completely to your liking, but it should at least
    contain the root `toctree` directive.
 
@@ -10,11 +10,8 @@ Welcome to uiautomator2's documentation!
    :maxdepth: 2
    :caption: Contents:
 
-.. automodule:: uiautomator2.Device
-   :members:
+   api
 
-.. automodule:: uiautomator2.Session
-   :members:
 
 Indices and tables
 ==================

diff --git a/docs/requirements.txt b/docs/requirements.txt
diff --git a/uiautomator2/ext/ocr/README.md → uiautomator2/ext-archived/ocr/README.md b/uiautomator2/ext/ocr/README.md → uiautomator2/ext-archived/ocr/README.md
diff --git a/uiautomator2/ext/ocr/__init__.py → uiautomator2/ext-archived/ocr/__init__.py b/uiautomator2/ext/ocr/__init__.py → uiautomator2/ext-archived/ocr/__init__.py
diff --git a/uiautomator2/ext/ocr/baiduOCR.py → uiautomator2/ext-archived/ocr/baiduOCR.py b/uiautomator2/ext/ocr/baiduOCR.py → uiautomator2/ext-archived/ocr/baiduOCR.py
diff --git a/uiautomator2/session.py b/uiautomator2/session.py
@@ -466,7 +466,7 @@ def drag(self, sx, sy, ex, ey, duration=0.5):
         return self.jsonrpc.drag(sx, sy, ex, ey, int(duration * 200))
 
     @retry((IOError, SyntaxError), delay=.5, tries=5, jitter=0.1,
-           max_delay=1)  # delay .5, .6, .7, .8 ...
+           max_delay=1, logger=logging)  # delay .5, .6, .7, .8 ...
     def screenshot(self, filename=None, format='pillow'):
         """
         Image format is JPEG
@@ -492,9 +492,12 @@ def screenshot(self, filename=None, format='pillow'):
                 f.write(r.content)
             return filename
         elif format == 'pillow':
-            from PIL import Image
-            buff = io.BytesIO(r.content)
-            return Image.open(buff).convert("RGB")
+            try:
+                from PIL import Image
+                buff = io.BytesIO(r.content)
+                return Image.open(buff).convert("RGB")
+            except Exception as ex:
+                raise IOError("PIL.Image.open IOError", ex)
         elif format == 'opencv':
             import cv2
             import numpy as np