Можно сократить тело функции до `return path.split('/')[0]`.
Но всё равно этот способ не учитывает гибкость именования файлов в уровнях.