PantrySentry.py

import threading
from enum import Enum

import keyring

import userdata
from Database import Database
from listmanager import ListManager
from opengtin import OpenGTIN, OpenGtinException


class PantrySentry:
    """
    The PantrySentry class is the top-level class to interact with the Pantry Sentry.
    """

    class Direction(Enum):
        ADD_TO_PANTRY = 1
        REMOVE_FROM_PANTRY = 2

    def __init__(self, local_ean_db_name='ean.db'):
        """
        Create a new PantrySentry
        """
        # Get the local, persistent EAN database that is queried before any web services are used
        self.local_ean_db = Database(local_ean_db_name)
        # Handle to the online Open GTIN database in case an item is not already in the local db
        self.online_ean_db = OpenGTIN(userdata.openGTIN_user_id)
        self.pantry = ListManager(userdata.google_keep_username,
                                  keyring.get_password('Google Keep', userdata.google_keep_username),
                                  list_name='Basement food')
        self.direction = self.Direction.ADD_TO_PANTRY

    def barcode_to_product_name(self, ean: int):
        """ Get the product name based on the barcode """
        product = self.local_ean_db[ean]
        if not product.name:
            try:
                db_entry = self.online_ean_db.query(ean)
                if db_entry['name']:
                    name = db_entry['name']
                if db_entry['detailname']:
                    name = db_entry['detailname']
                if name:
                    product = Database.Product(ean=ean, name=name, description=db_entry['descr'])
                    # Update the local database
                    self.local_ean_db[ean] = product
                else:
                    print("No name or detailed name available for this product")
            except OpenGtinException as e:
                print(e)
        return product.name

    def add_item(self, item_name: str):
        self.pantry.add(item_name)

    def remove_item(self, item_name: str):
        self.pantry.remove(item_name)

    def reset_mode(self):
        self.direction = self.Direction.ADD_TO_PANTRY
        print("Mode reset. Scanned items are added to pantry.")

    def switch_mode(self):
        print("Mode switch requested. Items scanned within the next 60 seconds will be removed from the pantry.")
        self.direction = self.Direction.REMOVE_FROM_PANTRY
        t = threading.Timer(60, self.reset_mode)
        t.start()

    def start(self):
        while True:
            print("Waiting for EAN input...")
            code = input()
            if code == 'SWITCHDIRECTION':
                self.switch_mode()
                continue
            else:
                ean = int(code)
            product_name = self.barcode_to_product_name(ean)
            if self.direction == self.Direction.ADD_TO_PANTRY:
                if product_name:
                    self.add_item(product_name)
                    print(f"Added {product_name} to the pantry.")
                else:
                    self.add_item(str(ean))
                    print(f"Could not find product name for {ean}. Added the EAN to the pantry instead.")
            else:  # Items should be removed from pantry
                if product_name:
                    try:
                        self.remove_item(product_name)
                        print(f"Removed {product_name} from the pantry.")
                    except ValueError as e:
                        print(e)

                else:
                    self.remove_item(str(ean))
                    print(f"Could not find product name for {ean}. Removed the EAN from the pantry instead.")


if __name__ == '__main__':
    pantry_sentry = PantrySentry()
    print("Starting the Pantry Sentry")
    pantry_sentry.start()