Refactor headers update code

thesadru · Jun 7, 2024 · fbc630a · fbc630a
1 parent 13bd94d
commit fbc630a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/genshin/client/components/base.py b/genshin/client/components/base.py
@@ -316,12 +316,9 @@ async def _request_hook(
             params = {k: v for k, v in params.items() if k != "authkey"}
             url = url.update_query(params)
 
-        headers_ = dict(headers or {})
-        if self.device_id:
-            headers_["x-rpc-device_id"] = self.device_id
-        if self.device_fp:
-            headers_["x-rpc-device_fp"] = self.device_fp
-        headers.update(headers_)
+        headers.update(
+            {k: v for k, v in {"x-rpc-device_id": self.device_id, "x-rpc-device_fp": self.device_fp}.items() if v}
+        )
 
         if data:
             self.logger.debug("%s %s\n%s", method, url, json.dumps(data, separators=(",", ":")))