uranusjr commented on code in PR #33355:
URL: https://github.com/apache/airflow/pull/33355#discussion_r1336849618
##########
airflow/operators/python.py:
##########
@@ -613,7 +625,83 @@ def _prepare_venv(self, venv_path: Path) -> None:
index_urls=self.index_urls,
)
+ def _calculate_cache_hash(self) -> tuple[str, str]:
+ """Helper to generate the hash of the cache folder to use.
+
+ The following factors are used as input for the hash:
+ - (sorted) list of requirements
+ - pip install options
+ - flag of system site packages
+ - python version
+ - Variable to override the hash with a cache key
+ - Index URLs
+
+ Returns a hash and the data dict which is the base for the hash as
text.
+ """
+ hash_dict = {
+ "requirements_list": self._requirements_list(),
+ "pip_install_options": self.pip_install_options,
+ "index_urls": self.index_urls,
+ "cache_key":
str(Variable.get("PythonVirtualenvOperator.cache_key", "")),
+ "python_version": self.python_version,
+ "system_site_packages": self.system_site_packages,
+ }
+ hash_text = json.dumps(hash_dict, sort_keys=True)
+ hash_object = hashlib_wrapper.md5(hash_text.encode())
+ requirements_hash = hash_object.hexdigest()
+ return requirements_hash[0:8], hash_text
Review Comment:
```suggestion
return requirements_hash[:8], hash_text
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
