Add type annotations

Author: chadrik

setup.py

@@ -32,7 +32,7 @@
     include_package_data=True,
     package_data={"": ["cacerts.txt", "cacert.pem"]},
     zip_safe=False,
-    python_requires=">=3.7.0",
+    python_requires=">=3.9.0",
     classifiers=[
         "Development Status :: 5 - Production/Stable",
         "Intended Audience :: Developers",

shotgun_api3/lib/mimetypes.py

@@ -26,6 +26,7 @@
 a patched version of the native mimetypes module and is used only in Python
 versions 2.7.0 - 2.7.9, which included a broken version of the mimetypes module.
 """
+from __future__ import print_function
 
 import os
 import sys
@@ -568,14 +569,14 @@ def _default_mime_types():
 """
 
     def usage(code, msg=''):
-        print USAGE
-        if msg: print msg
+        print(USAGE)
+        if msg: print(msg)
         sys.exit(code)
 
     try:
         opts, args = getopt.getopt(sys.argv[1:], 'hle',
                                    ['help', 'lenient', 'extension'])
-    except getopt.error, msg:
+    except getopt.error as msg:
         usage(1, msg)
 
     strict = 1
@@ -590,9 +591,9 @@ def usage(code, msg=''):
     for gtype in args:
         if extension:
             guess = guess_extension(gtype, strict)
-            if not guess: print "I don't know anything about type", gtype
-            else: print guess
+            if not guess: print("I don't know anything about type", gtype)
+            else: print(guess)
         else:
             guess, encoding = guess_type(gtype, strict)
-            if not guess: print "I don't know anything about type", gtype
-            else: print 'type:', guess, 'encoding:', encoding
+            if not guess: print("I don't know anything about type", gtype)
+            else: print('type:', guess, 'encoding:', encoding)

shotgun_api3/lib/mockgun/mockgun.py

@@ -120,6 +120,7 @@
 from ...shotgun import _Config
 from .errors import MockgunError
 from .schema import SchemaFactory
+from typing import Any
 
 # ----------------------------------------------------------------------------
 # Version
@@ -581,7 +582,7 @@ def _get_new_row(self, entity_type):
             row[field] = default_value
         return row
 
-    def _compare(self, field_type, lval, operator, rval):
+    def _compare(self, field_type: str, lval: Any, operator: str, rval: Any) -> bool:
         """
         Compares a field using the operator and value provide by the filter.
 
@@ -798,7 +799,7 @@ def _row_matches_filter(self, entity_type, row, sg_filter, retired_only):
 
             return self._compare(field_type, lval, operator, rval)
 
-    def _rearrange_filters(self, filters):
+    def _rearrange_filters(self, filters: list) -> None:
         """
         Modifies the filter syntax to turn it into a list of three items regardless
         of the actual filter. Most of the filters are list of three elements, so this doesn't change much.

shotgun_api3/lib/mockgun/schema.py

@@ -47,7 +47,7 @@ class SchemaFactory(object):
     _schema_cache_path = None
 
     @classmethod
-    def get_schemas(cls, schema_path, schema_entity_path):
+    def get_schemas(cls, schema_path: str, schema_entity_path: str) -> tuple:
         """
         Retrieves the schemas from disk.
 

shotgun_api3/lib/sgsix.py

@@ -58,7 +58,7 @@
     ShotgunSSLError = SSLHandshakeError
 
 
-def normalize_platform(platform, python2=True):
+def normalize_platform(platform: str, python2: bool = True) -> str:
     """
     Normalize the return of sys.platform between Python 2 and 3.
 

shotgun_api3/lib/sgutils.py

@@ -29,7 +29,7 @@
 """
 
 
-def ensure_binary(s, encoding='utf-8', errors='strict'):
+def ensure_binary(s, encoding='utf-8', errors='strict') -> bytes:
     """
     Coerce **s** to bytes.
 
@@ -44,7 +44,7 @@ def ensure_binary(s, encoding='utf-8', errors='strict'):
         raise TypeError(f"not expecting type '{type(s)}'")
 
 
-def ensure_str(s, encoding='utf-8', errors='strict'):
+def ensure_str(s, encoding='utf-8', errors='strict') -> str:
     """Coerce *s* to `str`.
 
       - `str` -> `str`