ranelpadon · gherceg · Nov 13, 2023 · Nov 14, 2023
diff --git a/autoload/python_copy_reference.vim b/autoload/python_copy_reference.vim
@@ -43,12 +43,12 @@ function! python_copy_reference#_get_module(path_format, separator)
 endfunction
 
 
-function! python_copy_reference#_get_attribute_parts(allow_nested)
+function! python_copy_reference#_get_attributes(allow_nested, separator)
     let current_word = expand('<cword>')
     let pattern = '\v^(class|def|async)'
 
     if match(current_word, pattern) == -1
-        return []
+        return ''
     endif
 
     let current_column = col('.')
@@ -57,15 +57,15 @@ function! python_copy_reference#_get_attribute_parts(allow_nested)
     " `def`/`class` is already at the gutter/edge.
     if current_column == 1
         " Return the file path + function/class.
-        return [name]
+        return name
     elseif !a:allow_nested
-        return []
+        return ''
     else
         call python_copy_reference#_jump_to_parent()
         let parent_name = python_copy_reference#_get_nearest_name()
 
         " Return the file path + function/class + inner function/class/method.
-        return [parent_name, name]
+        return parent_name . a:separator . name
     endif
 endfunction
 
@@ -111,11 +111,21 @@ function! python_copy_reference#_get_reference(format)
         let path_format = '%:r'
         let separator = '.'
         let allow_nested = 0
+    elseif a:format == 'nose'
+        " Path format (without file extension): foo/bar/baz
+        let path_format = '%:r'
+        let separator = '.'
+        let module_attr_separator = ':'
+        let allow_nested = 1
     endif
-
     let module = python_copy_reference#_get_module(path_format, separator)
-    let attribute_parts = python_copy_reference#_get_attribute_parts(allow_nested)
-    let reference = join([module] + attribute_parts, separator)
+    let attributes = python_copy_reference#_get_attributes(allow_nested, separator)
+    if empty(attributes)
+        let reference = module
+    else
+        let module_attr_separator = exists('module_attr_separator') ? module_attr_separator : separator
+        let reference = module . module_attr_separator . attributes
+    endif
 
     " Go back to marked/initial cursor for better UX.
     execute 'normal! `X'
@@ -149,3 +159,10 @@ function! python_copy_reference#import()
     echomsg 'Copied reference: ' . reference
     let @+ = reference
 endfunction
+
+
+function! python_copy_reference#nose()
+    let reference = python_copy_reference#_get_reference('nose')
+    echomsg 'Copied reference: ' . reference
+    let @+ = reference
+endfunction
diff --git a/plugin/python-copy-reference.vim b/plugin/python-copy-reference.vim
@@ -15,3 +15,4 @@ let g:loaded_python_copy_reference = 1
 command! -nargs=0 PythonCopyReferenceDotted call python_copy_reference#dotted()
 command! -nargs=0 PythonCopyReferencePytest call python_copy_reference#pytest()
 command! -nargs=0 PythonCopyReferenceImport call python_copy_reference#import()
+command! -nargs=0 PythonCopyReferenceNose call python_copy_reference#nose()
diff --git a/test/nose.vader b/test/nose.vader
@@ -0,0 +1,30 @@
+Do (copy reference):
+  :e! some/package/path.py\<cr>
+  :let g:python_copy_reference = {}\<cr>
+  :call python_copy_reference#nose()\<cr>
+
+Do (paste reference):
+  "+p
+
+Expect:
+  some.package.path
+
+Do (clean up):
+  let @+ = ''
+  :%bd!\<cr>
+
+Do (copy reference with attributes):
+  :e! some/package/path.py\<cr>
+  idef SomeClass():\<cr>  def some_function():\<cr>    pass\<esc>
+  k
+  :let g:python_copy_reference = {}\<cr>
+  :call python_copy_reference#nose()\<cr>
+
+Do (paste reference):
+  "+p
+
+Expect:
+  some.package.path:SomeClass.some_function
+
+Do (clean up):
+  :%bd!\<cr>
diff --git a/test/pytest.vader b/test/pytest.vader
@@ -11,3 +11,19 @@ Expect:
 
 Do (clean up):
   :%bd!\<cr>
+
+Do (copy reference with attributes):
+  :e! some/package/path.py\<cr>
+  idef SomeClass():\<cr>  def some_function():\<cr>    pass\<esc>
+  k
+  :let g:python_copy_reference = {}\<cr>
+  :call python_copy_reference#pytest()\<cr>
+
+Do (paste reference):
+  "+p
+
+Expect:
+  some/package/path.py::SomeClass::some_function
+
+Do (clean up):
+  :%bd!\<cr>