Сделали ORM систему с типами полей

main.py

@@ -2,62 +2,126 @@ import sqlite3
 
 db_name = 'mydb.db'
 
-class User:
-    def __init__(self,name,email,age):
-        self.__id = None
-        self.__name = name
-        self.__email = email
-        self.__age = age
+class Field:
+    def __init__(self,name,column_type):
+        self.name = name
+        self.type = column_type
+    def __str__(self):
+        return f'{self.name}:{self.type}'
 
-    @property
-    def name(self):
-        return self.__name
-    @name.setter
-    def name(self,value):
-        self.__name = value
-        
-    @property
-    def email(self):
-        return self.__email
+class IntegerField(Field):
+    def __init__(self,name):
+        super().__init__(name,'INTEGER')
 
-    @email.setter
-    def email(self,value):
-        if '@' in value:
-            self.__email = value
-        else:
-            raise ValueError('Не корректный адрес почты')
-    @property
-    def age(self):
-        return self.__age
-    @age.setter
-    def age(self,value):
-        if 0 <= value <= 120:
-            self.__age = value
-        else:
-            raise ValueError('Неправильно указан возраст')
+class StringField(Field):
+    def __init__(self,name):
+        super().__init__(name,'TEXT')
+    
+class ModelMeta(type):
+    def __new__(cls, name, bases, attrs):
+        if name == 'BaseModel':
+            return super().__new__(cls,name,bases,attrs)
+
+        fields = dict()
+        for k,v in attrs.items():
+            if isinstance(v,Field):
+                fields[k] = v
+
+        attrs['_fields'] = fields
+        return super().__new__(cls,name,bases,attrs)
         
+
+class BaseModel(metaclass=ModelMeta):
+    def __init__(self,**kwargs):
+        for k,v in kwargs.items():
+            setattr(self,k,v)   #{'name': 'Alex'} self.name = 'Alex'
+
+    @classmethod
+    def get_by_id(cls,id):
+        conn = sqlite3.connect(db_name)
+        cursor = conn.cursor()
+        property_list = ['id']+[f"{k}" for k, v in cls.__dict__.items() if isinstance(v, property)]
+        property_string = ','.join(property_list)
+        query = f"SELECT {property_string} FROM {cls.__name__} WHERE id = ?"
+        cursor.execute(query,(id,))
+        result = cursor.fetchone()
+        property_dict = dict()
+        for index in range(len(property_list)):
+            property_dict[property_list[index]] = result[index]
+        conn.close()
+        return cls(**property_dict)
+
     def save(self):
         conn = sqlite3.connect(db_name)
         cursor = conn.cursor()
-        property_list = [f"{k}" for k, v in User.__dict__.items() if isinstance(v, property)]
-        if '_User__id' in self.__dict__.keys():
+
+        fields = []
+        values = []
+        for name, field in self._fields.items():
+            fields.append(field.name)
+            values.append(getattr(self,name,None))
+
+        query =f'''INSERT INTO {self.__class__.__name__.lower()}({','.join(fields)})
+                    VALUES({','.join(['?']*len(values))})'''
+
+        print(query,values)
+        conn.close()
+'''
+        property_list = [f"{k}" for k, v in self.__class__.__dict__.items() if isinstance(v, property)]
+        
+        if self.__id != None:
             update_string = ','.join([f'`{p_name}`= ? ' for p_name in property_list])
             update_values = [getattr(self,p) for p in property_list] + [self.__id]
-            query = f"UPDATE {User.__name__} SET {update_string} WHERE id = ?"
+            query = f"UPDATE {self.__class__.__name__} SET {update_string} WHERE id = ?"
             print(query,update_values)
+            cursor.execute(query,update_values)
+            conn.commit()
         else:
-            property_count = ('?,'*len(property_list))[:-1] #','.join(['?' for _ in range(len(property_list))])
+            property_count = ('?,'*len(property_list))[:-1]
             property_names = ','.join(property_list)
             property_values = [getattr(self,p) for p in property_list]
-            query = f"INSERT INTO {User.__name__}({property_names}) VALUES({property_count})"
-            print(query,property_values)
+            query = f"INSERT INTO {self.__class__.__name__}({property_names}) VALUES({property_count})"
             cursor.execute(query,property_values)
             conn.commit()
             self.__id = cursor.lastrowid
-        conn.close()
-#print(User.__dict__)
+'''
 
-user1 = User('Igor','igor@example.com',44)
-user1.save()
-user1.age = 55
+def validate_email(func):
+    def wrapper(self,*args,**kwargs):
+        if hasattr(self,'email'):
+            if '@' not in self.email:
+                raise ValueError('Некоректный email')
+            
+        return func(self,*args,**kwargs)
+    return wrapper
+
+class User(BaseModel):
+    id = IntegerField('id')
+    name = StringField('name')
+    age = Field('age','REAL')
+    email = StringField('email')
+
+    @validate_email
+    def save(self):
+        super().save()
+
+'''
+class Product(BaseModel):
+    def __init__(self,name,category,price,id=None):
+        super().__init__(id)
+        self.__name = name
+        self.__category = category
+        self.__price = price
+    @property
+    def name(self):
+        return self.__name
+    @property
+    def category(self):
+        return self.__category
+    @property
+    def price(self):
+        return self.__price
+'''
+
+user1 = User(name='John',age=18, email='john@example.com')
 user1.save()