refactor: move internals for identity to private module

Author: Salakar

Diff for: .pylintrc

@@ -146,6 +146,7 @@ disable=abstract-method,
         wrong-import-order,
         xrange-builtin,
         zip-builtin-not-iterating,
+        import-outside-toplevel,
 
 
 [REPORTS]

Diff for: docs/generate.sh

@@ -86,6 +86,7 @@ PY_MODULES='firebase_functions
             firebase_functions.core
             firebase_functions.db_fn
             firebase_functions.https_fn
+            firebase_functions.identity_fn
             firebase_functions.options
             firebase_functions.params
             firebase_functions.pubsub_fn

Diff for: samples/identity/functions/main.py

@@ -14,11 +14,15 @@ def beforeusercreated(
     if not event.data.email:
         return None
     if "@cats.com" in event.data.email:
-        return identity_fn.BeforeCreateResponse(display_name="🐈",
-                                                custom_claims={"meow": True})
+        return identity_fn.BeforeCreateResponse(
+            display_name="🐈",
+            custom_claims={"meow": True},
+        )
     if "@dogs.com" in event.data.email:
-        return identity_fn.BeforeCreateResponse(display_name="🐕",
-                                                custom_claims={"woof": True})
+        return identity_fn.BeforeCreateResponse(
+            display_name="🐕",
+            custom_claims={"woof": True},
+        )
     return None
 
 
@@ -33,10 +37,21 @@ def beforeusersignedin(
     print(event)
     if not event.data.email:
         return None
+
     if "@cats.com" in event.data.email:
-        return identity_fn.BeforeSignInResponse(
-            display_name="🐈", session_claims={"temporary_meow": True})
+        return {
+            "display_name": "🐈",
+            "session_claims": {
+                "session_meow": True
+            },
+        }
+
     if "@dogs.com" in event.data.email:
-        return identity_fn.BeforeSignInResponse(
-            display_name="🐕", session_claims={"temporary_woof": True})
+        return {
+            "display_name": "🐕",
+            "session_claims": {
+                "session_woof": True
+            },
+        }
+
     return None