61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
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()
|