fix: type decl generator

mtshiba · Nov 30, 2024 · 4c57e6e · 4c57e6e
1 parent fa4cefd
commit 4c57e6e
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/crates/py2erg/gen_decl.rs b/crates/py2erg/gen_decl.rs
@@ -63,14 +63,23 @@ impl DeclFileGenerator {
     fn escape_type(&self, typ: String) -> String {
         typ.replace('%', "Type_")
             .replace("<module>", "")
+            .replace('/', ".")
             .trim_start_matches(self.filename.trim_end_matches(".d.er"))
             .trim_start_matches(&self.namespace)
             .to_string()
     }
 
     // e.g. `x: foo.Bar` => `foo = pyimport "foo"; x: foo.Bar`
     fn prepare_using_type(&mut self, typ: &Type) {
-        let namespace = Str::rc(typ.namespace().split('.').next().unwrap());
+        let namespace = Str::rc(
+            typ.namespace()
+                .split('/')
+                .next()
+                .unwrap()
+                .split('.')
+                .next()
+                .unwrap(),
+        );
         if namespace != self.namespace
             && !namespace.is_empty()
             && self.imported.insert(namespace.clone())