@Engineer_Mordimer: jak zwykle w takich zadaniach należy zakładać, że testowy input posiada wszystkie dane w prawidłowym formacie, jeżeli chcesz pisać obsługę takiego wyjątku to brane pod uwagę powinno być 5 pierwszych znaków rozszerzenia

@Engineer_Mordimer: to podejście ma ten plus, że nie trzeba pisać osobnej obsługi, bo pythonowy slicing jest dosyć elastyczny i użycie [:5] nie rzuci błędu nawet jeśli rozszerzenie ma na przykład 2 czy 3 znaki - wtedy zwróci je bez zmian (to coś porównywalnego do użycia modulo na incie dla ograniczania zakresu)