final app for testing and deployment

app/utils/data_manager.py

@@ -3,14 +3,39 @@ Data storage utilities for QR codes
 """
 
 import uuid
+import json
+import os
 from datetime import datetime
 
-# In-memory storage for QR codes (in production, use a database)
-qr_codes_db = {}
+# Data storage directory
+DATA_DIR = 'data'
+QR_CODES_FILE = os.path.join(DATA_DIR, 'qr_codes.json')
+
+# Ensure data directory exists
+os.makedirs(DATA_DIR, exist_ok=True)
 
 class QRDataManager:
     def __init__(self):
-        pass
+        self.qr_codes_db = self._load_qr_codes()
+    
+    def _load_qr_codes(self):
+        """Load QR codes from JSON file"""
+        try:
+            if os.path.exists(QR_CODES_FILE):
+                with open(QR_CODES_FILE, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            return {}
+        except Exception as e:
+            print(f"Error loading QR codes: {e}")
+            return {}
+    
+    def _save_qr_codes(self):
+        """Save QR codes to JSON file"""
+        try:
+            with open(QR_CODES_FILE, 'w', encoding='utf-8') as f:
+                json.dump(self.qr_codes_db, f, indent=2, ensure_ascii=False)
+        except Exception as e:
+            print(f"Error saving QR codes: {e}")
     
     def save_qr_record(self, qr_type, content, settings, image_data, page_id=None):
         """Save QR code record to database"""
@@ -27,32 +52,45 @@ class QRDataManager:
         if page_id:
             qr_record['page_id'] = page_id
             
-        qr_codes_db[qr_id] = qr_record
+        self.qr_codes_db[qr_id] = qr_record
+        self._save_qr_codes()  # Persist to file
         return qr_id
     
     def get_qr_record(self, qr_id):
         """Get QR code record"""
-        return qr_codes_db.get(qr_id)
+        # Reload data from file to ensure we have the latest data
+        self.qr_codes_db = self._load_qr_codes()
+        return self.qr_codes_db.get(qr_id)
+    
+    def get_qr_code(self, qr_id):
+        """Get QR code record (alias for compatibility)"""
+        return self.get_qr_record(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]
+        if qr_id in self.qr_codes_db:
+            del self.qr_codes_db[qr_id]
+            self._save_qr_codes()  # Persist to file
             return True
         return False
     
     def list_qr_codes(self):
         """List all QR codes"""
+        # Reload data from file to ensure we have the latest data
+        self.qr_codes_db = self._load_qr_codes()
         qr_list = []
-        for qr_id, qr_data in qr_codes_db.items():
+        for qr_id, qr_data in self.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"]}'
+                'preview': f'data:image/png;base64,{qr_data["image_data"]}',
+                'page_id': qr_data.get('page_id')  # Include page_id if it exists
             })
         return qr_list
     
     def qr_exists(self, qr_id):
         """Check if QR code exists"""
-        return qr_id in qr_codes_db
+        # Reload data from file to ensure we have the latest data
+        self.qr_codes_db = self._load_qr_codes()
+        return qr_id in self.qr_codes_db