from sqlalchemy import desc

from models import Notifikasi, Session

session = Session()


def hapus_notif():
    lima = session.query(Notifikasi).order_by(desc(Notifikasi.id)).limit(5)

    aman = []

    for item in lima:
        id = item.id

        aman.append(id)

    session.query(Notifikasi).filter(Notifikasi.id.not_in(aman)).delete(
        synchronize_session="fetch"
    )
    session.commit()


hapus_notif()