fstrings.vim

" ~/.vim/after/syntax/python.vim or ~/.config/nvim/after/syntax/python.vim
syn match pythonEscape +{{+ contained containedin=pythonfString,pythonfDocstring
syn match pythonEscape +}}+ contained containedin=pythonfString,pythonfDocstring

syn region pythonfString matchgroup=pythonQuotes
      \ start=+[fF]\@1<=\z(['"]\)+ end="\z1"
      \ contains=@Spell,pythonEscape,pythonInterpolation
syn region pythonfDocstring matchgroup=pythonQuotes
      \ start=+[fF]\@1<=\z('''\|"""\)+ end="\z1" keepend
      \ contains=@Spell,pythonEscape,pythonSpaceError,pythonInterpolation,pythonDoctest

syn region pythonInterpolation contained
      \ matchgroup=SpecialChar
      \ start=+{{\@!+ end=+}}\@!+ skip=+{{+ keepend
      \ contains=ALLBUT,pythonDecoratorName,pythonDecorator,pythonFunction,pythonDoctestValue,pythonDoctest

syn match pythonStringModifier /:\(.[<^=>]\)\?[-+ ]\?#\?0\?[0-9]*[_,]\?\(\.[0-9]*\)\?[bcdeEfFgGnosxX%]\?/ contained containedin=pythonInterpolation
syn match pythonStringModifier /![sra]/ contained containedin=pythonInterpolation
syn match pythonStringModifier /\zs *= *\ze[}:!]/ contained containedin=pythonInterpolation

hi link pythonfString String
hi link pythonfDocstring String
hi link pythonStringModifier PreProc

Thank you for posting this very useful gist, @phelipetls!

I found it useful to add the following after line 17, which enables highlighting the = in e.g. f"{foo=}":

syn match pythonStringModifier /\zs *= *\ze[}:!]/ contained containedin=pythonInterpolation

(See this change in context: https://gist.github.com/finite-state-machine/aa12c251fc4a7809d1484c4d82d9b064)

(I'd send a pull request, but those don't seem to be supported for gists.)

Thanks! I added it to the gist :)

Quick comment; I hope to look into this when I have more time, but it looks like something is missing, or my environment could be weird somehow.

Anyway, I don't think the highlighting is quite right for stuff like f"... {shlex.quote(SOURCEDIR)} ..." (where the rest of this string justifies why this isn't just using the value). It looks to me like quote(SOURCEDIR) isn't getting highlighted, although it's worth noting that I'm not getting the extra highlights at all without running https://github.com/jaxbot/semantic-highlight.vim, which I think is evidence towards "I did something weird with my environment".

For person who have a problem to use this script, put this file to ~/.config/nvim/after/syntax or ~/.vim/after/syntax. Then, rename this script from fstrings.vim to python.vim.

Thanks, @ninfueng. I added this as a comment to the gist.

@mwchase it seems to be working for me?

I just experimented a little, and it looks like it's a result of using https://github.com/vim-python/python-syntax instead of the builtin python syntax. I can try to work out what's working differently later.

Actually, from looking at using the built-in syntax, I... can't tell what using the other version was getting me, so I could just not bother.

That would make sense, this files assumes you're using the default highlight groups, bundled with Vim.

All groups will be highlighted, except things that aren't expressions (should probably contain more stuff here but I forgot):

syn region pythonInterpolation contained
      \ matchgroup=SpecialChar
      \ start=+{{\@!+ end=+}}\@!+ skip=+{{+ keepend
      \ contains=ALLBUT,pythonDecoratorName,pythonDecorator,pythonFunction,pythonDoctestValue,pythonDoctest

After a quick look, it seems this would make sense if you're using python-syntax:

syn region pythonInterpolation contained
      \ matchgroup=SpecialChar
      \ start=+{{\@!+ end=+}}\@!+ skip=+{{+ keepend
      \ contains=pythonExpression

Thank you! This should be part of the standard Vim distribution!