feat: method to help get compiler by name and version [APE-1618] (#103)

ethpm_types/manifest.py

@@ -297,6 +297,23 @@ def unpack_sources(self, destination: Path):
 
             source_path.write_text(content)
 
+    def get_compiler(self, name: str, version: str) -> Optional[Compiler]:
+        """
+        Get a compiler by name and version.
+
+        Args:
+            name (str): The name of the compiler, such as ``"vyper"``.
+            version (str): The version of the compiler.
+
+        Returns:
+            Optional[`~ethpm_types.source.Compiler`]
+        """
+        for compiler in self.compilers or []:
+            if compiler.name == name.lower() and compiler.version == version:
+                return compiler
+
+        return None
+
     def get_contract_compiler(self, contract_type_name: str) -> Optional[Compiler]:
         """
         Get the compiler used to compile the contract type, if it exists.

ethpm_types/source.py

@@ -53,6 +53,9 @@ def __eq__(self, other) -> bool:
             and self.settings == other.settings
         )
 
+    def __hash__(self) -> int:
+        return hash(f"{self.name}=={self.version}")
+
 
 class Checksum(BaseModel):
     """Checksum information about the contents of a source file."""

tests/test_package_manifest.py

@@ -190,6 +190,15 @@ def test_package_name_using_all_valid_characters():
     assert manifest.name == name
 
 
+def test_get_compiler():
+    compiler = Compiler(name="vyper", version="0.3.7", settings={}, contractTypes=["foobar"])
+    manifest = PackageManifest(
+        compilers=[compiler], contractTypes={"foobar": ContractType(contractNam="foobar")}
+    )
+    actual = manifest.get_compiler("vyper", "0.3.7")
+    assert actual == compiler
+
+
 def test_get_contract_compiler():
     compiler = Compiler(name="vyper", version="0.3.7", settings={}, contractTypes=["foobar"])
     manifest = PackageManifest(

tests/test_source.py

@@ -249,6 +249,23 @@ def test_compiler_equality():
     compiler_1.settings = compiler_2.settings
 
 
+def test_compiler_hash():
+    compiler_1 = Compiler(
+        name="yo", version="0.2.0", settings={"foo": "bar"}, contractType=["test1"]
+    )
+    compiler_2 = Compiler(
+        name="foo", version="0.1.0", settings={"foo": "bar"}, contractType=["test1", "test2"]
+    )
+    compiler_3 = Compiler(
+        name="yo", version="0.2.0", settings={"foo": "bar"}, contractType=["test1"]
+    )
+    a_set = {compiler_1, compiler_2, compiler_3}
+    assert len(a_set) == 2
+    assert compiler_1 in a_set
+    assert compiler_2 in a_set
+    assert compiler_3 in a_set
+
+
 def test_checksum_from_file():
     file = Path(tempfile.mktemp())
     file.write_text("foobartest123")