VSCode не подсвечивает переменные, классы и функции не встроенные в Python(если ты создал или импортировал класс, переменную или функцию, VSCode не будет подсвечивать).
И кстати, зачем тебе эта ТЯЖЕЛЕННАЯ VisualStudio если у тебя есть VSCode? VisualStudio нужна для C, C#, C++.
мне например как и другим разработчикам достаточно VSCode.
Можно попробовать PyCharm