small refactor

Author: charles-cooper

vyper/compiler/input_bundle.py

@@ -243,6 +243,8 @@ def _load_from_path(self, resolved_path: PurePath, original_path: PurePath) -> C
 # input bundle for vyper archives. similar to JSONInputBundle, but takes
 # a zipfile as input.
 class ZipInputBundle(InputBundle):
+    SOURCES_DIRECTORY = PurePath("sources")
+
     def __init__(self, archive: "ZipFile"):
         assert archive.testzip() is None
         self.archive = archive
@@ -259,7 +261,7 @@ def _load_from_path(self, resolved_path: PurePath, original_path: PurePath) -> C
         # zipfile.BadZipFile: File is not a zip file
 
         try:
-            p = PurePath("sources") / resolved_path
+            p = self.SOURCES_DIRECTORY / resolved_path
             value = self.archive.read(p.as_posix()).decode("utf-8")
         except KeyError:
             # zipfile literally raises KeyError if the file is not there

vyper/compiler/output_bundle.py

@@ -7,7 +7,7 @@
 from pathlib import PurePath
 from typing import Optional
 
-from vyper.compiler.input_bundle import CompilerInput, _NotFound
+from vyper.compiler.input_bundle import CompilerInput, ZipInputBundle, _NotFound
 from vyper.compiler.phases import CompilerData
 from vyper.compiler.settings import Settings
 from vyper.exceptions import CompilerPanic
@@ -252,7 +252,8 @@ def __del__(self):
 
     def write_sources(self, sources: dict[str, CompilerInput]):
         for path, c in sources.items():
-            path = f"sources/{_anonymize(path)}"
+            prefix = ZipInputBundle.SOURCES_DIRECTORY
+            path = str(prefix / _anonymize(path))
             self.archive.writestr(path, c.contents)
 
     def write_storage_layout_overrides(