xref: /aosp_15_r20/external/autotest/frontend/thread_local.py (revision 9c5db1993ded3edbeafc8092d69fe5de2ee02df7)
1*9c5db199SXin Liimport threading
2*9c5db199SXin Li
3*9c5db199SXin Li_store = threading.local()
4*9c5db199SXin Li_store.user = None
5*9c5db199SXin Li
6*9c5db199SXin Lidef set_user(user):
7*9c5db199SXin Li    """\
8*9c5db199SXin Li    Sets the current request's logged-in user.  user should be a
9*9c5db199SXin Li    afe.models.User object.
10*9c5db199SXin Li    """
11*9c5db199SXin Li    _store.user = user
12*9c5db199SXin Li
13*9c5db199SXin Li
14*9c5db199SXin Lidef get_user():
15*9c5db199SXin Li    'Get the currently logged-in user as a afe.models.User object.'
16*9c5db199SXin Li    return _store.user
17