Сделали BaseModel

main.py

@@ -2,9 +2,50 @@ import sqlite3
 
 db_name = 'mydb.db'
 
-class User:
+class BaseModel:
-    def __init__(self,name,email,age):
+    def __init__(self,id):
-        self.__id = None
+        self.__id = id
+
+    @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 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 {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]
+            property_names = ','.join(property_list)
+            property_values = [getattr(self,p) for p in property_list]
+            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()
+
+class User(BaseModel):
+    def __init__(self,name,email,age,id = None):
+        super().__init__(id)
         self.__name = name
         self.__email = email
         self.__age = age
@@ -36,28 +77,30 @@ class User:
         else:
             raise ValueError('Неправильно указан возраст')
 
-    def save(self):
+class Product(BaseModel):
-        conn = sqlite3.connect(db_name)
+    def __init__(self,name,category,price,id=None):
-        cursor = conn.cursor()
+        super().__init__(id)
-        property_list = [f"{k}" for k, v in User.__dict__.items() if isinstance(v, property)]
+        self.__name = name
-        if '_User__id' in self.__dict__.keys():
+        self.__category = category
-            update_string = ','.join([f'`{p_name}`= ? ' for p_name in property_list])
+        self.__price = price
-            update_values = [getattr(self,p) for p in property_list] + [self.__id]
+    @property
-            query = f"UPDATE {User.__name__} SET {update_string} WHERE id = ?"
+    def name(self):
-            print(query,update_values)
+        return self.__name
-        else:
+    @property
-            property_count = ('?,'*len(property_list))[:-1] #','.join(['?' for _ in range(len(property_list))])
+    def category(self):
-            property_names = ','.join(property_list)
+        return self.__category
-            property_values = [getattr(self,p) for p in property_list]
+    @property
-            query = f"INSERT INTO {User.__name__}({property_names}) VALUES({property_count})"
+    def price(self):
-            print(query,property_values)
+        return self.__price
-            cursor.execute(query,property_values)
-            conn.commit()
-            self.__id = cursor.lastrowid
-        conn.close()
-#print(User.__dict__)
 
+product1 = Product('Свинина, шея','Мясные продукты',250)
+product1.save()
+'''
+user2 = User.get_by_id(9)
+user2.name = 'Andrey'
+user2.save()
 user1 = User('Igor','igor@example.com',44)
 user1.save()
 user1.age = 55
 user1.save()
+'''