extract source from “hf_token”

app.py

@@ -45,7 +45,7 @@ def run(hf_token, service, game, functionality, nlp_command):
     """
 
     # reuse hf_token field as json string
-    token, user_name, redirect, _ = extract(hf_token)
+    token, user_name, redirect, source, _ = extract(hf_token)
     if user_name is None:
         user_name = "__fake__"
 
@@ -90,7 +90,7 @@ def run(hf_token, service, game, functionality, nlp_command):
     else:
         try:
             assert "games" in service_list
-            if service == "games":            
+            if service == "games":
                 print(f"{beijing()} [{user_name}] [{game}] {nlp_command}")
             client = Client(
                 url,
@@ -114,6 +114,8 @@ def run(hf_token, service, game, functionality, nlp_command):
             outp["proxy-version"] = proxy_version
             outp["user"] = user_name
             outp["game"] = game
+            if source:
+                outp["source"] = source
             calling_start = beijing()
             print(f"calling logger starts at {beijing()}")
             app_util.call_logger(outp, identity, token)

extract.py

@@ -1,29 +1,35 @@
 import json
 
+
 def extract(hf_token):
     """
-    Extract token, user, redirect, and info from input hf_token.
+    Extract token, user, redirect, source, and info from input hf_token.
     If hf_token is simple, it is the token itself.
     """
-    info = {} # a copy of hf_token in json format
+    info = {}  # a copy of hf_token in json format
     try:
         info = json.loads(hf_token)
     except json.decoder.JSONDecodeError:
-        return hf_token, None, None, None
+        return hf_token, None, None, None, None
 
     try:
-        token = info['token']
+        token = info["token"]
     except KeyError:
         token = None
 
     try:
-        user = info['user']
+        user = info["user"]
     except KeyError:
         user = None
 
     try:
-        redirect = info['redirect']
+        redirect = info["redirect"]
     except KeyError:
         redirect = None
 
-    return token, user, redirect, info
+    try:
+        source = info["source"]
+    except KeyError:
+        source = None
+
+    return token, user, redirect, source, info