Add USE_WRAPPED toggle to users

Author: hugetim

.anvil_editor.yaml

@@ -2,6 +2,7 @@ unique_ids:
   forms: {}
   modules:
     _demo_startup: '1719087683748644303970981.0312'
+    _users: '1721065127336894381056321.5935'
     auto_batch: '1720557189347558051799744.9072'
     tables: '1720711436314890422194670.1993'
     tables._tables: '1721062590178633669568195.047'

client_code/_users.py

@@ -0,0 +1,12 @@
+from .auto_batch import debatchify_inputs, BatchTable
+import anvil.users
+
+
+def force_login(*args, **kwargs):
+    return debatchify_inputs(anvil.users.force_login)(*args, **kwargs)
+
+
+def get_user(*args, **kwargs):
+    user_row = anvil.users.get_user(*args, **kwargs)
+    batch_table = BatchTable.of_row(user_row)
+    return batch_table.get_batch_row(user_row)

client_code/users.py

@@ -1,16 +1,10 @@
-from .auto_batch import debatchify_inputs, BatchTable
+from .auto_batch import USE_WRAPPED
+from . import _users
 import anvil.users
 
 
-def force_login(*args, **kwargs):
-    return debatchify_inputs(anvil.users.force_login)(*args, **kwargs)
-
-
-def get_user(*args, **kwargs):
-    user_row = anvil.users.get_user(*args, **kwargs)
-    batch_table = BatchTable.of_row(user_row)
-    return batch_table.get_batch_row(user_row)
-
-
 def __getattr__(attr):
-    return getattr(anvil.users, attr)
+    if USE_WRAPPED and attr in ('force_login', 'get_user'):
+        return getattr(_users, attr)
+    else:
+        return getattr(anvil.users, attr)