Commit 979a80d5 authored by Jeremy Pallats's avatar Jeremy Pallats 💬
Browse files

Zero OCRTracker fort and um properties on cycle tick. 

- Prevent OCR from picking up wrong numbers.
parent efc13c32

cog/actions.py

+2 −0
Original line number Diff line number Diff line
@@ -353,6 +353,8 @@ class Admin(Action): 
            RemoteError - The sheet/tab combination could not be resolved. Tab needs creating.

        """

        await self.top(5)

        # Zero trackers for new ocr data

        cogdb.query.ocr_zero_live_trackers(self.session)

        self.bot.deny_commands = True

        scanner_configs = cog.util.get_config('scanners')

        lines = [['Document', 'Active Page']]

cogdb/query.py

+13 −0
Original line number Diff line number Diff line
@@ -1226,3 +1226,16 @@ Current Consolidation: {}% 
        msg += "\n" + str(prep)



    return msg





def ocr_zero_live_trackers(session):

    """

    Post cycle tick 0 out existing OCRTracker objects fort and um fields.



    Args:

        session: Session on to the db.

    """

    for tracker in session.query(OCRTracker):

        tracker.fort = 0

        tracker.um = 0

    session.commit()

tests/cogdb/test_query.py

+8 −0
Original line number Diff line number Diff line
@@ -819,3 +819,11 @@ Current Consolidation: 77% 
Cubeo: 5531, updated at 2021-08-25 02:33:00

Rhea: 0, updated at 2021-08-25 02:33:00"""

    assert cogdb.query.ocr_prep_report(session) == expect





def test_ocr_zero_live_trackers(session, f_ocr_testbed):

    cogdb.query.ocr_zero_live_trackers(session)



    for tracker in session.query(OCRTracker):

        assert tracker.fort == 0

        assert tracker.um == 0