Setelah itu Budi terjun ke dunia politik dengan menjadi Anggota DPR RI dari Fraksi Partai Gerindra, yakni periode 2017-2019 dan 2019-2024. Di periode keduanya sebagai wakil rakyat, Budi didapuk menjadi Wakil Ketua Komisi IV DPR RI. Sedangkan di Partai Gerindra, Budi juga menjabat sebagai Waketum.