util path to string

youknowone · youknowone · commit c0de1eafe413 · 2022-04-20T00:04:08.000+09:00
diff --git a/derive/src/pymodule.rs b/derive/src/pymodule.rs
@@ -184,17 +184,14 @@ where
     let mut pyattrs = Vec::new();
     for (i, attr) in iter {
         // take py items but no cfgs
-        let attr_name = if let Some(ident) = attr.get_ident() {
-            ident.to_string()
-        } else {
-            continue;
-        };
+        let attr_name = attr.path.to_string();
         if attr_name == "cfg" {
             return Err(syn::Error::new_spanned(
                 attr,
                 "#[py*] items must be placed under `cfgs`",
             ));
         }
+        let attr_name = attr_name.replace("::", "");
 
         let attr_name = match AttrName::from_str(attr_name.as_str()) {
             Ok(name) => name,
diff --git a/derive/src/util.rs b/derive/src/util.rs
@@ -25,6 +25,11 @@ pub(crate) const ALL_ALLOWED_NAMES: &[&str] = &[
     "extend_class",
 ];
 
+pub fn path_to_string(path: &syn::Path) -> String {
+    let x = format!("{}", quote! {#path});
+    x.replace(' ', "")
+}
+
 #[derive(Default)]
 pub(crate) struct ItemNursery(IndexMap<(String, Vec<Attribute>), TokenStream>);
 
@@ -346,6 +351,7 @@ pub(crate) fn path_eq(path: &Path, s: &str) -> bool {
 }
 
 pub(crate) trait AttributeExt: SynAttributeExt {
+    fn path_string(&self) -> String;
     fn promoted_nested(&self) -> Result<PunctuatedNestedMeta>;
     fn ident_and_promoted_nested(&self) -> Result<(&Ident, PunctuatedNestedMeta)>;
     fn try_remove_name(&mut self, name: &str) -> Result<Option<syn::NestedMeta>>;
@@ -355,9 +361,12 @@ pub(crate) trait AttributeExt: SynAttributeExt {
 }
 
 impl AttributeExt for Attribute {
+    fn path_string(&self) -> String {
+        path_to_string(&self.path)
+    }
     fn promoted_nested(&self) -> Result<PunctuatedNestedMeta> {
         let list = self.promoted_list().map_err(|mut e| {
-            let name = self.get_ident().unwrap().to_string();
+            let name = self.path_string();
             e.combine(syn::Error::new_spanned(
                 self,
                 format!(
