requests_aws4auth optional lib added.

bachetesla · Dec 7, 2016 · 7190e3b · 7190e3b
1 parent 52a890a
commit 7190e3b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/requirements-optional.txt b/requirements-optional.txt
@@ -1,2 +1,3 @@
 python-magic >= 0.4.12
 wand >= 0.4.3
+requests-aws4auth >= 0.9
diff --git a/sqlalchemy_media/optionals.py b/sqlalchemy_media/optionals.py
@@ -57,3 +57,22 @@ def ensure_wand():
 
     if wand is None:  # pragma: no cover
         raise OptionalPackageRequirementError('wand')
+
+
+# requests-aws4auth
+try:
+    # noinspection PyPackageRequirements
+    from requests_aws4auth import AWS4Auth
+except ImportError:  # pragma: no cover
+    AWS4Auth = None
+
+
+def ensure_aws4auth():
+    """
+
+    .. warning:: :exc:`.OptionalPackageRequirementError` will be raised if ``requests-aws4auth`` is not installed.
+
+    """
+
+    if AWS4Auth is None:  # pragma: no cover
+        raise OptionalPackageRequirementError('requests-aws4auth')