Automatic Commit By liuyuqi

models.py

@@ -4,6 +4,13 @@ from datetime import datetime
 db = SQLAlchemy()
 
 class ImageModel(db.Model):
-    """ 用户 """
+    """ image model """
     id = db.Column(db.Integer, primary_key=True)
+    image = db.Column(db.LargeBinary, nullable=False)
+    created_at = db.Column(db.DateTime, default=datetime.utcnow)
+    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    def __init__(self, image):
+        self.image = image
+        
     

test/my_unittest.py

@@ -0,0 +1,15 @@
+import unittest
+from flask_image_matcher.models import ImageModel
+
+class TestImageModel(unittest.TestCase):
+
+    def test_init(self):
+        # Test with valid image
+        image = b'valid_image_data'
+        model = ImageModel(image)
+        self.assertEqual(model.image, image)
+
+        # Test with empty image
+        image = b''
+        with self.assertRaises(ValueError):
+            ImageModel(image)