from peewee import *
from datetime import datetime, time
import json

# Create database instance
db = SqliteDatabase('scanner.db')

class BaseModel(Model):
    class Meta:
        database = db

class BarcodeDB(BaseModel):
    barcode = CharField(unique=True)
    product_info = TextField()

    class Meta:
        table_name = 'barcode_db'

class ScanLog(BaseModel):
    barcode = CharField()
    scan_at = DateTimeField(default=datetime.now)

    class Meta:
        table_name = 'scan_logs'

    @classmethod
    def get_today_logs(cls):
        # Get today's start and end timestamps
        today = datetime.now().date()
        today_start = datetime.combine(today, time.min)  # Start of day (00:00:00)
        today_end = datetime.combine(today, time.max)    # End of day (23:59:59.999999)
        
        # Query logs for today
        return cls.select().where(cls.scan_at.between(today_start, today_end))
    
    def serialize(self):
        # Try to get product info from BarcodeDB
        name = None
        try:
            product = BarcodeDB.get(BarcodeDB.barcode == self.barcode)
            product_info = json.loads(product.product_info)
            name = product_info.get('name')
        except BarcodeDB.DoesNotExist:
            pass
            
        return {
            'barcode': self.barcode,
            'name': name,
            'scan_at': self.scan_at
        }



# Create tables
def create_tables():
    with db:
        db.create_tables([BarcodeDB, ScanLog])

if __name__ == '__main__':
    create_tables()