59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
"""
|
|
Data storage utilities for QR codes
|
|
"""
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
# In-memory storage for QR codes (in production, use a database)
|
|
qr_codes_db = {}
|
|
|
|
class QRDataManager:
|
|
def __init__(self):
|
|
pass
|
|
|
|
def save_qr_record(self, qr_type, content, settings, image_data, page_id=None):
|
|
"""Save QR code record to database"""
|
|
qr_id = str(uuid.uuid4())
|
|
qr_record = {
|
|
'id': qr_id,
|
|
'type': qr_type,
|
|
'content': content,
|
|
'settings': settings,
|
|
'created_at': datetime.now().isoformat(),
|
|
'image_data': image_data
|
|
}
|
|
|
|
if page_id:
|
|
qr_record['page_id'] = page_id
|
|
|
|
qr_codes_db[qr_id] = qr_record
|
|
return qr_id
|
|
|
|
def get_qr_record(self, qr_id):
|
|
"""Get QR code record"""
|
|
return qr_codes_db.get(qr_id)
|
|
|
|
def delete_qr_record(self, qr_id):
|
|
"""Delete QR code record"""
|
|
if qr_id in qr_codes_db:
|
|
del qr_codes_db[qr_id]
|
|
return True
|
|
return False
|
|
|
|
def list_qr_codes(self):
|
|
"""List all QR codes"""
|
|
qr_list = []
|
|
for qr_id, qr_data in qr_codes_db.items():
|
|
qr_list.append({
|
|
'id': qr_id,
|
|
'type': qr_data['type'],
|
|
'created_at': qr_data['created_at'],
|
|
'preview': f'data:image/png;base64,{qr_data["image_data"]}'
|
|
})
|
|
return qr_list
|
|
|
|
def qr_exists(self, qr_id):
|
|
"""Check if QR code exists"""
|
|
return qr_id in qr_codes_db
|