fix: handle non alpha chars in contract names (#663)

ApeWorX · Apr 17, 2022 · 2a653a5 · 2a653a5
1 parent d85544e
commit 2a653a5
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 1 deletion.
diff --git a/src/ape/managers/project/types.py b/src/ape/managers/project/types.py
@@ -43,7 +43,9 @@ def sources(self) -> List[Path]:
 
         for extension in self.compiler_manager.registered_compilers:
             r_ext = extension.replace(".", "\\.")
-            files.extend(get_all_files_in_directory(self.contracts_folder, pattern=rf"\w+{r_ext}"))
+            files.extend(
+                get_all_files_in_directory(self.contracts_folder, pattern=rf"[\w|-]+{r_ext}")
+            )
 
         return files
 

diff --git a/tests/integration/cli/projects/one-interface/contracts/Interface-with-hyphens.json b/tests/integration/cli/projects/one-interface/contracts/Interface-with-hyphens.json
@@ -0,0 +1,3 @@
+[
+    {"name":"foo","type":"fallback", "stateMutability":"nonpayable"}
+]
diff --git a/tests/integration/cli/projects/one-interface/contracts/InterfaceWithNumber123.json b/tests/integration/cli/projects/one-interface/contracts/InterfaceWithNumber123.json
@@ -0,0 +1,3 @@
+[
+    {"name":"foo","type":"fallback", "stateMutability":"nonpayable"}
+]
diff --git a/tests/integration/cli/projects/one-interface/contracts/Interface_with_underscores.json b/tests/integration/cli/projects/one-interface/contracts/Interface_with_underscores.json
@@ -0,0 +1,3 @@
+[
+    {"name":"foo","type":"fallback", "stateMutability":"nonpayable"}
+]