Вот модификация скрипта, которую я внес на тот момент. Правда, это полный бардак: моя база данных напоминала монстра Франкенштейна, и многие строки закомментированы из-за тестовых правок — в таком виде её использовать нельзя:
Обратите внимание на def import_pm — там находится интересующая вас часть.
И не забудьте:
Удачи! 
Также настоятельно рекомендую сравнить содержимое с официальным импортером, чтобы вы могли точно увидеть, что изменилось.